def zip_files()

in stratozone-aws-export.py [0:0]

• 10 lines of code
• 6 McCabe index (conditional complexity)

def zip_files(dir_name, zip_file_name):
  """Compress generated files into zip file for import into stratozone.

  Args:
    dir_name: source directory
    zip_file_name: name of the file to be created

  """
  csv_filter = lambda name: 'csv' in name or 'json' in name

  if os.path.exists(zip_file_name):
    os.remove(zip_file_name)

  with zipfile.ZipFile(zip_file_name, 'w') as zip_obj:
    # Iterate over all the files in directory
    for folder_name, sub_folder, file_names in os.walk(dir_name):
      for file_name in file_names:
        if csv_filter(file_name):
          file_path = os.path.join(folder_name, file_name)
          zip_obj.write(file_path, os.path.basename(file_path))