def parse_arguments()

in composer/tools/composer_migrate.py [0:0]

• 50 lines of code
• 1 McCabe index (conditional complexity)

def parse_arguments() -> Dict[Any, Any]:
    """Parses command line arguments."""
    argument_parser = argparse.ArgumentParser(
        usage="Script for migrating environments from Composer 2 to Composer 3.\n"
    )

    argument_parser.add_argument(
        "--project",
        type=str,
        required=True,
        help="Project name of the Composer environment to migrate.",
    )
    argument_parser.add_argument(
        "--location",
        type=str,
        required=True,
        help="Location of the Composer environment to migrate.",
    )
    argument_parser.add_argument(
        "--source_environment",
        type=str,
        required=True,
        help="Name of the Composer 2 environment to migrate.",
    )
    argument_parser.add_argument(
        "--target_environment",
        type=str,
        required=True,
        help="Name of the Composer 3 environment to create.",
    )
    argument_parser.add_argument(
        "--target_airflow_version",
        type=str,
        default="2",
        help="Airflow version for the Composer 3 environment.",
    )
    argument_parser.add_argument(
        "--dry_run",
        action="store_true",
        default=False,
        help=(
            "If true, script will only print the config for the Composer 3"
            " environment."
        ),
    )
    argument_parser.add_argument(
        "--sdk_endpoint",
        type=str,
        default="https://composer.googleapis.com/",
        required=False,
    )

    return argument_parser.parse_args()