You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.3 KiB
79 lines
2.3 KiB
import os
|
|
import shutil
|
|
import ExtensionName as ext
|
|
|
|
FileName = "FileOrganizeAutomation.py"
|
|
ExtensionModule = "ExtensionName.py"
|
|
|
|
paths = {
|
|
"Downloads" : "Downloads/",
|
|
"Desktop" : "Desktop/",
|
|
"Documents" : "Documents/",
|
|
"C Drive" : "C:/",
|
|
"D Drive" : "D:/"
|
|
}
|
|
|
|
print("\n\n1:Organize Files\n2:Backup To Drive\n3:Rename Unordered Files")
|
|
choice = int(input("\nEnter choice:"))
|
|
|
|
def organizeFiles():
|
|
print("\n\nFolders...\n")
|
|
for key in paths.keys():
|
|
print(key)
|
|
|
|
print("\n1-->Choose Path from above\n2-->Set custom path\n")
|
|
choice = int(input("Enter choice:"))
|
|
if choice == 1:
|
|
path = paths[input("\nEnter the folder to be organized:")]
|
|
|
|
elif choice == 2:
|
|
path = input("Enter custom path:") + '/'
|
|
print("\n")
|
|
else:
|
|
print("Invalid Choice!")
|
|
return 0
|
|
|
|
if (path == "C:/"):
|
|
print("Cannot make changes to system folder")
|
|
return 0
|
|
list_files = os.listdir(path)
|
|
list_extension_names = set([os.path.splitext(file)[1].strip(".") for file in list_files])
|
|
#os.path.splitext returns a tuple with file name and extension split
|
|
#Eg: Filename = Hello.txt
|
|
#-----> os.path.splitext(file) returns ("Hello", ".txt")
|
|
|
|
#Create New Folders
|
|
for extension in list_extension_names:
|
|
folder_name = ext.foldername(extension)
|
|
if(folder_name == None): #Folder_name = None --> Its a folder
|
|
continue #No need to make new folder
|
|
if not os.path.exists(path + folder_name):
|
|
os.makedirs(path + folder_name)
|
|
|
|
#Move Files To Respective Folder
|
|
for file in list_files:
|
|
if(file == FileName or file == ExtensionModule):
|
|
continue
|
|
file_extension = os.path.splitext(file)[1].strip(".")
|
|
folder_name = ext.foldername(file_extension)
|
|
if(folder_name == None):
|
|
continue
|
|
if not os.path.exists(path + folder_name + "/" + file):
|
|
shutil.move(path + file, path + folder_name + "/" + file)
|
|
print("Moved {} to {}".format(file, path + folder_name))
|
|
return 0
|
|
|
|
|
|
if(choice == 1):
|
|
organizeFiles()
|
|
elif(choice == 2):
|
|
print("Backup feature not available yet")
|
|
elif(choice == 3):
|
|
print("Feature to be added soon")
|
|
else:
|
|
print("\nInvalid Choice")
|
|
|
|
|
|
|
|
|