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

最为原始的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结合！！！@contextmanagerdef auto_commit_db(self):try:yieldself.session.commit()except Exception as e:# 加入数据库commit提交失败，必须回滚！！！self.session.rollback()raise edb = SQLAlchemy()

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

from kirin_app.db_models.table_user import User
from kirin_app.db_models import dbclass 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)

在这里插入图片描述

————————————————
版权声明：本文为CSDN博主「比特币爱好者007」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_43343144/article/details/87106213