Hello, 极简AES!

from cryptography.fernet import Fernet
import base64
from hashlib import sha256


# 1. 把任意密码转成Fernet合法密钥（核心简化）
def get_key(password: str) -> bytes:
    # 用SHA256哈希+Base64处理，确保32字节
    return base64.urlsafe_b64encode(sha256(password.encode()).digest())


# 2. 加密函数
def encrypt(data: str, password: str) -> str:
    key = get_key(password)
    return Fernet(key).encrypt(data.encode()).decode()


# 3. 解密函数
def decrypt(encrypted_data: str, password: str) -> str:
    key = get_key(password)
    return Fernet(key).decrypt(encrypted_data.encode()).decode()


# 使用示例（复制即用）
if __name__ == "__main__":
    _password = "123"  # 你的密码（任意字符串）
    _data = "Hello, 极简AES!"  # 要加密的内容

    # 加密
    encrypted = encrypt(_data, _password)
    print("加密后:", encrypted)

    # 解密
    decrypted = decrypt(encrypted, _password)
    print("解密后:", decrypted)

加密后: gAAAAABofIi4QPserqlSQifF8f6jthnpyJMZ52stQsWaQDoI3y6DGuW2eNbfYvK89XcNaNfbzihlCB_Zn4NO2PvkdOuZlM2stkKVAVEQuLd6PFm_1lcE_nQ=
解密后: Hello, 极简AES!

gAAAAABofIq3S6H8-h_3CepD70HLxo8SFQzgYQ6MWrOoGvOZd2FGbNdtvTaxO_3t6H6MSnXDaoPayN0_-l69KjBcOEvc-w9f0M9mcw9mei5NnD55kN61TKGalXppNXjDAAwHoKdfx61KaRJT_MRBjx4H8MSTpAW4aCF7EDVZ9ThoKZW4OTFE7ljoA5AWTimnwNUDepOSmDme