您的位置:首页 > 其它

修饰器

2016-02-26 14:15 190 查看
函数修饰器通常是用来给现存的函数增加每次函数被调用时都会运行的一层逻辑。它们可以用来记录函数的日志或者调用次数,检查参数的类型,用户权限等。

如果我想记录每个函数执行的时间,

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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: