python django事務(wù)實例源碼解析

這篇文章主要講解了“python django事務(wù)實例源碼解析”，文中的講解內(nèi)容簡單清晰，易于學(xué)習(xí)與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學(xué)習(xí)“python django事務(wù)實例源碼解析”吧！

python Django事務(wù)

# atomic() 方法   
# from django.db import transaction
###################
# atomic()
###################
def atomic(using=None, savepoint=True):    #  裝飾器和上下文管理器必須.()調(diào)用方法，因為真正的處理是該方法返回的實例，不是該方法本身
    if callable(using):
        return Atomic(DEFAULT_DB_ALIAS, savepoint)(using)
    # Decorator: @atomic(...) or context manager: with atomic(...): ...
    else:
        return Atomic(using, savepoint)
##########################################
#  Atomic類  省略了非核心內(nèi)容
############################################
class Atomic(ContextDecorator):
  def __init__(self, using, savepoint):
    self.using = using
    self.savepoint = savepoint
  def __enter__(self):
    connection = get_connection(self.using)
    sid = connection.savepoint()            # 進(jìn)入with創(chuàng)建一個保存點
    # .............do
  def __exit__(self, exc_type, exc_value, traceback):
    if connection.in_atomic_block:
    # do.............
    if sid is not None:
        try:
            connection.savepoint_commit(sid)        #  提交事務(wù)
        except DatabaseError:
            try:
                connection.savepoint_rollback(sid)    #  捕獲數(shù)據(jù)庫異?；貪L
                connection.savepoint_commit(sid)
            except Error:
                connection.needs_rollback = True
            raise
    ## 還有一段代碼是exec_type收到其他程序異常時候 全局回滾，此處省略
    # do.................
###############################
# ContextDecorator
#################################
class ContextDecorator(object):
    def __call__(self, func):
        def inner(*args, **kwargs):
            with self:              #  把函數(shù)放進(jìn)self的with上下文管理器，效果with相同，只是控制細(xì)粒度不同
                return func(*args, **kwargs)
        return inner

python MySQLdb

class Tran():
    def __init__(self, conn=None, close=True):
        if conn is None:                    # 創(chuàng)建數(shù)據(jù)庫鏈接
            print 'init'
            self.conn = conn_tbkt()
            self.cur = self.conn.cursor()
            self.sql = []

    def __enter__(self):                            # 上下文管理器返回 sql語句列表  with Tran('tbkt_pxb') as sqls:
        print 'enter'
        return self.sql     # sql.append('select 1')

    def __exit__(self, exc_type, exc_val, exc_tb):
        print 'exit'
        try:

            print self.sql                              # 執(zhí)行sql
            for s in self.sql:
                self.cur.execute(s)
            self.conn.commit()
        except:                                             # 可以捕獲所有異常(django事務(wù)如果中間出現(xiàn)程序異常終止無法回滾)
            try:                    # 回滾本身也是sql執(zhí)行，也有可能失敗
                import traceback
                traceback.print_exc()
                print 'rollback'
                self.conn.rollback()
            except:
                print u'回滾失敗'
        finally:
            self.cur.close()
            self.conn.close()

更細(xì)粒度的回滾:

# 在事務(wù)塊中@atomic() 或者 with atomic():
sid = transaction.savepoint('tbkt_pxb')
try:
  # do ..........
except:
  transaction.savepoint_rollback(sid, 'tbkt_pxb')

注意：如果有多個數(shù)據(jù)庫有路由，則需要指定和路由返回一致的useing： math3下的model需要事務(wù)，即使ziyuan_new和default是同一個庫,也必須使用useing=ziyuan_new

        ziyuan_app = ['math3', 'ziyuan']
        if model._meta.app_label in ziyuan_app:
            return "ziyuan_new"

        return 'default'

調(diào)用時候必須.()方法調(diào)用

atomic塊中必須注意try的使用，如果手動捕獲了程序錯誤會導(dǎo)致atomic包裝器捕獲不到異常，也就不會回滾。要么try內(nèi)代碼不影響事務(wù)操作，要么就捕獲異常后raise出，讓atomic可以正?；貪L(就是因為沒有注意到這個問題，導(dǎo)致嘗試了好幾天都沒成功，切記)

感謝各位的閱讀，以上就是“python django事務(wù)實例源碼解析”的內(nèi)容了，經(jīng)過本文的學(xué)習(xí)后，相信大家對python django事務(wù)實例源碼解析這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關(guān)知識點的文章，歡迎關(guān)注！