給Python函數執行前后添加額外行為

以前想在函數前后添點啥額外功能（比如過濾、計時等）時，總是首先想到裝飾器。比如這個計量所花費時間的程序：

from functools import wraps, partial
from time import time

def timing(func=None, frequencies=1):
    if func is None:
        # print("+None")
        return partial(timing, frequencies=frequencies)
    # else:
        # print("-None")

    @wraps(func)
    def _wrapper(*args, **kwargs):
        start_time = time()
        for t in range(frequencies):
            result = func(*args, **kwargs)
        end_time = time()
        print('運行花費時間：{:.6f}s。'.format(end_time-start_time))
        return result

    return _wrapper


@timing
def run():
    l = []
    for i in range(5000000):
        l.extend([i])
    return len(l)

運行如下：

In [4]: run()
運行花費時間：2.383398s。
Out[4]: 5000000

（喜歡刨根問底的可以去掉注釋，并思考預計會有什么樣的輸出）。

今天無意間看到了Python的 上下文管理器（Context Manager） ，發現也非常不錯，其實這跟 with 語句是息息相關的，竟然以前一直未在意。

from time import time

def run2():
    l = []
    for i in range(5000000):
        l.extend([i])
    return len(l)

class ElapsedTime():
    def __enter__(self):
        self.start_time = time()
        return self

    def __exit__(self, exception_type, exception_value, traceback):
        self.end_time = time()
        print('運行花費時間：{:.6f}s。'.format(self.end_time - self.start_time))

with ElapsedTime():
    run2()

初略看了一點官方文檔，上下文管理還是有點都內容的。Python發展到現在，其實不簡單了。說簡單，只是你自己不夠與時俱進，掌握的都是老式三板斧而已。 所以，知識需要不斷更新，才能彌補自己的盲點，這是我最想說的。

來自：http://www.2gua.info/post/63