Compare commits

...

3 commits

12
main.py
View file

@ -70,18 +70,24 @@ def main():
extension: str = file.rpartition(".")[-1] or "" extension: str = file.rpartition(".")[-1] or ""
if extension != "": if extension != "":
extension = "." + extension extension = "." + extension
newname = input("New name: ") newname = input("Enter new name: ")
else: else:
newname = file newname = file
if args.classify: if args.classify:
category: str = input("Category: ") print(f"Existing categories: {os.listdir(outputFolder)}")
if not os.path.exists(outputFolder + category) or not os.path.isdir( 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 outputFolder + category
): ):
os.mkdir(outputFolder + category) os.mkdir(outputFolder + category)
else: else:
category: str = "" category: str = ""
path: str = outputFolder + category + "/" + newname + extension 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) shutil.move(inputFolder + file, path)
if args.reformat: if args.reformat:
if extension != ".mp4": if extension != ".mp4":