diff --git a/main.py b/main.py index 076b86a..48477fe 100755 --- a/main.py +++ b/main.py @@ -70,24 +70,18 @@ def main(): extension: str = file.rpartition(".")[-1] or "" if extension != "": extension = "." + extension - newname = input("Enter new name: ") + newname = input("New name: ") else: newname = file if args.classify: - 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( + 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 - 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":