raphael/meme-classifier
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added option to delete files

Browse source
This commit is contained in:
R Dittberner 2025-10-21 23:24:54 +02:00
parent 010c555e5a
commit 33ea029c37
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
main.py
View file

@ -74,8 +74,11 @@ def main():
else: else:
newname = file newname = file
if args.classify: if args.classify:
print(f"Existing categories: {os.listdir(outputFolder)}")
category: str = input("Category: ") category: str = input("Category: ")
if not os.path.exists(outputFolder + category) or not os.path.isdir( 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)