diff --git a/main.py b/main.py index 48477fe..076b86a 100755 --- a/main.py +++ b/main.py @@ -70,18 +70,24 @@ def main(): extension: str = file.rpartition(".")[-1] or "" if extension != "": extension = "." + extension - newname = input("New name: ") + newname = input("Enter new name: ") else: newname = file if args.classify: - category: str = input("Category: ") - if not os.path.exists(outputFolder + category) or not os.path.isdir( + print(f"Existing categories: {os.listdir(outputFolder)}") + category: str = input("Enter category or delete to permanently delete file: ") + if category == "delete": + os.remove(inputFolder + file) + elif 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 + if os.path.exists(path): + newname = input("Name already in use, please enter another name: ") + path = outputFolder + category + "/" + newname + extension shutil.move(inputFolder + file, path) if args.reformat: if extension != ".mp4":