diff --git a/main.py b/main.py deleted file mode 100755 index 48477fe..0000000 --- a/main.py +++ /dev/null @@ -1,101 +0,0 @@ -# Python project to classify memes - -import os -import subprocess -import argparse -import time -import shutil - - -def main(): - parser = argparse.ArgumentParser() - - parser.add_argument("INPUT", help="folder with input files", type=str) - - parser.add_argument("OUTPUT", help="folder for output", type=str) - - parser.add_argument("-n", "--name", action="store_true") - - parser.add_argument("-c", "--classify", action="store_true") - - parser.add_argument( - "-r", - "--reformat", - action="store_true", - help="automatically convert videos to mp4 using ffmpeg", - ) - - parser.add_argument("-v", "--verbose", action="store_true") - - args = parser.parse_args() - - inputFolder: str = args.INPUT - if not inputFolder.endswith("/"): - inputFolder = inputFolder + "/" - - outputFolder: str = args.OUTPUT - if not outputFolder.endswith("/"): - outputFolder = outputFolder + "/" - - if not os.path.isdir(inputFolder): - print(f"\033[91mERROR: {inputFolder} does not exist") - exit(1) - if not os.path.isdir(outputFolder): - query = input(f"{outputFolder} does not exist. Do you want to create it? [Y/n]") - yes: tuple(str) = "yes", "y", "" - no: tuple(str) = "no", "n" - if query.lower() in yes: - if args.verbose: - print(f"Creating folder {outputFolder}") - os.mkdir(outputFolder) - elif query.lower() in no: - print("\033[91mNo output folder, aborting...") - exit(1) - - files: [str] = os.listdir(inputFolder) - for index, file in enumerate(files): - if os.path.isdir(inputFolder + file): - files.pop(index) - if args.verbose: - for file in files: - print(file) - time.sleep(0.0001) - print(f"\nFound {len(files)} files to process") - - for id, file in enumerate(files): - print(f"Processing file {id + 1}/{len(files)}") - subprocess.run(["mpv", f"{inputFolder}{file}"]) - if args.name: - print(f"Old name: {file}") - extension: str = file.rpartition(".")[-1] or "" - if extension != "": - extension = "." + extension - newname = input("New name: ") - else: - newname = file - if args.classify: - category: str = input("Category: ") - if not os.path.exists(outputFolder + category) or not os.path.isdir( - outputFolder + category - ): - os.mkdir(outputFolder + category) - else: - category: str = "" - path: str = outputFolder + category + "/" + newname + extension - shutil.move(inputFolder + file, path) - if args.reformat: - if extension != ".mp4": - subprocess.run( - ["ffmpeg"], - ["-i"], - [path], - ["-o"], - ["mp4"], - [f"{outputFolder}{category}{newname}.mp4"], - ) - if args.verbose: - print(f"converted {newname} from {extension} to .mp4") - - -if __name__ == "__main__": - main()