main.py

import typer
import os
import pathlib
import csv
from rich.console import Console
from rich.table import Table
from rich.tree import Tree
from rich.text import Text
from rich.filesize import decimal
from rich.markup import escape

app = typer.Typer(rich_markup_mode="rich")
console = Console()


@app.command(
    epilog="Made with :heart:  by [blue][link=https://github.com/Syntoxine]Syntoxine[/link][/blue]"
)
def main(
    directory: str = typer.Argument(
        ".\\",
        help="The directory which will be checked for duplicates.",
        show_default="Current directory",
    ),
    show_tree: bool = typer.Option(False, help="Show a tree of the working directory."),
    export: bool = typer.Option(False, help="Exports duplicates to a .csv file.")
) -> None:
    """Utility script to check for file duplicates."""
    if show_tree:

        def walk_directory(directory: pathlib.Path, tree: Tree) -> None:
            """
            Recursively builds a Tree with directory contents.
            Provided by Will McGugan
            """
            paths = sorted(
                pathlib.Path(directory).iterdir(),
                key=lambda path: (path.is_file(), path.name.lower()),
            )
            for path in paths:
                # Remove hidden files
                if path.name.startswith("."):
                    continue
                if path.is_dir():
                    style = "dim" if path.name.startswith("__") else ""
                    branch = tree.add(
                        f"[bold magenta]:open_file_folder: [link file://{path}]{escape(path.name)}",
                        style=style,
                        guide_style=style,
                    )
                    walk_directory(path, branch)
                else:
                    text_filename = Text(path.name, "green")
                    text_filename.highlight_regex(r"\..*$", "red")
                    text_filename.stylize(f"link file://{path}")
                    file_size = path.stat().st_size
                    text_filename.append(f" ({decimal(file_size)})", "blue")
                    match path.suffix.lower():
                        case ".py": icon = "🐍 "
                        case ".png" | ".jpeg" | ".jpg": icon = "🎨 "
                        case ".mov" | ".mp4": icon = "🎥 "
                        case _: icon = "📄 "
                    tree.add(Text(icon) + text_filename)
        
        fdirectory = os.path.abspath(directory)
        tree = Tree(
            f":open_file_folder: [link file://{fdirectory}]{fdirectory}",
            guide_style="bright_blue",
        )
        walk_directory(pathlib.Path(fdirectory), tree)
        console.print(tree)

    elif export:
        
        duplicates, file_paths = dir_walk(directory)
        if duplicates:
            with open('doubloon-duplicates.csv', 'w', newline='') as file:
                writer = csv.writer(file, dialect = 'excel')
                
                writer.writerows([["Duplicates"],["Name", "Filepath"]])
                for i in duplicates:
                    for j in file_paths[i]:
                        writer.writerow([i, j])
            
            console.print(f"Duplicates successfully exported to [blue]{directory}doubloon-duplicates.csv[/blue].")
            
        else: console.print("[blue]No duplicates[/blue] found!")
    
    else:
        table = Table("Name", "Path", show_lines=True, title="Duplicates")

        duplicates, file_paths = dir_walk(directory)
            
        for i in duplicates:
            for j in file_paths[i]:
                table.add_row(i, j)
            
        if duplicates : console.print(table)
        else: console.print("[blue]No duplicates[/blue] found!")


def dir_walk(directory : str) -> tuple[list,dict]:
    file_paths = {}
    duplicates = []
        
    for root, dirs, files in os.walk(directory):
        files = [f for f in files if not f[0] == '.']
        dirs[:] = [d for d in dirs if not d[0] == '.']
        for i in files:
            file_path = os.path.join(root, i)
            try:
                file_paths[i] = [*file_paths[i],file_path]
                duplicates.append(i)
            except KeyError:
                file_paths[i] = [file_path]
                
    return (duplicates, file_paths)

if __name__ == "__main__":
    app()