python学习笔记之---装饰器

≯℡__Kan透↙ 提交于 2019-12-02 14:58:21
装饰器:也是一种设计模式   decorator
核心作用:给不同的函数(方法用),可以增加不同的公用的功能。
装饰器的标志:带@符号的,如:
@classmethod
@staticmethod
 
闭包:函数对象+函数部分所需要使用的一个外部变量=函数
返回的整体。
 
例子1:
#函数对象可以做为返回值返回
def pri():
    return sum   #函数执行后返回了另外一个函数对象sum
print(type(pri()))

D:\>py -3 a.py

<class 'builtin_function_or_method'>
 
函数对象:就是函数不加()----sum,如果加()就是函数调用----sum()
 
 
 
例子2:
#函数的参数可以使用函数对象
def func1(func):
    return func([1,2,3])
    
func1(sum)

'''
func1(sum) #--->6
    #sum->sum([1,2,3])
    #pri-->pri([1,2,3])--->不符合函数定义,会报错
'''

执行结果:   

D:\>py -3 a.py
<class 'builtin_function_or_method'>
Traceback (most recent call last):
  File "a.py", line 11, in <module>
    func1(pri)
  File "a.py", line 7, in func1
    return func([1,2,3])
TypeError: pri() takes 0 positional arguments but 1 was given
 
函数名---》函数对象
函数名()--->调用函数
 
 
 
 
#闭包:
def func():
    x =100
    def func1():
        print(x)
    return func1
 
 
func()()---等价于
               a=func()
               a()
C:\Users\dell\Desktop\练习\6>py -3 0617.py
100
    #函数名---》返回了函数对象  return func1
    #函数名(参数或无参数)--->调用函数  
    return func1()-->打印100,返回None
 
-----》》》分析过程:
>>> def func():
...     x =100
...     def func1():
...         print(x)
...     return func1
...
>>> func()  #加一个括号是调用了func这个函数,执行完后返回的是func1的函数对象;如果想调用func1()打印x的话就得再加一个(),即func()()
<function func.<locals>.func1 at 0x00000150A43FA620>
>>> func()()
100

 

 
闭包部分:函数+需要的外部变量一起返回了。闭包必须返回一个函数
x =100
    def func1():
        print(x)
        return None
    return func1
func()-->func函数调用后,返回了func1+x
a=func()--->a=func1,把func1+x赋值给a
a()---调用func1()--->打印了100,返回了None
 
#函数对象:就是函数所在的内存地址
 
 
 
 为什么要使用装饰器?
#例:单独的函数,代码都是重复的
def a():
    time1=time.time()
    i =1
    while i<100000000:
        i+=1
    time2=time.time()
    print("time elapsed:%s" %(time2-time1))
 
def b():
    time1=time.time()
    i =1
    while i<990000000:
        i+=1
    time2=time.time()
    print("time elapsed:%s" %(time2-time1))
 
 
#通过函数来尝试解决:
def timer(func):
    time1=time.time()
    func()  #做了一次函数调用 
    time2=time.time()
    print("time elapsed:%s" %(time2-time1))
 
 
def a():
    i =1
    while i<100000000:
        i+=1
 
def b():
    i =1
    while i<990000000:
        i+=1
 
timer(a)
timer(b)

 

 
 
#加装饰器:给函数增加共有的功能,装饰函数一定要有闭包
import time
def timer(func):
    def func1():
        time1=time.time()
        func()  #做一次函数调用
        time2=time.time()
        print("time elapsed:%s" %(time2-time1))
    return func1
    #闭包:func+func1
 
 
@timer
def a():
    i =1
    while i<100000000:
        i+=1
 
 
a()
a  #a等价于timer(a),由python编译器自动完成的。
a()  #a()等价于timer(a)()

 

 
 
#装饰器执行过程解析:
import time
def timer(func):  #装饰函数
    def wrapper():
        time1=time.time()#定义函数调用前干的事儿
        print("******")
        func()  #做一次函数调用
        time2=time.time()#定义函数调用后干的事儿
        print("******-----------")
        print("time elapsed:%s" %(time2-time1))
    return wrapper
    #闭包:func(闭包变量)+wrapper(闭包函数)
 
@timer
def a():
    i =1
    while i<100000000:
        i+=1
    print("a is over!")
 
