SALAlchemy的事务操作-Connection

默认情况,MySQL执行的SQL是autocommit的,SALAlchemy 查询语句也是 autocommit的,就是说如果没有明确声明事务的begin,每个单独的SQL都是一个独立的事务。但是在做交易系统时,比如银行给用户A转账给用户B时,有两个操作,从A里面减100,然后给B加100。这两个操作必须放在一个事务里面才行,否是就会出现钱扣了,对方又没到账的情况。

通过connection.begin 方法可以获取事务对象,执行完sql后,要把事务commit提交,如果出现异常,则把事务rollback,保证数据的最终一致性。

手动开始事务的几种方式:

connection = engine.connect(close_with_result=True)
trans = connection.begin()
try:
    r1 = connection.execute("update account set blance-=100 where id=1")
    r1 = connection.execute("update account set blance+=100 where id=2")
    trans.commit()
except:
    trans.rollback()
    raise

 

当然,事务还有更优雅的写法,使用上下文管理器的特性,用with语句实现

with engine.begin() as connection:
    r1 = connection.execute("update account set blance-=100 where id=1")
    r1 = connection.execute("update account set blance+=100 where id=2")
  • 少了好多行代码,执行完sql会自动提交,如果报异常会自动rollback 。
  • with执行完,连接自动关闭
  • connection关闭后连接自动回收到连接池
上一篇:Python:访问数据库


下一篇:推广自动化的波折