python装饰器、with和contextmanager

2019-11-08 01:00:56

字体：大中小

来源：转载

供稿：网友

1，装饰器：

装饰器是一种简单的记号而已，有如下等价的转换规则（带不带参数的装饰器都一样，只要是一个可调用对象即可）。

@foo

def bar:

pass

等价于：

x = foo

bar = x(bar)

'----------------'

@foo('abc')

def bar:

pass

等价于：

x = foo('abc')

bar = x(bar)

'----------------'

@foo.X('abc').Y

def bar:

pass

等价于：

x = foo.X('abc').Y

bar = x(bar)

真正起作用的是"函数式编程"方面的知识，包括闭包，LEGB访问规则，自由变量等。

装饰器的用处：返回一个增强版的函数，主要是对函数的上文和下文，都可以进行处理。

比如：参数检查，缓存数据（建立输入与输出的字典），代理，上下文提供者

from threading import RLocklock = RLock()def locker(function):    def _locker(*args,**kw):        lock.acquire()        try:            return function(*args,**kw)        finally:            lock.release()    return _locker@lockerdef thread_safe():    pass以上装饰器的定义都没有问题，但还差最后一步！！
因为我们讲了函数也是对象，它有__name__等属性，但经过decorator装饰之后的函数，
它们的__name__已经从原来的'thread_safe'变成了'_locker'。所以不完美。
不需要直接编写wrapper.__name__ = func.__name__这样的代码，Python内置的functools.wraps就是干这个事的。
import functoolsdef locker(function):    @functools.wraps(function)    def _locker(*args,**kw):       lock.acquire()       try:           return function(*args,**kw)       finally:           lock.release()    return _locker
2，with语句：
任何实现了 __enter__和__exit__的类都可以和 with使用。class Context(object):    def __enter__(self):        PRint 'entering the zone'    def __exit__(self, exception_type, exception_value,                 exception_traceback):        print 'leaving the zone'        if exception_type is None:            print 'with no error'        else:            print 'with an error (%s)' % exception_value        '异常是会被新的异常覆盖的        raise TypeError('i am the second bug')  try:    with Context():        print 'i am the buggy zone'        raise TypeError('i am the bug')except Exception,s:    print 'I got :',Exception,s3，contextmanager装饰器：
一种和yield结合使用的方便的上下文管理方式，切记yield是协程技术。
Typical usage:     # @contextmanager     def some_generator(<arguments>):         <setup>         try:             yield <value>         finally:             <cleanup> This makes this:     with some_generator(<arguments>) as <variable>:         <body> equivalent to this:     <setup>     try:         <variable> = <value>         <body>     finally:         <cleanup>'---------自己实现contextmanager装饰器---------------'class MyGeneratorContextManager(object):    def __init__(self, gen):        print("__init__ called")        self.gen = gen    def __enter__(self):        print("__enter__ called")        return self.gen.next()    def __exit__(self, exc_type, exc_val, exc_tb):        print("__exit__called exc_type = %s, exc_val = %s, exc_tb = %s"/              % (exc_type, exc_val, exc_tb))        try:            ' 即便不调用next也是没有关系的，只是不会做清理的处理，不会执行print("end foo")，但是finally还是程序结束前总会被执行的'            return self.gen.next()         except StopIteration:             '是不会捕获with中的异常，但是这里抛出的异常可以覆盖with中抛出的异常！'            print 'nothing to do with StopIteration'        def MyContextManager(func):    def _deco(*args,**kwargs):        print("func info:", func)        return MyGeneratorContextManager(func(*args,**kwargs))    return _deco@MyContextManagerdef foo(*args,**kwargs):    try: # 尝试用老方法捕捉错误        print("start foo:", args,kwargs)        yield [1, 2, 3]        print("end foo") # 需要调用self.gen.next()才能输出    except (Exception, AssertionError):        print("EXCEPTION ENCOUNTERED!")    finally:        print("FINALLY")with foo("oh no!") as tmp: # tmp的值就是__enter__返回的值    print("START WITH")    print 'tmp:',tmp    assert 1>2    # 出错之后直接从with中跳出去，下面不可能被执行    print("END WITH")（如果有什么说的不对的地方，欢迎大家多多批评指正。）