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):        3，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")（如果有什么說的不對的地方，歡迎大家多多批評指正。）