diff options
Diffstat (limited to 'soledad_sync.py')
| -rw-r--r-- | soledad_sync.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/soledad_sync.py b/soledad_sync.py index e291d68..e53f49c 100644 --- a/soledad_sync.py +++ b/soledad_sync.py @@ -1,4 +1,5 @@ import os +import json from ConfigParser import ConfigParser from leap.soledad.client.api import Soledad from twisted.internet import defer @@ -11,6 +12,8 @@ HOST = parser.get('server', 'host') UUID = parser.get('client', 'uuid') NUM_DOCS = int(parser.get('sync', 'num_docs')) PAYLOAD = parser.get('sync', 'payload') +AUTH_TOKEN = parser.get('sync', 'auth-token') +STATS_FILE = parser.get('test', 'stats-file') @@ -20,6 +23,7 @@ DO_THESEUS = os.environ.get('THESEUS', False) phase = 0 + def _get_soledad_instance_from_uuid(uuid, passphrase, basedir, server_url, cert_file, token): secrets_path = os.path.join(basedir, '%s.secret' % uuid) @@ -55,7 +59,12 @@ def upload_soledad_stuff(): t.install() s = _get_soledad_instance_from_uuid( - UUID, 'pass', '/tmp/soledadsync', HOST, '', 'an-auth-token') + UUID, 'pass', '/tmp/soledadsync', HOST, '', AUTH_TOKEN) + + def dumpStats(_): + stats = s.sync_stats() + with open(STATS_FILE, 'w') as f: + f.write(json.dumps(stats)) def do_sync(_): global phase @@ -63,6 +72,7 @@ def upload_soledad_stuff(): phase += 1 d = s.sync() d.addCallback(onSyncDone) + d.addCallback(dumpStats) return d def stop_tracing(_): |