def main()

in image_generation/collect_scenes.py [0:0]

• 30 lines of code
• 5 McCabe index (conditional complexity)

def main(args):
  input_files = os.listdir(args.input_dir)
  scenes = []
  split = None
  for filename in os.listdir(args.input_dir):
    if not filename.endswith('.json'):
      continue
    path = os.path.join(args.input_dir, filename)
    with open(path, 'r') as f:
      scene = json.load(f)
    scenes.append(scene)
    if split is not None:
      msg = 'Input directory contains scenes from multiple splits'
      assert scene['split'] == split, msg
    else:
      split = scene['split']
  scenes.sort(key=lambda s: s['image_index'])
  for s in scenes:
    print(s['image_filename'])
  output = {
    'info': {
      'date': args.date,
      'version': args.version,
      'split': split,
      'license': args.license,
    },
    'scenes': scenes
  }
  with open(args.output_file, 'w') as f:
    json.dump(output, f)