一、 装饰器

  装饰器就是闭包函数的一种应用场景

  什么是闭包函数?我们再来回忆一下:

  闭包函数:

    定义在函数内部的函数,并且该函数包含对外部函数作用域(强调:对全局作用域名字的引用不算闭包)名字的引用,该函数称为闭包函数

  说到作用域我们再回忆下:

 

  作用域:
      全局范围:内置+全局
        全局有效,全局存活
      局部范围:局部
        局部有效,局部存活
  作用域关系是在函数定义阶段就规定死,与调用位置无关,
  也就是说,无论函数在哪儿调用,都必须回到当初定义函数时的位置找作用域关系

 

一 、为何要用装饰器

#开放封闭原则:
    #软件一旦上线后,就应该遵循开放封闭原则,对修改源代码是封闭的,对功能的扩展是开放的,也就是我们必须找到一种解决方法:
    #能够在不修改一个功能源代码以及调用方式的前提下,为其加上新功能.

二 、什么是装饰器

#装饰器他人的器具,本身可以是任意可调用对象,被装饰者也可以是任意可调用对象。
#强调装饰器的原则:
    #1 、不修改被装饰对象的源代码 
    #2 、不修改被装饰对象的调用方式
#装饰器的目标:
    #在遵循1和2的前提下,为被装饰对象添加上新功能

三 、装饰器的使用

#我们现在给下面函数增加一个运行时间的功能
import time

def index():
    time.sleep(3)
    print('welcome to index page')

#修改一
def index():
    start_time=time.time()
    time.sleep(3)
    print('welcome to index page')
    stop_time=time.time()
    print('run time is %s' %(stop_time-start_time))

index()   #功能实现

#评语:直接改源代码,这么搞被开了。。。
#再来一位童靴,来实现功能
#修改二
import time

def index():
    time.sleep(1)
    print('welcome to index page')

start_time=time.time()
index()
stop_time = time.time()
print('run time is %s' % (stop_time - start_time))

#评语:有好多函数要实现这个功能,写N遍变这个代码,
#后期维护一脸懵逼,还是被开。。。
#再来一位童靴,功能重复实现看我用强大的函数
#修订三:
import time

def index():
    time.sleep(3)
    print('welcome to index page')

def wrapper(func):  #func=index
    start_time=time.time()
    func() #index()
    stop_time = time.time()
    print('run time is %s' % (stop_time - start_time))

wrapper(index) 

#评语:修改了原函数的调用方式,依然被开。。。
#终于来了位小牛的童靴,函数的值可以返回,然后我再把重新定义index
#修订四:
import time

def index():
    time.sleep(3)
    print('welcome to index page')

def outter(func): #func=最原始的index
    # func=最原始的index
    def wrapper():
        start_time=time.time()
        func()
        stop_time=time.time()
        print(stop_time-start_time)
    return wrapper

index=outter(index)  # 新的index=wrapper
index()  #wrapper()  功能基本实现
#评语:在原值没有返回值是没问题,但是有返回值的情况下,这么搞就会发现返回的是None
#这次小牛牛童靴路过看到这情况,say这么搞!
#修订五
import time
def index():
    time.sleep(1)
    print('welcome to index page')
    return 123     #假使这里返回123     返回值可以是任意类型 

#==============下面就是装饰器
def timmer(func):
    #func=最原始的index
    def wrapper(*args,**kwargs):     #可变长参数
        start_time=time.time()
        res=func(*args,**kwargs)  #调用最原始的index
        stop_time=time.time()
        print(stop_time-start_time)
        return res        #index()运行的返回值
    return wrapper

index=timmer(index) # 新的index=wrapper

print(index())    #功能已经实现,返回值123

#评语:这里的装饰的功能已经实现,返回的值也得到,小牛牛不是白叫的
#天上五彩红光,大牛童靴出现!各位童靴火速围观!
import time
def timmer(func):
    def wrapper(*args,**kwargs):
        start_time=time.time()
        res=func(*args,**kwargs)
        stop_time=time.time()
        print(stop_time-start_time)
        return res
    return wrapper

@timmer #index=timmer(index)      装饰器的标准格式!
def index():
    time.sleep(1)
    print('welcome to index page')
    return 123

@timmer # home=timmer(home)
def home(name):
    time.sleep(2)
    print('welcome %s to home page' %name)

# index()
#home('egon')   

#评语:大牛就是大牛!大牛say:教你们绝技,不会的童靴可以照下面的模板实现:

#无参装饰器模板
def outer(func):     #outer,inner名字功能随意
    def inner(*args,**kwargs):
        res=func(*args,**kwargs)
        return res
    return inner

@outer            #装饰器要在装饰函数的上方
def duoduo():
    pass                      

四 、装饰器语法

#被装饰函数的正上方,单独一行
@deco1
@deco2
@deco3
def foo():
    pass
#foo=deco1(deco2(deco3(foo)))
#这里的思想就是最上面装饰器,装饰的下面所有的函数(deco2,deco3,foo)
#然后deco2装饰(deco3,foo),最后deco3装饰foo
#功能的不同,放的顺序也要注意,不然装饰的效果可能实现的就不对了!

五、多个装饰器的使用:

import time
current_user={
    'username':None,
    # 'login_time':None
}

def auth(func):
    # func=index
    def wrapper(*args,**kwargs):
        if current_user['username']:   #这里是认证过的,下次就不用认证
            print('已经登陆过了')
            res=func(*args,**kwargs)
            return res

        uname=input('用户名>>: ').strip()
        pwd=input('密码>>: ').strip()
        if uname == 'egon' and pwd == '123':
            print('登陆成功')
            current_user['username']=uname
            res=func(*args,**kwargs)
            return res
        else:
            print('用户名或密码错误')
    return wrapper

def timmer(func):
    def wrapper(*args,**kwargs):
        start_time=time.time()
        res=func(*args,**kwargs)
        stop_time=time.time()
        print(stop_time-start_time)
        return res
    return wrapper

@timmer # timmer 统计的是auth+index的执行时间
@auth             #我们只装饰index的话就要把@timmer紧跟index
def index():
    time.sleep(1)
    print('welcome to index page')
    return 123

@auth
@timmer     #这里统计的就是home运行的时间
def home(name):
    time.sleep(2)
    print('welcome %s to home page' %name)

#index()
#home("duoduo")

六、有参数的装饰器的使用:

import time
current_user={
    'username':None,
    # 'login_time':None
}

def auth(engine):    #道理还是那个道理,在外面包了一层engine的值
    # engine='file'  #这个值外面穿什么进来就是什么
    def auth2(func):
        # func=index
        def wrapper(*args,**kwargs):
            if engine == 'file':
                if current_user['username']:
                    print('已经登陆过了')
                    res=func(*args,**kwargs)
                    return res

                uname=input('用户名>>: ').strip()
                pwd=input('密码>>: ').strip()
                if uname == 'egon' and pwd == '123':
                    print('登陆成功')
                    current_user['username']=uname
                    res=func(*args,**kwargs)
                    return res
                else:
                    print('用户名或密码错误')
            elif engine == 'mysql':         #engine 值得判断情况
                print('基于MyQL的认证')
            elif engine == 'ldap':
                print('基于LDAP的认证')
        return wrapper
    return auth2     #这里也要返回auth2的内存地址

@auth('ldap') #@auth2 #index=auth2(index) #index=wrapper
def index():
    time.sleep(1)
    print('welcome to index page')
    return 123


index() # wrapper()

 

版权声明:本文为ManyQian原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://www.cnblogs.com/ManyQian/p/8670190.html