def b():
    i =1
    while i<100000000:
        i+=1
 
#a  #a等价于timer(a),由python编译器自动完成的。
a()  #a()等价于timer(a)()
 
b()
C:\Users\dell\Desktop\练习\6>py -3 0617.py
******
a is over!
******-----------
time elapsed:5.6724371910095215
 
 
a-->timer(a)-->返回wrapper函数对象
a()-->调用了闭包函数wrapper()
 
wrapper+a:
        time1=time.time()#定义函数调用前干的事儿
        print("******")
        a()  #做一次函数调用
        time2=time.time()#定义函数调用后干的事儿
        print("******-----------")
        print("time elapsed:%s" %(time2-time1))
 
 
#有参装饰器
import time
def timer(func):  #装饰函数
    def wrapper(arg):
        time1=time.time()#定义函数调用前干的事儿
        print("******")
        func(arg)  #做一次函数调用
        time2=time.time()#定义函数调用后干的事儿
        print("******-----------")
        print("time elapsed:%s" %(time2-time1))
    return wrapper
    #闭包:func(闭包变量)+wrapper(函数)
 
@timer
def a(count):
    print("执行次数:",count)
    i =1
    while i<count:
        i+=1
    print("a is over!")
 
a(20)
C:\Users\dell\Desktop\练习\6>py -3 0617.py
******
执行次数: 20
a is over!
******-----------
time elapsed:0.0013477802276611328
 
 
 
#多个参数的装饰器:可变参数
import time
def timer(func):  #装饰函数
    def wrapper(*arg,**kw):
        time1=time.time()#定义函数调用前干的事儿
        print("******")
        func(*arg,**kw)  #做一次函数调用
        time2=time.time()#定义函数调用后干的事儿
        print("******-----------")
        print("time elapsed:%s" %(time2-time1))
    return wrapper
    #闭包:func(闭包变量)+wrapper(函数)
 
@timer
def a(count):
    print("执行次数:",count)
    i =1
    while i<count:
        i+=1
    print("a is over!")
 
@timer
def b(count1,count2,count3):
    print("执行次数:",count1+count2+count3)
 
a(100)
b(200,300,count3=1000)
C:\Users\dell\Desktop\练习\6>py -3 0617.py
******
执行次数: 100
a is over!
******-----------
time elapsed:0.0009958744049072266
******
执行次数: 1500
******-----------
time elapsed:0.0
 
 
 
 
总结:
闭包:执行时需要的外部变量+函数对象
 
装饰器:
定义一个装饰函数,函数必须返回一个闭包函数,
并且func(被装饰的函数)会被python自动传递给
装饰函数,作为装饰函数的一个参数。
 
装饰函数中的闭包函数结构如下:
 
def wrapper(*arg,**kw):
    xxxxx干一些事情
    func(*arg,**kw)  #执行被装饰函数
    yyy   干另外一些事情
    
 
使用装饰器,需要在其他函数的上面写个@xxxxx
 
 
 

装饰器九步法:
第一步:最简单的函数,准备附加额外功能

# -*- coding:utf-8 -*-
'''示例1: 最简单的函数,表示调用了两次'''
def myfunc():
    print ("myfunc() called.")

myfunc()
myfunc()

E:\>py -3 a.py
myfunc() called.
myfunc() called.

 

 

第二步:使用装饰函数在函数执行前和执行后分别附加额外功能

'''示例2: 替换函数(装饰)装饰函数的参数是被装饰的函数对象,返回原函数对象
装饰的实质语句: myfunc = deco(myfunc)'''
def deco(func):
    print ("before myfunc() called.")
    func()
    print (" after myfunc() called.")
    return func
def myfunc():
    print (" myfunc() called.")

myfunc = deco(myfunc)
myfunc()
myfunc()

E:\>py -3 a.py
before myfunc() called.
myfunc() called.
after myfunc() called.
myfunc() called.
myfunc() called.

 

 

第三步:

'''示例3: 使用语法@来装饰函数,相当于“myfunc = deco(myfunc)”
但发现新函数只在第一次被调用,且原函数多调用了一次'''
def deco(func):
    print ("before myfunc() called.")
    func()
    print (" after myfunc() called.")
    return func
@deco  #等价于:deco(myfunc)
def myfunc():
    print (" myfunc() called.")

