给Python函数执行前后添加额外行为

给Python函数执行前后添加额外行为 2016 10 19 17:46:04 以前想在函数前后添点啥额外功能(比如过滤、计时等)时,总是首先

给Python函数执行前后添加额外行为

2016.10.19 17:46:04

给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发展到现在,其实不简单了。说简单,只是你自己不够与时俱进,掌握的都是老式三板斧而已。 所以,知识需要不断更新,才能弥补自己的盲点,这是我最想说的。

后续再找时间好好探究一番。

未登录用户
全部评论0
到底啦