def main(args=None):
if not args:
args, pass_throughs = parse_args()
else:
args, pass_throughs = parse_args(args)
if args.debug:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.INFO)
if args.subparser_name == 'submit':
args_dict = args.__dict__
if not args_dict.get('config_root_dir'):
args_dict['config_root_dir'] = args_dict['base_name']
execute.submit(pass_throughs=pass_throughs, **args_dict)
elif args.subparser_name == 'bootstrap':
execute.bootstrap(pass_throughs=pass_throughs, **args.__dict__)
elif args.subparser_name == 'one-off':
execute.submit_one_off(pass_throughs=pass_throughs, **args.__dict__)
elif args.subparser_name == 'batch':
execute.submit_batch(pass_throughs=pass_throughs, **args.__dict__)
elif args.subparser_name == 'rm':
execute.rm_pipeline(pass_throughs=pass_throughs, **args.__dict__)
elif args.subparser_name == 'trigger':
execute.trigger_pipeline(pass_throughs=pass_throughs, **args.__dict__)
elif args.subparser_name == 'abort':
execute.abort_pipeline(pass_throughs=pass_throughs, **args.__dict__)
else:
# this is here so that if future subcommands are added, you don't forget to add a bit
# here to enable them.
raise NotImplementedError("Command {} is not implemented".format(
args.subparser_name))
def main(args=None):
if not args:
args = parse_args()
else:
args = parse_args(args)
if args.debug:
logging.basicConfig(level=logging.DEBUG)
else:
logging.basicConfig(level=logging.INFO)
if args.subparser_name == 'submit':
args_dict = args.__dict__
if not args_dict.get('config_root_dir'):
args_dict['config_root_dir'] = args_dict['base_name']
execute.submit(**args_dict)
elif args.subparser_name == 'bootstrap':
execute.bootstrap(**args.__dict__)
elif args.subparser_name == 'examine':
execute.compute_builds(**args.__dict__)
elif args.subparser_name == 'one-off':
execute.submit_one_off(**args.__dict__)
else:
# this is here so that if future subcommands are added, you don't forget to add a bit
# here to enable them.
raise NotImplementedError("Command {} is not implemented".format(args.subparser_name))
def test_submit(mocker):
mocker.patch.object(execute, 'subprocess')
pipeline_file = os.path.join(test_config_dir, 'plan_director.yml')
execute.submit(pipeline_file,
base_name="test",
pipeline_name="test-pipeline",
src_dir='.',
config_root_dir=os.path.join(test_data_dir, 'config-test'))