函数修饰器通常是用来给现存的函数增加每次函数被调用时都会运行的一层逻辑。它们可以用来记录函数的日志或者调用次数,检查参数的类型,用户权限等。
如果我想记录每个函数执行的时间,
1,可以在每个函数里面加逻辑,如果函数太多,代码改动太大
# -*- coding:utf-8 -*-
import time
def foo():
start= time.clock()
time.sleep(2.5)
end= time.clock()
print end - start
2,最大限度地少改动
# -*- coding:utf-8 -*-
import time
def timeit(func):
def wrapper():
start= time.clock()
func()
end= time.clock()
print end - start
return wrapper
@timeit
def foo():
time.sleep(2.5)
还有一个问题如果输出foo.__name__
print foo.__name__
wrapper
带参数的修饰器
# -*- coding:utf-8 -*-
import time
import functools
def permission(level):
def decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwds):
start= time.clock()
func(*args, **kwds)
end= time.clock()
print "level=%s"%level, func.__name__, end - start
return wrapper
return decorator
@permission(level=2)
def foo(l):
print l
foo(10)
可以使用functools模块
def timeit(func):
@functools.wraps(func)
def wrapper():
start= time.clock()
func()
end= time.clock()
print end - start
return wrapper
@timeit
def foo():
time.sleep(0.5)
foo()
print foo.__name__
Python还提供其他的内置的装饰器比如staticmethod、classmethod和property