def get_script_params()

in scripts/update_auto_launch_config.py [0:0]

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

def get_script_params(cli_args=None):
    parser = ArgumentParser(description="Helper script to update Nimble Studio Auto Workstation Scheduler config for a studio user at a certain start time.")

    parser.add_argument("-u", "--user-name", dest="user_name", help="The user name of the studio user to update configuration for",
                        required=False)
    parser.add_argument("-o", "--sso-id", dest="sso_id", help="The SSO ID of the user to update configuration for. This takes precedence over user name",
                        required=False)
    parser.add_argument("-s", "--studio-id", dest="studio_id", help="The studio ID to set configuration for",
                        required=False)
    parser.add_argument("-t", "--start-time", dest="start_time", help="The time (UTC) to start the launch of the workstation formatted as HH:MM",
                        required=False)
    parser.add_argument("-d", "--days", dest="days", help="The weekdays to launch the workstation at this time, comma delimited (ex: Monday,Tuesday)",
                        required=False)
    parser.add_argument("-l", "--launch-profile", dest="launch_profile", help="The ID of the launch profile to launch for the workstation",
                        required=False)
    parser.add_argument("-i", "--streaming-image", dest="streaming_image", help="The streaming image component ID to use for the launch profile",
                        required=False)
    parser.add_argument("-n", "--instance-type", dest="instance_type", help="The instance type and size to use for the workstation (ex: g4dn.2xlarge)",
                        required=False)
    parser.add_argument("-e", "--enable", dest='enabled', action='store_true', help="Enable auto launch for this config")
    parser.add_argument("-D", "--disable", dest='enabled', action='store_false', help="Disable auto launch for this config")
    parser.set_defaults(enabled=True)
    parser.add_argument("-x", "--skip-validation", dest="skip_validation", action='store_true', help="Skip input validation and prompts")
    parser.set_defaults(skip_validation=False)

    if not cli_args:
        return parser.parse_args()
    else:
        return parser.parse_args(cli_args.split())