Python 装饰器

# （1）装饰器定义
def log(func):
    def wrapper(*arg, **kwargs):
        print "函数%s被调用了:" % func.__name__
        return func(*arg, **kwargs)
    return wrapper

@log                    # 等价于：test = log(test)
def test():
    print 100

test()
print test.__name__      # 此时test等价log的内函数wrapper

#------------输出----------
# 函数test被调用了:
# 100
# wrapper
#-------------------------

# （2）希望test.__name__获取它自身的名字
import functools
def log2(func):
    @functools.wraps(func)
    def wrapper(*arg, **kwargs):
        print "函数%s被调用了:" % func.__name__
        return func(*arg, **kwargs)
    return wrapper

@log                    # 等价于：test = log(test)
def test():
    print 100

test()
print test.__name__

#------------输出----------
# 函数test被调用了:
# 100
# test
#-------------------------


# （3）定义带参数的装饰器函数
import functools
def log3(param1, param2):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*arg, **kwargs):
            print param1, param2
            print "函数%s被调用了:" % func.__name__
            return func(*arg, **kwargs)
        return wrapper
    return decorator

@log3(1, 2)                   # 等价于：test = log(test)
def test():
    print 100

test()

#------------输出----------
# 1 2
# 函数test被调用了:
# 100
#-------------------------