How to Archive a Repository as a ZIP in Python

import zipfile
import io
import os
from pathlib import Path


def archive_repo_mock(repo_path, output_path="repo_archive.zip"):
    """Create a zip archive of a repository directory (mock export)."""
    repo = Path(repo_path)
    if not repo.exists():
        raise FileNotFoundError(f"Repository not found: {repo}")

    with zipfile.ZipFile(output_path, "w", zipfile.ZIP_DEFLATED) as zf:
        for root, dirs, files in os.walk(repo):
            # Skip hidden directories and __pycache__
            root_path = Path(root)
            dirs[:] = [d for d in dirs if not d.startswith(".") and d != "__pycache__"]
            for file in files:
                if file.startswith("."):
                    continue
                full_path = root_path / file
                arcname = full_path.relative_to(repo)
                zf.write(full_path, arcname)

    # Show the archive contents
    with zipfile.ZipFile(output_path, "r") as zf:
        print(f"Archive created: {output_path}")
        print("Contents:")
        for name in sorted(zf.namelist()):
            print(f"  {name}")


if __name__ == "__main__":
    # Create a small mock repository structure
    with tempfile.TemporaryDirectory() as tmpdir:
        tmp_path = Path(tmpdir)
        (tmp_path / "README.md").write_text("# Mock Repo\n")
        (tmp_path / "src").mkdir()
        (tmp_path / "src" / "main.py").write_text("print('hello')\n")
        (tmp_path / ".git").mkdir()
        (tmp_path / ".git" / "config").write_text("[core]\n")

        archive_repo_mock(tmp_path, os.path.join(tmpdir, "archive.zip"))