myfunc()
myfunc()

E:\>py -3 a.py
before myfunc() called.
myfunc() called.
after myfunc() called.
myfunc() called.
myfunc() called.

------->>>>>>

执行逻辑:
deco(myfunc)
print ("before myfunc() called.")
myfunc()
print ("  after myfunc() called.")
print (" myfunc() called.")
print (" myfunc() called.")
 
 

第四步:使用内嵌包装函数来确保每次新函数都被调用

装饰器的规则:
规则1:
函数func上面定义了@xxxx,那么等价于 func = xxxx(func)
规则2:
装饰函数xxxx,必须返回一个闭包(一个内置函数+func)
'''示例4: 使用内嵌包装函数来确保每次新函数都被调用,
内嵌包装函数的形参和返回值与原函数相同,装饰函数返回内嵌包装
函数对象'''
def deco(func):
    def _deco():
        print ("before myfunc() called.")
        func()
        print (" after myfunc() called.")
        # 不需要返回func,实际上应返回原函数的返回值
    return _deco

@deco
def myfunc():
    print (" myfunc() called.")
    return 'ok'

myfunc() #---->执行闭包函数_deco()
myfunc() #---->执行闭包函数_deco()
#myfunc=deco(myfunc)----->返回一个闭包:_deco的函数+myfunc

E:\>py -3 a.py
before myfunc() called.
myfunc() called.
after myfunc() called.
before myfunc() called.
myfunc() called.
after myfunc() called.

分析:
def deco(func):
    def _deco():
        print ("before myfunc() called.")
        func()
        print ("  after myfunc() called.")
        # 不需要返回func,实际上应返回原函数的返回值
    return _deco
闭包:_deco+func
 
@deco   
def myfunc():
    print (" myfunc() called.")
    return 'ok'
 
myfunc=deco(myfunc)
myfunc是什么?是闭包:_deco+myfunc
myfunc()--->_deco()
myfunc()--->_deco()
 
def outer(name):
     def inner():
         print(name)
     return inner
闭包:inner+name
执行过程解释:
装饰函数deco
被装饰函数myfunc
@deco   
def myfunc():   --->myfunc= deco(myfunc)
    myfunc= deco(myfunc)干了什么呢?
                1 调用了deco(myfunc)
                2  返回闭包:_deco+外包变量myfunc
                3 闭包赋值给了myfunc
                4 提醒myfunc变为了闭包函数对象
  
myfunc()---》干了什么呢?
         1 _deco()执行了
        2 print ("before myfunc() called.")
        3 myfunc()
        4 print ("  after myfunc() called.")
 
myfunc()---》干了什么呢?
         1 _deco()执行了
        2 print ("before myfunc() called.")
        3 myfunc()
        4 print ("  after myfunc() called.")

 

 

第五步:对带参数的函数进行装饰

'''示例5: 对带参数的函数进行装饰,
内嵌包装函数的形参和返回值与原函数相同,装饰函数返回内嵌包装
函数对象'''
def deco(func):
    def _deco(a, b):
        print ("before myfunc() called.")
        ret = func(a, b)
        print (" after myfunc() called. result: %s" % ret)
        return ret
    return _deco

@deco
def myfunc(a, b):
    print (" myfunc(%s,%s) called." % (a, b))
    return a + b

myfunc(1, 2)
myfunc(3, 4)

E:\>py -3 a.py
before myfunc() called.
myfunc(1,2) called.
after myfunc() called. result: 3
before myfunc() called.
myfunc(3,4) called.
after myfunc() called. result: 7

  

 

第六步:对参数数量不确定的函数进行装饰

'''示例6: 对参数数量不确定的函数进行装饰,
参数用(*args, **kwargs),自动适应变参和命名参数'''
def deco(func):
    def _deco(*args, **kwargs):
        print ("before %s called." % func.__name__)
        ret = func(*args, **kwargs)
        print (" after %s called. result: %s" % (func.__name__, ret))
        return ret
    return _deco

@deco
def myfunc(a, b):
    print (" myfunc(%s,%s) called." % (a, b))
    return a+b

@deco
def myfunc2(a, b, c):
    print (" myfunc2(%s,%s,%s) called." % (a, b, c))
    return a+b+c

