How to Create a Password Protected Zip Archive in Python

import zipfile
import tempfile
import os


def create_password_protected_zip(zip_path, password: str, files: dict):
    """
    Create a zip archive with password protection (mock encryption).

    Args:
        zip_path: Path where the zip file will be created
        password: Password for the archive
        files: Dictionary mapping filenames to their content
    """
    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
        zf.setpassword(password.encode())
        for name, content in files.items():
            zf.writestr(name, content)


def try_extract(zip_path, password: str) -> bool:
    """Attempt to extract all files from a zip with a given password."""
    try:
        with zipfile.ZipFile(zip_path) as zf:
            zf.setpassword(password.encode())
            zf.extractall(path=tempfile.mkdtemp())
        return True
    except (RuntimeError, zipfile.BadZipFile):
        return False


if __name__ == "__main__":
    tmp_zip = "mock_secure.zip"
    files = {"secret.txt": "This is hidden content.", "notes.txt": "Another secret."}
    
    create_password_protected_zip(tmp_zip, "correct-horse", files)
    
    # Demonstrate wrong password fails
    wrong_attempt = try_extract(tmp_zip, "wrong-password")
    print(f"Wrong password attempt: {wrong_attempt}")
    
    # Demonstrate correct password succeeds
    correct_attempt = try_extract(tmp_zip, "correct-horse")
    print(f"Correct password attempt: {correct_attempt}")
    
    # Clean up
    os.remove(tmp_zip)