Create a ZIP Archive of a Folder in Python

import zipfile
from pathlib import Path

def zip_folder(source_dir: str, archive_path: str) -> None:
    """Zip all files in source_dir recursively into archive_path."""
    source = Path(source_dir)
    with zipfile.ZipFile(archive_path, "w", zipfile.ZIP_DEFLATED) as archive:
        for file_path in source.rglob("*"):
            if file_path.is_file():
                # Store files relative to source directory
                archive.write(file_path, file_path.relative_to(source))

if __name__ == "__main__":
    # Create a test directory with some sample files
    test_dir = Path("sample_folder")
    test_dir.mkdir(exist_ok=True)
    (test_dir / "readme.txt").write_text("Hello, world!")
    (test_dir / "data.csv").write_text("name,value\nAlice,42\nBob,17\n")
    sub_dir = test_dir / "subfolder"
    sub_dir.mkdir(exist_ok=True)
    (sub_dir / "config.json").write_text('{"enabled": true}')

    # Zip the folder and print the contents of the archive
    zip_folder("sample_folder", "sample_folder.zip")
    with zipfile.ZipFile("sample_folder.zip") as archive:
        for name in archive.namelist():
            print(name)