python装饰器

python装饰器

什么是装饰器？

装饰器本质是一个python函数，可以让其他函数在不需要做任何代码啊变动的前提下增加额外功能，装饰器的返回值也是一个函数对象。它经常用于有切面需求的场景，比如：插入日志、性能检测、事物处理、缓存、权限校验等场景。

1、非装饰器时的打印函数打印日志：

import time

def deco(func):
    print "starttime: %s" % time.time()
    func()
    print "endtime:%s" % time.time()

def myfunc():
    print "start myfunc"
    print "Test myfunc"
    print "end my func"

def main():
    deco(myfunc)

运行后的结果：

starttime: 1502033836.49
start myfunc
Test myfunc
end my func
endtime:1502033836.49

可以看到，直接传入函数参数，可以实现计时（打印）的能力，但是在调用的时候，需要使用deco调用，这样造成的后果是，每次在想使用myfunc的时候，都需要使用deco调用，并且，在已实现的逻辑中，需要改动调用方式。

 

2、使用装饰器打印：

def deco1(func):
    def wrapper():
        print "starttime: %s" % time.time()
        func()
        print "endtime:%s" % time.time()
    return wrapper

def myfunc():
    print "start myfunc"
    print "Test myfunc"
    print "end my func"

def main():
    #deco(myfunc)
    myfunc1 = deco1(myfunc)
    myfunc1()
    pass
if __name__ == "__main__":
    main()

可以看到，在这种情况下可以用myfunc1 = deco(myfunc)方式调用，这是上面那种传入函数名称所不能达到的。但是存在的问题是，函数名称会冲突，不得不使用myfunc1方式。

 

3、使用@语法糖的装饰器：

def deco1(func):
    def wrapper():
        print "starttime: %s" % time.time()
        func()
        print "endtime:%s" % time.time()
    return wrapper

@deco1
def myfunc():
    print "start myfunc"
    print "Test myfunc"
    print "end my func"

def main():
    myfunc()

则使用了@语法糖之后，可以直接使用myfunc进行调用。并且，很好的实现了时间统计、日志打印，这是最简单的装饰器场景了。

 

4、带固定参数的装饰器：

def deco2(func):
    def wrapper(a, b, c):
        print "starttime: %s" % time.time()
        func(a, b, c)
        print "endtime:%s" % time.time()
    return wrapper

@deco2
def myfunc1(a, b, c=100):
    print "start myfunc1(a, b )"
    print "a is %s" % a
    print "b is %s" % b
    print "c is %s" % c
    print "end my func1(a, b )"

def main():
    myfunc1(1, 2, c=3)

执行结果：

starttime: 1502118568.76
start myfunc1(a, b )
a is 1
b is 2
c is 3
end my func1(a, b )
endtime:1502118568.76

5、对参数数量不确定的函数进行装饰：

def deco3(func):
    def wrapper(*args, **kargs):
        print "starttime: %s" % time.time()
        func(*args, **kargs)
        print "endtime:%s" % time.time()
    return wrapper

@deco3
def myfunc1(a, b, c=100):
    print "start myfunc1(a, b )"
    print "a is %s" % a
    print "b is %s" % b
    print "c is %s" % c
    print "end my func1(a, b )"

def main():
    myfunc1(1, 2, c=3)

针对不同参数的情况，可以看到结果正常执行。

 

6、装饰器支持函数入参：

def deco4(*args0, **kargs0):
    def _deco4(func):
        def wrapper(*args, **kargs):
            print args0[0]
            print kargs0["b"]
            print "starttime: %s" % time.time()
            func(*args, **kargs)
            print "endtime:%s" % time.time()
        return wrapper
    return _deco4

@deco4("x", b="y")
def myfunc1(a, b, c=100):
    print "start myfunc1(a, b )"
    print "a is %s" % a
    print "b is %s" % b
    print "c is %s" % c
    print "end my func1(a, b )"

def main():
    myfunc1(1, 2, c=3)

执行后可以看到，函数正常的执行了：

x
y
starttime: 1502120041.85
start myfunc1(a, b )
a is 1
b is 2
c is 3
end my func1(a, b )
endtime:1502120041.85

 

7、装饰器入参类：

class TestDeco():
    def __init__(self):
        print "init TestDeco"

    def info(self):
        print "Now begin to Test Deco"
    info = classmethod(info)

def deco5(cls):
    def _deco4(func):
        def wrapper(*args, **kargs):
            print "starttime: %s" % time.time()
            print cls.info()
            func(*args, **kargs)
            print "endtime:%s" % time.time()
        return wrapper
    return _deco4

@deco5(TestDeco)
def myfunc1(a, b, c=100):
    print "start myfunc1(a, b )"
    print "a is %s" % a
    print "b is %s" % b
    print "c is %s" % c
    print "end my func1(a, b )"

def main():
    myfunc1(1, 2, c=3)

执行后发现，装饰器正常的入参了类，并且打印了正常的信息：

starttime: 1502120586.49
Now begin to Test Deco
None
start myfunc1(a, b )
a is 1
b is 2
c is 3
end my func1(a, b )
endtime:1502120586.49