organizer.py

import os
import shutil
from datetime import datetime

# File categories
FILE_TYPES = {
    "Images": [".jpg", ".jpeg", ".png", ".gif", ".bmp"],
    "Documents": [".pdf", ".doc", ".docx", ".txt"],
    "Spreadsheets": [".xls", ".xlsx", ".csv"],
    "Videos": [".mp4", ".avi", ".mov", ".mkv"],
    "Audio": [".mp3", ".wav", ".aac"],
    "Archives": [".zip", ".rar", ".7z"]
}


def get_unique_filename(destination_folder, filename):
    """
    Prevent overwriting files with same name.
    """

    base, extension = os.path.splitext(filename)

    counter = 1

    new_filename = filename

    while os.path.exists(
        os.path.join(destination_folder, new_filename)
    ):
        new_filename = f"{base}_{counter}{extension}"
        counter += 1

    return new_filename


def organize_folder(folder_path):

    stats = {}

    log_entries = []

    for item in os.listdir(folder_path):

        item_path = os.path.join(folder_path, item)

        if os.path.isdir(item_path):
            continue

        moved = False

        for category, extensions in FILE_TYPES.items():

            if item.lower().endswith(tuple(extensions)):

                category_folder = os.path.join(
                    folder_path,
                    category
                )

                os.makedirs(
                    category_folder,
                    exist_ok=True
                )

                safe_name = get_unique_filename(
                    category_folder,
                    item
                )

                shutil.move(
                    item_path,
                    os.path.join(category_folder, safe_name)
                )

                stats[category] = stats.get(category, 0) + 1

                log_entries.append(
                    f"{item} -> {category}"
                )

                moved = True
                break

        if not moved:

            other_folder = os.path.join(
                folder_path,
                "Others"
            )

            os.makedirs(
                other_folder,
                exist_ok=True
            )

            safe_name = get_unique_filename(
                other_folder,
                item
            )

            shutil.move(
                item_path,
                os.path.join(other_folder, safe_name)
            )

            stats["Others"] = stats.get(
                "Others",
                0
            ) + 1

            log_entries.append(
                f"{item} -> Others"
            )

    create_log(folder_path, stats, log_entries)


def create_log(folder_path, stats, log_entries):

    log_file = os.path.join(
        folder_path,
        "organizer_log.txt"
    )

    with open(log_file, "w") as file:

        file.write(
            "SMART FILE ORGANIZER REPORT\n"
        )

        file.write(
            "=" * 35 + "\n\n"
        )

        file.write(
            f"Generated: {datetime.now()}\n\n"
        )

        file.write("Statistics:\n")

        total = 0

        for category, count in stats.items():

            file.write(
                f"{category}: {count}\n"
            )

            total += count

        file.write(
            f"\nTotal Files: {total}\n\n"
        )

        file.write(
            "Files Moved:\n"
        )

        file.write(
            "-" * 35 + "\n"
        )

        for entry in log_entries:
            file.write(entry + "\n")

    print("\nOrganization Complete!")
    print(f"Total files moved: {total}")
    print(f"Log saved to: {log_file}")


folder = input(
    "Enter folder path to organize: "
)

organize_folder(folder)