compress_folders.py

import sys

from mtlogger import logger
from mtprompt import Prompt
from mtvalidation import to_bool, to_dir, to_int, to_list

from _common import compress_child_folders
from _constants import FAILED, INCOMPLETE, SUCCEEDED, ZIP_TYPES


def main():
  parent_dir = (
    to_dir(sys.argv[1])
    if len(sys.argv) > 1
    else Prompt.dir(
      'Enter the path to the directory containing the folders you want to compress'
    )
  )
  output_type = sys.argv[2] if len(sys.argv) > 2 else ZIP_TYPES[0]
  delete_original = to_bool(sys.argv[3]) if len(sys.argv) > 3 else False
  exclusion_patterns = to_list(sys.argv[4]) if len(sys.argv) > 4 else None
  min_depth = to_int(sys.argv[5]) if len(sys.argv) > 5 else 1
  max_depth = to_int(sys.argv[6]) if len(sys.argv) > 6 else min_depth

  logger.log(f'Compressing folders in "{parent_dir}"...')
  status = compress_child_folders(
    parent_dir,
    output_type,
    delete_original,
    exclusion_patterns,
    min_depth,
    max_depth,
  )

  if status == SUCCEEDED:
    logger.success(f'Finished compressing folders in "{parent_dir}".')

  elif status == INCOMPLETE:
    logger.warn(f'Compression did not complete for folders in "{parent_dir}".')

  elif status == FAILED:
    logger.error(f'Failed to compress some folders in "{parent_dir}".')

  return status


if __name__ == '__main__':
  try:
    status = main()
  except Exception as ex:
    logger.unhandled_error(ex)
    status = FAILED

  Prompt.enter_to_exit(timeout=status == SUCCEEDED)