compress-files.py

from PIL import Image
import os

from lib.inputs import yesNo
from lib.files import traverse


# Absolute path to the directory you want to search
ROOT = r"C:\Users\[Users]\[pythonFiles]"
PREFIX = ''
SUFFIX = ''


def compressFiles(path):
    assert os.path.isdir(path)
    files = [f for f in os.listdir(path) if os.path.isfile(os.path.join(path, f)) if str(f).startswith(PREFIX) and str(f).endswith(SUFFIX)]

    print("---FILES---")
    for file in files:
        print(file)

    if yesNo("Compress files (y/n): "):
        count = 0
        for file in files:
            try:
                maxWidth = 1920
                filePath = os.path.join(path, file)
                originalSize = os.path.getsize(filePath)
                image = Image.open(filePath)
                width, height = image.size
                aspectRatio = width / height
                newHeight = maxWidth / aspectRatio
                image = image.resize((maxWidth, round(newHeight)))
                image.save(filePath, optimize=True, quality=85)
                finalSize = os.path.getsize(filePath)
                print(f"Removed {round((originalSize-finalSize)/1024/1024, 3)}MB")
                count += 1
            except Exception as e:
                print(e)
        print(f"Compressed {count} files")

    input("Press ENTER to exit")


def main():
    assert os.path.isdir(ROOT)

    sourceDir = traverse(ROOT, "Choose source directory (press ENTER to choose CWD): ")

    compressFiles(sourceDir)


if __name__ == '__main__':
    main()