数据库加密双十一优惠活动

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from cryptography.fernet import Fernet

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    username = Column(String)
    email = Column(String)

# 生成密钥
key = Fernet.generate_key()
cipher_suite = Fernet(key)

# 加密函数
def encrypt_data(data):
    return cipher_suite.encrypt(data.encode())

# 解密函数
def decrypt_data(encrypted_data):
    return cipher_suite.decrypt(encrypted_data).decode()

# 创建会话
engine = create_engine('sqlite:///example.db')
Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
session = Session()

# 添加用户（加密邮箱）
new_user = User(username='john_doe', email=encrypt_data('john@example.com'))
session.add(new_user)
session.commit()

# 查询用户（解密邮箱）
user = session.query(User).filter_by(username='john_doe').first()
print(f"Username: {user.username}, Email: {decrypt_data(user.email)}")