Compress and Extract ZIP Files Programmatically in Python

import zipfile
from pathlib import Path
import tempfile
import os

def create_sample_zip(zip_path: str, files: dict) -> None:
    """Create a ZIP file containing the given files (name -> content mapping)."""
    with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zf:
        for filename, content in files.items():
            zf.writestr(filename, content)

def extract_zip(zip_path: str, extract_dir: str) -> list:
    """Extract all files from a ZIP archive and return list of extracted paths."""
    with zipfile.ZipFile(zip_path, 'r') as zf:
        zf.extractall(extract_dir)
    return [str(p) for p in Path(extract_dir).rglob('*') if p.is_file()]

if __name__ == "__main__":
    # Create a temporary directory for demonstration
    with tempfile.TemporaryDirectory() as tmpdir:
        zip_path = os.path.join(tmpdir, "sample.zip")
        
        # Create sample ZIP with two text files
        files_to_zip = {
            "hello.txt": "Hello, world!",
            "data/notes.txt": "This is a nested file."
        }
        create_sample_zip(zip_path, files_to_zip)
        
        # Extract to a subdirectory
        extract_path = os.path.join(tmpdir, "extracted")
        extracted_files = extract_zip(zip_path, extract_path)
        
        # Verify contents
        for file_path in sorted(extracted_files):
            with open(file_path, 'r') as f:
                print(f"{os.path.relpath(file_path, extract_path)}: {f.read()}")