为了保持一个事物的完整性,我们一般都会这样写:
try:
sql1
sql2
sql.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:
@contextmanager
def commit(self):
try:
yield
sql.commit()
except:
sql.rollback()
bb = B()
with bb.commit():
sql1= "sql1"
sql2 = "sql2"
先执行commit(),遇到yield后,跳出来执行sql1,sql2,然后再进入commit()执行上一次跳出来的位置