def main(argv=None): mod_docstring = sys.modules[__name__].__doc__ arg_parser = ArgumentParser(description=mod_docstring) add_common_submit_args(arg_parser) arg_parser.add_argument('--url', default=DEFAULT_CLIENT_URL, help=HELP_URL) arg_parser.add_argument('--amqp_url', default=DEFAULT_CLIENT_URL, help=HELP_AMQP_URL) arg_parser.add_argument('--private_token', default=None, help=HELP_PRIVATE_TOKEN) # TODO: choices... arg_parser.add_argument('--default_file_action', default="none") arg_parser.add_argument('--file_action_config', default=None) arg_parser.add_argument('--transport', default=None, choices=["urllib", "curl"], help=HELP_TRANSPORT) # set to curl to use pycurl arg_parser.add_argument('--suppress_output', default=False, action="store_true", help=HELP_SUPPRESS_OUTPUT) arg_parser.add_argument('--disable_cleanup', dest="cleanup", default=True, action="store_false", help=HELP_DISABLE_CLEANUP) arg_parser.add_argument('--server', default=False, action="store_true", help=HELP_SERVER) arg_parser.add_argument('--job_id', default=None, help=HELP_JOB_ID) arg_parser.add_argument('--command', help=HELP_COMMAND) arg_parser.add_argument('--working_directory', default=".", help=HELP_WORKING_DIRECTORY) arg_parser.add_argument('--result_json', default=None) arg_parser.add_argument('--output', default=[], action="append", help=HELP_OUTPUT) arg_parser.add_argument('--output_pattern', default=[], action="append", help=HELP_OUTPUT_PATTERN) args = arg_parser.parse_args(argv) if args.server: return run_server_for_job(args) else: failed = _run_client_for_job(args) if failed: return 1 else: return 0
def main(args=None): mod_docstring = sys.modules[__name__].__doc__ arg_parser = ArgumentParser(description=mod_docstring) add_common_submit_args(arg_parser) args = arg_parser.parse_args(args) run_server_for_job(args)