file_management.py

import os
import shutil
class Organize:
    def __init__(self, main_path, file_folders=[[]]):
        self.main_path = main_path
        self.main_path_list = os.listdir(main_path)
        self.file_folders = file_folders

    def create_directories(self):
        #need to add input for paths but that can come later
        add_new_folder = True
        while(add_new_folder):
            add_new_folder_input = input("Would you like to add a new folder(Y/N): ").lower()
            if add_new_folder_input == "y":
                name_of_folder = input("What is the name of the folder: ")
                corresponding_ext = input("what extension do you want to organize in this folder: ")
                self.file_folders.append([name_of_folder,corresponding_ext])
            else:
                add_new_folder = False 
            
    def move_folders(self):
        for i in range(len(self.file_folders)):
            path = os.path.join(self.main_path, self.file_folders[i][0])
            #make directory
            if not os.path.exists(path):
                os.mkdir(path)
            else:
                print("folder already exists")
    
    def organize_files(self):
       #loop through the number of folders the user created
       for i in range(len(self.file_folders)):
           #save path string in a variable
            path = os.path.join(self.main_path, self.file_folders[i][0])
            for j in range(len(self.main_path_list)):
                #check file extention of the paths loose files, with the file the extention the user chose
                if self.main_path_list[j].endswith(self.file_folders[i][1]):
                    #move all loose files to the folder
                    shutil.move(self.main_path + "/" + self.main_path_list[j], path)
            




initialize = Organize("/Users/user/Desktop", [])

initialize.create_directories()
initialize.move_folders()
initialize.organize_files()