myfunc(1, 2)
myfunc(3, 4)
myfunc2(1, 2, 3)
myfunc2(3, 4, 5)

E:\>py -3 a.py
before myfunc called.
myfunc(1,2) called.
after myfunc called. result: 3
before myfunc called.
myfunc(3,4) called.
after myfunc called. result: 7
before myfunc2 called.
myfunc2(1,2,3) called.
after myfunc2 called. result: 6
before myfunc2 called.
myfunc2(3,4,5) called.
after myfunc2 called. result: 12

 

 

第七步:被装饰函数加参数:带参数的装饰器本质都是两层闭包

'''示例7: 在示例4的基础上,让装饰器带参数,
和上一示例相比在外层多了一层包装。
装饰函数名实际上应更有意义些'''
def deco(arg):
    def _deco(func):
        def __deco():
            print ("before %s called [%s]." % (func.__name__, arg))
            func()
            print (" after %s called [%s]." % (func.__name__, arg))
        return __deco
    return _deco

@deco("mymodule")
def myfunc():
    print (" myfunc() called.")

@deco("module2")
def myfunc2():
    print (" myfunc2() called.")

myfunc()
myfunc2()

'''
1)多了一步:deco("hello") ---》返回了闭包:__deco+s
deco=闭包:__deco+s
2)@deco--->__deco(func)+s--->返回了一个闭包_deco+func+s
后面的过程跟上一步的过程一样。
'''

E:\>py -3 a.py
before myfunc called [mymodule].
myfunc() called.
after myfunc called [mymodule].
before myfunc2 called [module2].
myfunc2() called.
after myfunc2 called [module2].

 

 

 

第八步:让装饰器带 类 参数

'''示例8: 装饰器带类参数'''
class locker:
    def __init__(self):
        print ("locker.__init__() should be notcalled.")

    @staticmethod
    def acquire():
        print ("locker.acquire() called.(这是静态方法)")

    @staticmethod
    def release():
        print (" locker.release() called.(不需要对象实例)")
def deco(cls):
    '''cls 必须实现acquire和release静态方法'''
    def _deco(func):
        def __deco():
            print("before %s called [%s]." %(func.__name__, cls))
            cls.acquire()
            try:
                return func()
            finally:
                cls.release()
        return __deco
    return _deco

@deco(locker)
def myfunc():
    print (" myfunc() called.")

myfunc()
myfunc()

E:\>py -3 a.py
before myfunc called [<class '__main__.locker'>].
locker.acquire() called.(这是静态方法)
myfunc() called.
locker.release() called.(不需要对象实例)
before myfunc called [<class '__main__.locker'>].
locker.acquire() called.(这是静态方法)
myfunc() called.
locker.release() called.(不需要对象实例)

 

第九步:装饰器带类参数,并分拆公共类到其他py文件中,同时演示了对一个函数应用多个装饰器

'''mylocker.py: 公共类 for 示例9.py'''
class mylocker:
    def __init__(self):
        print("mylocker.__init__() called.")
    @staticmethod
    def acquire():
        print("mylocker.acquire() called.")
    @staticmethod
    def unlock():
        print(" mylocker.unlock() called.")

class lockerex(mylocker):
    @staticmethod
    def acquire():
        print("lockerex.acquire() called.")
    @staticmethod
    def unlock():
        print(" lockerex.unlock() called.")


def lockhelper(cls):
    '''cls 必须实现acquire和release静态方法'''
    def _deco(func):
        def __deco(*args, **kwargs):
            print("before %s called." %func.__name__)
            cls.acquire()
            try:
                return func(*args, **kwargs)
            finally:
                cls.unlock()
        return __deco
    return _deco

 

'''示例9: 装饰器带类参数,并分拆公共
类到其他py文件中
同时演示了对一个函数应用多个装饰
器'''
from mylocker import *
class example:
    @lockhelper(mylocker)
    def myfunc(self):
        print (" myfunc() called.")
    @lockhelper(mylocker)
    @lockhelper(lockerex)
    def myfunc2(self, a, b):
        print (" myfunc2() called.")
        return a + b

if __name__=="__main__":
    a = example()
    a.myfunc()
    print(a.myfunc())
    print (a.myfunc2(1, 2))
    print (a.myfunc2(3, 4))

 

 

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!