commit 384775613778ae6994ccedb365530ca14491b9bb Author: Isis Lovecruft <isis@torproject.org> Date: Tue Nov 5 08:39:46 2013 +0000 Change runner.generateDescriptors() to not need whole options class. * CHANGE the parameters to bridgedb.runner.generateDescriptors() function to only need specific settings, rather than needing the whole instance of the MainOptions class. --- lib/bridgedb/Main.py | 3 ++- lib/bridgedb/runner.py | 7 ++++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/lib/bridgedb/Main.py b/lib/bridgedb/Main.py index 2ab1f5a..4f5203f 100644 --- a/lib/bridgedb/Main.py +++ b/lib/bridgedb/Main.py @@ -472,7 +472,8 @@ def runSubcommand(options, config): logging.debug("Running BridgeDB command: '%s'" % options.subCommand) if 'descriptors' in options.subOptions: - runner.generateDescriptors(options) + runner.generateDescriptors(options.subOptions['descriptors'], + config.RUNDIR) if options.subCommand == 'test': if options.subOptions['trial']: diff --git a/lib/bridgedb/runner.py b/lib/bridgedb/runner.py index 425c0e8..b477c35 100644 --- a/lib/bridgedb/runner.py +++ b/lib/bridgedb/runner.py @@ -19,7 +19,7 @@ from __future__ import print_function import sys -def generateDescriptors(options): +def generateDescriptors(count=None, rundir=None): """Run a script which creates fake bridge descriptors for testing purposes. This will run scripts/gen_bridge_descriptors to create a bridge router @@ -31,11 +31,12 @@ def generateDescriptors(options): :returns: The process returncode from the gen_bridge_descriptors script. """ import subprocess + import os.path proc = None - rundir = options['rundir'] script = 'gen_bridge_descriptors' - count = options.subOptions['descriptors'] + rundir = rundir if os.path.isdir(rundir) else None + count = count if count else 3 try: proc = subprocess.Popen([script, '-n', str(count)], close_fds=True, cwd=rundir)