commit 18385338507bf4c3f5959e0de68a3ad33f291353 Author: Arturo Filastò arturo@filasto.net Date: Wed Mar 30 14:08:01 2016 +0200
[Refactoring] oonicli logic --- ooni/oonicli.py | 37 ++++++++++++++++--------------------- 1 file changed, 16 insertions(+), 21 deletions(-)
diff --git a/ooni/oonicli.py b/ooni/oonicli.py index 9509a18..63312b5 100644 --- a/ooni/oonicli.py +++ b/ooni/oonicli.py @@ -209,25 +209,22 @@ def setupAnnotations(global_options): global_options["annotations"] = annotations return annotations
-def setupCollector(global_options, net_test_loader): - collector = None - if not global_options['no-collector']: - if global_options['collector']: - collector = global_options['collector'] - elif 'collector' in config.reports \ - and config.reports['collector']: - collector = config.reports['collector'] - elif net_test_loader.collector: - collector = net_test_loader.collector - - if collector and collector.startswith('httpo:') \ +def setupCollector(global_options, collector_address): + if global_options['collector']: + collector_address = global_options['collector'] + elif 'collector' in config.reports \ + and config.reports['collector']: + collector_address = config.reports['collector'] + + if collector_address.startswith('httpo:') \ and (not (config.tor_state or config.tor.socks_port)): raise errors.TorNotRunning - return collector + return collector_address
def createDeck(global_options, url=None): - log.msg("Creating deck for: %s" % (url)) + if url: + log.msg("Creating deck for: %s" % (url))
if global_options['no-yamloo']: log.msg("Will not write to a yamloo report file") @@ -311,18 +308,16 @@ def runTestWithDirector(director, global_options, url=None, # If a collector is not specified in the deck, or the # deck is a singleton, the default collector set in # ooniprobe.conf will be used - - collector = setupCollector(global_options, net_test_loader) - - if collector and collector.startswith('httpo:') \ - and (not (config.tor_state or config.tor.socks_port)): - raise errors.TorNotRunning + collector_address = None + if not global_options['no-collector']: + collector_address = setupCollector(global_options, + net_test_loader.collector)
net_test_loader.annotations = global_options['annotations']
yield director.startNetTest(net_test_loader, global_options['reportfile'], - collector, + collector_address, global_options['no-yamloo'])
d.addCallback(setup_nettest)