修饰器

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


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

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


你可能感兴趣的:(修饰器)