最為原始的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)