attach-fixer.py

from pathlib import Path
import filetype
import logging

logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s - %(levelname)s - %(message)s",
    filename=Path.cwd() / "attach_fixer.log",
)


def find_references(key: Path, search_dir: Path):
    for item in search_dir.iterdir():
        if item.is_dir():
            ref = find_references(key, item)
            if ref:
                return ref
        elif item.is_file():
            with open(item, "r", encoding="utf-8", errors="ignore") as file:
                contents = file.read()
                if key.name in contents:
                    return item


def detect_extension(file: Path) -> str:
    return filetype.guess(file).extension


def apply_changes(attachment: Path, new_extension: str, ref: Path):
    attachment.rename(attachment.with_suffix(f".{new_extension}"))
    logging.info(
        f"Renamed: {attachment} to {attachment.with_suffix(f'.{new_extension}')}"
    )
    with open(ref, "r", encoding="utf-8", errors="ignore") as file:
        contents = file.read()
        contents = contents.replace(
            attachment.name, attachment.with_suffix(f".{new_extension}").name
        )
        with open(ref, "w", encoding="utf-8", errors="ignore") as file:
            file.write(contents)
            logging.info(f"Updated reference: {ref}")


def attach_fixer(attach_dir: Path, ref_dir: Path):
    count = 0
    found = 0
    print("File   |   Detected ext   |   Reference")
    for item in attach_dir.iterdir():
        count += 1
        if item.is_dir():
            attach_fixer(item)
        # Skip items with suffix
        elif not item.suffix:
            found += 1

            ext = detect_extension(item)
            ref = find_references(item, ref_dir)

            # apply_changes(item, ext, ref)

            print(f"{item.name}   {ext}   {ref}")

    print(f"Total items reviewed: {count}")
    print(f"Total items found: {found}")