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和default是同一个库,也必须使用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可以正常回滚(就是因为没有注意到这个问题，导致尝试了好几天都没成功，切记)