数据库加密双11活动

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)

    def __init__(self, username, email):
        self.username = username
        self.email = Fernet(key).encrypt(email.encode())

# 初始化数据库连接
engine = create_engine('sqlite:///example.db')
Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
session = Session()

# 添加新用户
new_user = User(username='john_doe', email='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: {Fernet(key).decrypt(user.email).decode()}")