数据库加密双十二促销活动

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()

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

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

    def set_encrypted_email(self, email):
        self.email = cipher_suite.encrypt(email.encode())

    def get_decrypted_email(self):
        return cipher_suite.decrypt(self.email).decode()

# 创建会话
Session = sessionmaker(bind=engine)
session = Session()

# 添加用户
new_user = User(username='john_doe')
new_user.set_encrypted_email('john@example.com')
session.add(new_user)
session.commit()

# 查询并解密用户邮箱
user = session.query(User).filter_by(username='john_doe').first()
print(user.get_decrypted_email())