python3 1.7.3 装饰器(decorator)

装饰器(decorator)功能

1.   引入日志

2.    函数执行时间统计

3.    执行函数前预备处理

4.    执行函数后清理功能

5.    权限校验等场景

6.    异常的处理

7.    缓存

装饰器示例

 例1:无参数的函数

from time import ctime, sleep

deftimefun(func):

    defwrappedfunc():

        print("%s called at %s"%(func.__name__, ctime()))

        func()

              return wrappedfunc

@timefun

deffoo():

    print("I am foo")

foo()

sleep(2)

foo()

上面代码理解装饰器执行行为可理解成

foo = timefun(foo)

#foo先作为参数赋值给func后,foo接收指向timefun返回的wrappedfuncfoo()

#调用foo(),即等价调用wrappedfunc()

#内部函数wrappedfunc被引用，所以外部函数的func变量(自由变量)并没有释放

#func里保存的是原foo函数对象

 例2:被装饰的函数有参数

from time import ctime, sleep 

deftimefun(func):    

defwrappedfunc(a, b):

        print("%s called at %s"%(func.__name__, ctime()))

        print(a, b)

        func(a, b)

            return wrappedfunc

@timefun

deffoo(a, b):

    print(a+b)

foo(3,5)

sleep(2)

foo(2,4)

 例3:被装饰的函数有不定长参数

from time import ctime, sleep

deftimefun(func):

    defwrappedfunc(*args, **kwargs):

        print("%s called at %s"%(func.__name__, ctime()))

        func(*args, **kwargs)

             return wrappedfunc

@timefun

deffoo(a, b, c):

    print(a+b+c)

foo(3,5,7)

sleep(2)

foo(2,4,9)

 例4:装饰器中的return

from time import ctime, sleep 

deftimefun(func):

    defwrappedfunc():

        print("%s called at %s"%(func.__name__, ctime()))

        func()

             return wrappedfunc

@timefun

deffoo():

    print("I am foo") 

@timefun

defgetInfo():    

    return'----hahah---' 

foo()

sleep(2)

foo() 

print(getInfo())

执行结果:

foo called at Fri Nov  4 21:55:35 2016

I am foo

foo called at Fri Nov  4 21:55:37  2016

I am foo

getInfo called at Fri Nov  4 21:55:37 2016

None

如果修改装饰器为return func()，则运行结果：

foo called at Fri Nov  4 21:55:57 2016

I am foo

foo called at Fri Nov  4 21:55:59 2016

I am foo

getInfo called at Fri Nov  4 21:55:59 2016

----hahah---

 

总结：

·        一般情况下为了让装饰器更通用，可以有return

1.1.1.5       例5:装饰器带参数,在原有装饰器的基础上，设置外部变量

#decorator2.py

from time import ctime, sleep 

def timefun_arg(pre="hello"):

        def timefun(func):

                def wrappedfunc():

            print("%s called at %s %s"%(func.__name__, ctime(), pre))

                          return func()

                return wrappedfunc

        return timefun

@timefun_arg("wangcai")

deffoo():

    print("I am foo") 

@timefun_arg("python")

deftoo():

    print("I am too") 

foo()

sleep(2)

foo()

too()

sleep(2)

too()

可以理解为

foo()==timefun_arg("wangcai")(foo)()

 例6：类装饰器（扩展，非重点）

装饰器函数其实是这样一个接口约束，它必须接受一个callable对象作为参数，然后返回一个callable对象。在Python中一般callable对象都是函数，但也有例外。只要某个对象重写了 __call__() 方法，那么这个对象就是callable的。

classTest():

    def__call__(self):

        print('call me!')

t = Test()

t()  # call me

类装饰器demo

class Test(object):

        def__init__(self, func):

        print("---初始化---")

        print("func name is %s"%func.__name__)

        self.__func = func

        def__call__(self):

        print("---装饰器中的功能---")

        self.__func()

#说明：

#1. 当用Test来装作装饰器对test函数进行装饰的时候，首先会创建Test的实例对象

#    并且会把test这个函数名当做参数传递到__init__方法中

#    即在__init__方法中的func变量指向了test函数体

#2. test函数相当于指向了用Test创建出来的实例对象

#3. 当在使用test()进行调用时，就相当于让这个对象()，因此会调用这个对象的__call__方法

#4. 为了能够在__call__方法中调用原来test指向的函数体，所以在__init__方法中就需要一个实例属性来保存这个函数体的引用

#    所以才有了self.__func = func这句代码，从而在调用__call__方法中能够调用到test之前的函数体

@Test

def test():

    print("----test---")

test()

showpy()#如果把这句话注释，重新运行程序，依然会看到"--初始化--"

运行结果如下：

---初始化---

func name is test

---装饰器中的功能---

----test---