diff --git a/FFMpeg-Compressor/main.py b/FFMpeg-Compressor/main.py index 6b36d29..9d853df 100755 --- a/FFMpeg-Compressor/main.py +++ b/FFMpeg-Compressor/main.py @@ -3,14 +3,20 @@ from modules import compressor from modules import printer from modules import utils +import sys import os -orig_folder = input("Folder: ").replace("'", "") +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!") diff --git a/FFMpeg-Compressor/modules/utils.py b/FFMpeg-Compressor/modules/utils.py index 0c4f43b..84e16d7 100644 --- a/FFMpeg-Compressor/modules/utils.py +++ b/FFMpeg-Compressor/modules/utils.py @@ -23,3 +23,8 @@ def get_compression(orig, 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