folder_sort.py

import os


def get_profile_name():
    """Get the username of the current user.
    This function retrieves the username of the current user by extracting it
    from the current working directory path.
    Returns:
        str: The username of the current user.
    Raises:
        Exception: If the current working directory is not within the
        home directory.
    """
    
    pwd = os.getcwd() 
    dir_split = pwd.split("/")
    home = dir_split.index("home")
    try:  
        return dir_split[home+1]
    except:
        print("currently in the home directory")


def get_target_path(profile):
    """Get the target path for a user's Downloads folder.
    Args:
        profile (str): The username or profile name of the user.
    Returns:
        str: The target path for the user's Downloads folder, formatted as
        "/home/{profile}/Downloads".
    """
    return f"/home/{profile}/Downloads"


def change_directory(directory):
    """Change the current working directory to the specified directory.
    Args:
        directory (str): The path of the directory to change to.
    Raises:
        FileNotFoundError: If the specified directory does not exist.
        PermissionError: If the user does not have permission to access 
        the specified directory.
    """
    os.chdir(directory)


def modified_ext_directory2(extension)-> str:
    """Determine the category of a file based on its extension.
    Args:
        extension (str): The file extension to categorize.
    Returns:
        str: The category of the file based on its extension.
    """
    
    categories = {
        "videos": ["mp4", "mkv", "avi", "mov", "wmv", "flv", "webm"],
        "images": ["jpeg", "jpg", "png", "gif", "bmp", "svg", "tiff"],
        "audio": ["mp3", "wav", "flac", "aac", "ogg", "wma", "m4a"],
        "documents": ["pdf", "doc", "docx", "xls", "xlsx", "ppt", "pptx", "txt"],
        "archives": ["zip", "rar", "7z", "tar", "gz"],
        "executables": ["exe", "bat", "sh", "bin"],
        "programming files": ["py", "c", "cpp", "java", "html", "css", "js"],
        "spreadsheets": ["xls", "xlsx", "ods"],
        "presentations": ["ppt", "pptx", "odp"],
        "database": ["sql", "db", "sqlite", "mdb"],
        "fonts": ["ttf", "otf", "woff", "woff2"],
        "vector_graphics": ["svg", "ai", "eps"],
        "3d_models": ["obj", "stl", "fbx", "blend"],
        "data_files": ["csv", "json", "xml", "yaml"],
    }

    for category, extension_raw in categories.items():
        if extension in extension_raw:
            return category
    return "other"
    

def modified_ext_directory1(extensions_list)-> set:
    """Determine the categories of multiple file extensions.
    Args:
        extensions_list (list): A list of file extensions to categorize.
    Returns:
        set: A set containing the categories of the file extensions.
    Raises:
        KeyError: If any extension in the list is not found in any category.
    """
    
    ext_modded = set
    for extension in extensions_list:
        extension_extract = modified_ext_directory2(extension)
        ext_modded.add(extension_extract)
    return ext_modded


def create_directory(extensions):
    """Create directories for each extension if they don't already exist.
    Args:
        extensions (list): A list of directory names to create.
    Raises:
        FileExistsError: If a directory with the same name already exists.
    """
    
    for extension_dir in extensions:
        if os.path.exists(extension_dir):
            print(f"Direcory {extension_dir} already exists" )
            return
        else:
            os.mkdir(f"{extension_dir}")
            print("i shoulda be making a directory", extension_dir)


def files_id() -> list[str]:
    """function to identify all the files in a directory
    return: list of all the available files in the directoroy"""
    dir_list = os.listdir()
    return [file for file in dir_list if os.path.isfile(file)]


def extension_id(file_list) -> list[str]:
    """Extract the file extensions from a list of filenames.
    Args:
        file_list (list): A list of filenames.
    Returns:
        list[str]: A list containing the file extensions extracted
        from the filenames.
    """
    return [os.path.splitext(file)[1][1:] for file in file_list]


def move_files(profile):
    """Move files in the Downloads directory to categorized folders based on
    their extensions.
    This function iterates through the files in the Downloads directory
    associated with the given profile.
    Each file is categorized based on its extension, and then moved to a
    corresponding subdirectory within the Downloads directory.

    Args:
        profile (str): The username or profile name of the user.

    Returns:
        None

    Raises:
        Exception: If an error occurs during the file moving process.
    """
    
    os.chdir(f"/home/{profile}/Downloads")
    files = files_id()
    source_dir = f"/home/{profile}/Downloads"
    for file in files:
        ext = os.path.splitext(file)[1][1:]
        ext_folder = modified_ext_directory2(ext)
        source_path = os.path.join(source_dir, file)
        destination_path = os.path.join(source_dir, ext_folder, file)
        
        try:
            print(os.getcwd())
            os.makedirs(os.path.join(source_dir,ext_folder), exist_ok=True)
            os.rename(source_path, destination_path)
            print(f"Moved {file} to {destination_path}\n")
            
        except Exception as e:
            print(f"Error moving '{file}': {e}")
    print("Operations Complete")
            
if __name__ == "__main__":
    profile  = get_profile_name()
    move_files(profile)