@contextmanager 和yield 的结合使用法

为了保持一个事物的完整性,我们一般都会这样写:

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()执行上一次跳出来的位置