關於python django事務transaction源碼分析介紹

這篇文章主要介紹了python django事務transaction源碼分析詳解的相關資料,需要的朋友可以參考下

python Django事務

網路上關於django1.6的事務資料很多，但是1.8的卻搜不到任何資料，自己要用的時候費了不少勁就是不行，現在記下要用的人少走彎路version：Django 1.8 事務官方文檔事務中文文檔裡面介紹很多方法，不一一贅述，按照文檔即可，下面只分析下atomic方法的源碼按照官方文檔transaction.atomic 有兩種用法裝飾器和上下文管理器

# atomic() 方法 
# from django.db import transaction
###################
# atomic()
###################
def atomic(using=None, savepoint=True): # 装饰器和上下文管理器必须.()调用方法，因为真正的处理是该方法返回的实例，不是该方法本身
 if callable(using):
  return Atomic(DEFAULT_DB_ALIAS, savepoint)(using)
 # Decorator: @atomic(...) or context manager: with atomic(...): ...
 else:
  return Atomic(using, savepoint)
##########################################
# Atomic类 省略了非核心内容
############################################
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()   # 进入with创建一个保存点
 # .............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)  # 提交事务
  except DatabaseError:
   try:
    connection.savepoint_rollback(sid) # 捕获数据库异常回滚
    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:    # 把函数放进self的with上下文管理器，效果with相同，只是控制细粒度不同
    return func(*args, **kwargs)
  return inner

python MySQLdb

class Tran():
 def init(self, conn=None, close=True):
  if conn is None:     # 创建数据库链接
   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        # 执行sql
   for s in self.sql:
    self.cur.execute(s)
   self.conn.commit()
  except:            # 可以捕获所有异常(django事务如果中间出现程序异常终止无法回滚)
   try:     # 回滚本身也是sql执行，也有可能失败
    import traceback
    traceback.print_exc()
    print 'rollback'
    self.conn.rollback()
   except:
    print u'回滚失败'
  finally:
   self.cur.close()
   self.conn.close()

更細粒度的回滾:

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

注意：如果有多個資料庫有路由，則需要指定和路由回傳一致的useing ： math2下的model需要事務，即使ziyuan_new和d​​efault是同一個函式庫,也必須使用useing=ziyuan_new

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

# 方法呼叫

###atomic區塊中必須注意try的使用，如果手動捕獲了程式錯誤會導致atomic包裝器捕獲不到異常，也就不會回滾。要嘛try內程式碼不影響事務操作，要嘛就捕獲異常後raise出，讓atomic可以正常回滾(就是因為沒有註意到這個問題，導致嘗試了好幾天都沒成功，切記)######感謝閱讀，希望能幫助大家，謝謝大家對本站的支持！ ###

以上是關於python django事務transaction源碼分析介紹的詳細內容。更多資訊請關注PHP中文網其他相關文章！