python decorator 修饰器

decorator 就是给函数加一层皮，好用！

 

from time import ctime

def deco(func):
    def wrappedFunc(*args, **kwargs):
        print '[%s] %s called' % (ctime(), func.__name__)
        ret = func(*args, **kwargs)
        print '[%s] %s call end' % (ctime(), func.__name__)
        return ret

    return wrappedFunc

@deco
def foo(a=0,b=1):
    print "foo's here!"
    return a,b


def main():
    print foo(3,2)


if __name__ == '__main__':
    main()

在 12 行加上 @deco 相当于定义好 foo 函数之后执行一下 foo = deco(foo)

@deco
def foo(a=0,b=1):
    print "foo's here!"
    return a,b

等价于

def foo(a=0,b=1):
    print "foo's here!"
    return a,b

foo = deco(foo)

  

输出：

[Sun Nov 22 22:14:19 2015] foo called
foo's here!
[Sun Nov 22 22:14:19 2015] foo call end
(3, 2)

 

带参数的修饰器

def deco_with_params(a, b, c):
    def real_deco(func):
        def wrappedFunc(*args, **kwargs):
            ret = func(*args, **kwargs)
            return ret
        return wrappedFunc
    return real_deco

 

 

开心玩耍吧！