#!/bin/python3

from modules import compressor
from modules import printer
from modules import utils
import sys
import os

try:
    orig_folder = sys.argv[1]
except IndexError:
    print(utils.help_message())
    exit()

try:
    os.mkdir(f"{orig_folder}_compressed")
    printer.info(f"Created {orig_folder}_compressed folder")
except OSError:
    printer.warning(f"{orig_folder}_compressed already exist!")
    pass

printer.info("Compression started!")
compressor.compress(orig_folder)

if len(os.listdir(path=orig_folder)) == len((os.listdir(path=f"{orig_folder}_compressed"))):
    printer.info("Success!")
    utils.get_compression(orig_folder, f"{orig_folder}_compressed")
else:
    printer.warning("Some files failed to compress!")
    utils.get_compression(orig_folder, f"{orig_folder}_compressed")