def main()

in cookbooks/aws-parallelcluster-slurm/files/default/head_node_slurm/slurm/pcluster_fleet_config_generator.py [0:0]

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

def main():
    try:
        logging.basicConfig(
            level=logging.INFO, format="%(asctime)s - [%(name)s:%(funcName)s] - %(levelname)s - %(message)s"
        )
        log.info("Running ParallelCluster Fleet Config Generator")
        parser = argparse.ArgumentParser(description="Take in fleet configuration generator related parameters")
        parser.add_argument("--output-file", help="The output file for generated json fleet config", required=True)
        parser.add_argument(
            "--input-file",
            help="Yaml file containing pcluster CLI configuration file with default values",
            required=True,
        )
        args = parser.parse_args()
        generate_fleet_config_file(args.output_file, args.input_file)
    except Exception as e:
        log.exception("Failed to generate Fleet configuration, exception: %s", e)
        raise