copy_dir.py

# /// script
# requires-python = ">=3.11"
# dependencies = []
# ///

import logging
import shutil
import sys
from pathlib import Path


def copy_files(source: Path, destination: Path) -> None:
    if destination.exists():
        shutil.rmtree(destination)
    destination.mkdir()

    for file in source.iterdir():
        if file.is_file():
            shutil.copy(file, destination / file.name)
        else:
            copy_files(file, destination / file.name)


if __name__ == "__main__":
    if len(sys.argv) != 3:  # noqa
        logging.error(
            "Script used incorrectly!\nUsage: python copy_dir.py <source_dir> <destination>"
        )
        sys.exit(1)

    root_dir = Path(__file__).parent.parent.parent
    src = Path(root_dir / sys.argv[1])
    dest = Path(root_dir / sys.argv[2])
    print(f"Copying files from '{sys.argv[1]}' to '{sys.argv[2]}'...")  # noqa: T201

    if not src.exists():
        logging.error("Source directory %s does not exist", src)
        sys.exit(1)

    copy_files(src, dest)