def main()

in build_artifacts/v2/v2.5/v2.5.0/dirs/etc/sagemaker-ui/workflows/workflow_client.py [0:0]

• 20 lines of code
• 2 McCabe index (conditional complexity)

def main():
    parser = argparse.ArgumentParser(description="Workflow local runner client")
    subparsers = parser.add_subparsers(dest="command", help="Available commands")

    update_status_parser = subparsers.add_parser(
        "update-local-runner-status", help="Update status of local runner"
    )
    update_status_parser.add_argument("--status", type=str, required=True, help="Status to update")
    update_status_parser.add_argument(
        "--detailed-status", type=str, required=False, help="Detailed status text"
    )

    start_parser = subparsers.add_parser("start-local-runner", help="Start local runner")

    stop_parser = subparsers.add_parser("stop-local-runner", help="Stop local runner")

    args = parser.parse_args()

    # create the request session
    session = requests.Session()
    # populate XSRF cookie
    session.get(JUPYTERLAB_URL)

    kwargs = vars(args) | {"session": session}

    if args.command in COMMAND_REGISTRY:
        COMMAND_REGISTRY[args.command](**kwargs)
    else:
        parser.print_help()