Python Flask 项目统一数据回滚功能实现

在 Python Flask 项目中，可以使用装饰器来统一添加数据回滚功能。下面是一个示例代码：

from flask import Flask, request, g
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'your_database_uri'
db = SQLAlchemy(app)

@app.before_request
def before_request():
    g.rollback_data = []

@app.teardown_request
def teardown_request(exception=None):
    if exception:
        db.session.rollback()
    else:
        db.session.commit()

@app.route('/add_data', methods=['POST'])
def add_data():
    data = request.json

    try:
        # 在这里添加你需要执行的数据库操作
        # 例如：
        # new_data = MyModel(**data)
        # db.session.add(new_data)
        # db.session.commit()
        # 如果发生异常，会在teardown_request中自动回滚数据

        # 将数据添加到回滚列表中
        g.rollback_data.append(new_data)

        return 'Data added successfully'
    except Exception as e:
        # 发生异常时，将回滚列表中的数据删除
        g.rollback_data.remove(new_data)
        raise e

if __name__ == '__main__':
    app.run()

在上面的示例代码中，before_request装饰器会在每个请求开始前创建一个空的回滚列表，teardown_request装饰器会在请求结束后根据是否有异常来决定是否回滚数据。在添加数据的路由函数中，我们可以将需要回滚的数据添加到回滚列表中，如果发生异常，则在回滚列表中删除该数据。这样可以确保在发生异常时，数据库中的数据会回滚到请求开始前的状态。