数据库加密新购活动

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)
    password = Column(String)

    def set_password(self, password):
        key = Fernet.generate_key()
        f = Fernet(key)
        self.password = f.encrypt(password.encode())

    def get_password(self):
        key = Fernet.generate_key()
        f = Fernet(key)
        return f.decrypt(self.password.encode()).decode()

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

# 使用示例
new_user = User(username='john_doe')
new_user.set_password('secret')
session.add(new_user)
session.commit()