from modules import printer
import glob
import os


def get_dir_size(directory, files):
    total_size = 0
    for f in files:
        fp = glob.glob(f'{directory}/{f}*')[0]
        if not os.path.islink(fp):
            total_size += os.path.getsize(fp)
    return total_size


def get_compression(orig, comp):
    processed_files = []
    for file in os.listdir(comp):
        processed_files.append(os.path.splitext(file)[0])

    comp = 100 - int((get_dir_size(comp, processed_files) / get_dir_size(orig, processed_files)) * 100)
    if comp < 0:
        printer.warning(f'Compression: {comp}%')
        printer.warning("The resulting files are larger than the original ones!")
    else:
        printer.info(f'Compression: {comp}%')


def help_message():
    text = "Usage: main.py {folder}"
    return text