def main(HarnessClass): (options, args, database_options) = parse_args(HarnessClass) pickles_path = join(args['pickles_directory'], HarnessClass.__name__, options.run_name) try: makedirs(pickles_path) except OSError, e: if e.errno != EEXIST: raise harness = HarnessClass(options) process_sessions(harness, args['database_backend'], args['database_name'], database_options, pickles_path, options.temp_pickles_dir, options.workers, options.ignore_pickles, options.cached_global_context) # You can't run this harness, since it's an abstract class. To run your own # harness, put the following lines in the module with your harness, replacing # "Harness" with the name of your Harness subclass. # # from bismarkpassive import main # ... # if __name__ == '__main__': # main(Harness)
parse_coordinator_args(parser) options, args = parser.parse_args() if len(args) != 4: parser.error('Missing required option') mandatory = { 'harness': args[0], 'updates_directory': args[1], 'index_filename': args[2], 'pickles_directory': args[3] } return options, mandatory def main(): (options, args) = parse_args() pickles_path = join(args['pickles_directory'], args['harness']) try: makedirs(pickles_path) except OSError, e: if e.errno != errno.EEXIST: raise coordinators = map(lambda cl: cl(options), harnesses[args['harness']]) process_sessions(coordinators, args['updates_directory'], args['index_filename'], pickles_path, options.temp_pickles_dir, options.workers, not options.disable_refresh) if __name__ == '__main__': main()
help='Disable refresh of index before processing') parse_coordinator_args(parser) options, args = parser.parse_args() if len(args) != 4: parser.error('Missing required option') mandatory = { 'harness': args[0], 'updates_directory': args[1], 'index_filename': args[2], 'pickles_directory': args[3] } return options, mandatory def main(): (options, args) = parse_args() pickles_path = join(args['pickles_directory'], args['harness']) try: makedirs(pickles_path) except OSError, e: if e.errno != errno.EEXIST: raise coordinators = map(lambda cl: cl(options), harnesses[args['harness']]) process_sessions(coordinators, args['updates_directory'], args['index_filename'], pickles_path, options.temp_pickles_dir, options.workers, not options.disable_refresh) if __name__ == '__main__': main()