为了保持一个事物的完整性,我们一般都会这样写:
try:sql1sql2sql.commit()
except:sql.rollback()
但是每次都这样try...except...就会显得很重复。
所以要引用装饰器@contextmanager,它需要与with语法结合使用
先看下简单的例子:
from contextlib import contextmanager
class A:def f1(self):print('2')@contextmanager
def get_A():print('1')yield A()print('3')with get_A() as a:a.f1()
#执行结果会依次打印:
1
2
3
上面这个例子,我们可以把print('1') 和 print('3')看成是try和except,要执行代码的部分就是他们中间的部分,进行以下更改
class B:@contextmanagerdef commit(self):try:yieldsql.commit()except:sql.rollback()bb = B()with bb.commit():sql1= "sql1"sql2 = "sql2"
先执行commit(),遇到yield后,跳出来执行sql1,sql2,然后再进入commit()执行上一次跳出来的位置