Encrypt and Decrypt Files Using Python

import os
from pathlib import Path
from cryptography.fernet import Fernet

def generate_key(key_file: Path) -> bytes:
    key = Fernet.generate_key()
    key_file.write_bytes(key)
    return key

def load_key(key_file: Path) -> bytes:
    return key_file.read_bytes()

def encrypt_file(input_path: Path, key: bytes, output_path: Path = None) -> Path:
    fernet = Fernet(key)
    encrypted_data = fernet.encrypt(input_path.read_bytes())
    out = output_path or input_path.with_suffix(input_path.suffix + '.enc')
    out.write_bytes(encrypted_data)
    return out

def decrypt_file(encrypted_path: Path, key: bytes, output_path: Path = None) -> Path:
    fernet = Fernet(key)
    decrypted_data = fernet.decrypt(encrypted_path.read_bytes())
    out = output_path or encrypted_path.with_suffix('.dec')
    out.write_bytes(decrypted_data)
    return out

if __name__ == "__main__":
    key_file = Path("secret.key")
    original = Path("example.txt")
    original.write_text("Hello, secure world!")

    key = generate_key(key_file)
    enc_path = encrypt_file(original, key)
    print(f"Encrypted: {enc_path}")

    dec_path = decrypt_file(enc_path, key)
    print(f"Decrypted: {dec_path}")
    print(f"Content: {dec_path.read_text()}")

    os.remove(key_file)
    os.remove(original)
    os.remove(enc_path)
    os.remove(dec_path)