flask-SQLAlchemy数据库模型插入数据的时候使用session.commit()必须处理异常回滚db.session.rollback()-CSDN博客

本文链接：https://blog.csdn.net/weixin_43343144/article/details/87106213

本文介绍了一种使用Python的contextlib.contextmanager来优化数据库操作的方法，通过自定义SQLAlchemy类实现自动提交和错误回滚，避免了繁琐的try/except代码重复。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最为原始的try/except办法，多次插入数据就要写多次，很麻烦，使用python原生的contextlib.contextmanager简化代码！

         try:
            user_db = User(email=self.email, nickname=self.nickname, password=self.password)
            db.session.add(user_db)
            #所有的数据处理准备好之后，执行commit才会提交到数据库！
            db.session.commit()
        except Exception as e:
            #加入数据库commit提交失败，必须回滚！！！
            db.session.rollback()
            raise e

最优化的一种方式，这种方法很牛逼！（对于多个try/except很好的重复性）

from flask_sqlalchemy import SQLAlchemy as BaseSQLAlchemy
from contextlib import contextmanager

#自定义一个SQLAlchemy继承flask_sqlalchemy的,方便自定义方法！！！
class  SQLAlchemy(BaseSQLAlchemy):

    #利用contextmanager管理器,对try/except语句封装，使用的时候必须和with结合！！！
    @contextmanager
    def auto_commit_db(self):
        try:
            yield
            self.session.commit()
        except Exception as e:
           # 加入数据库commit提交失败，必须回滚！！！
           self.session.rollback()
           raise e

db = SQLAlchemy()

利用with上下文管理方法调用！

from kirin_app.db_models.table_user import User
from kirin_app.db_models import db


class RegisterViewModel:

    def __init__(self,form_data):
        self.email = form_data["email"]
        self.nickname = form_data["nickname"]
        self.password = form_data["password"]

        self.__add_db_data()

    # 把数据添加进入数据库
    def __add_db_data(self):
        with db.auto_commit_db():
            user_db = User(email=self.email, nickname=self.nickname, password=self.password)
            db.session.add(user_db)