def main()

in tools/genconfig/genconfig.py [0:0]

• 23 lines of code
• 4 McCabe index (conditional complexity)

def main(argv):
  if len(argv) < 3:
    usage()
    sys.exit(1)

  output_dir = argv[3] if len(argv) == 4 else '.'

  urls = []
  with open(argv[2]) as f:
    urls = [line.rstrip() for line in f]

  config = get_config(urls, argv[1])

  # Write generated template.
  with open(output_dir + '/generated.jinja', 'w') as f:
    f.write(yaml.dump(config, default_flow_style=False))
  with open(output_dir + '/generated.jinja.schema', 'w') as f:
    f.write(yaml.dump(get_generated_schema(), default_flow_style=False))

  # Write yaml config which uses template.
  with open(output_dir + '/config.yaml', 'w') as f:
    f.write(yaml.dump(get_config_dot_yaml(), default_flow_style=False))

  print(''.join(['All done! See files generated in output directory "',
                 output_dir, '".']),
        file=sys.stderr)
  print(file=sys.stderr)
  print(''.join(['You may want to modify them to have different resource names,'
                 , ' parameterized properties,'
                 , ' or references between related resources.']),
        file=sys.stderr)