数据库凭据存储新年优惠活动

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

# 生成密钥并创建加密对象
key = Fernet.generate_key()
cipher_suite = Fernet(key)

Base = declarative_base()

class Promotion(Base):
    __tablename__ = 'promotions'
    id = Column(Integer, primary_key=True)
    code = Column(String)
    discount = Column(Integer)
    encrypted_code = Column(String)

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

# 添加新的优惠活动
def add_promotion(code, discount):
    encrypted_code = cipher_suite.encrypt(code.encode())
    new_promotion = Promotion(code=encrypted_code.decode(), discount=discount)
    session.add(new_promotion)
    session.commit()

# 获取优惠活动信息
def get_promotion(promotion_id):
    promotion = session.query(Promotion).filter_by(id=promotion_id).first()
    if promotion:
        decrypted_code = cipher_suite.decrypt(promotion.encrypted_code.encode()).decode()
        return {
            'id': promotion.id,
            'code': decrypted_code,
            'discount': promotion.discount
        }
    return None

# 示例调用
add_promotion('NEWYEAR2023', 20)
print(get_promotion(1))