From 9ea9495266ad2a9934a6a1231bc096b0874986fb Mon Sep 17 00:00:00 2001 From: OleSTEEP Date: Thu, 1 Jun 2023 16:57:40 +0300 Subject: [PATCH] FFMpeg-Compressor: Command line argumets --- FFMpeg-Compressor/main.py | 8 +++++++- FFMpeg-Compressor/modules/utils.py | 5 +++++ 2 files changed, 12 insertions(+), 1 deletion(-) 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