Py修行路 python基础 （十）装饰器

装饰器

一、定义

    装饰器：顾名思义，就是对某个东西起到装饰修饰的功能。

python中的装饰器，其本质上就是一个python函数，它可以让其他函数在不需要任何代码变动的前提下增加额外功能。通俗理解就是 函数 闭包 的实用。

二、语法及注意事项

　　1、书写规范 @ *** ***指装饰器的函数名

　　2、装饰器作为一个函数，他会把其下一行的主函数名作为变量，传递到自己的函数去调用。再重新赋值主函数。

　　3、装饰器必须放到被装饰的函数上边，并且独占一行；多个装饰器下 ，按照先下后上执行。

三、为什么要用装饰器？

开放封闭原则

源代码上线之后，主函数的定义和调用方式尽量避免更改。装饰器本身在源代码中就是可调用的对象。

四、装饰器使用举例

原函数如下：

import time
def index():
    time.sleep(3)
    print("welcome to oldboy")
index()

执行结果：

welcome to oldboy

在原函数不被更改的情况下，增加新的功能：计算主函数执行的时间！

1、最原始的无参装饰器（无位置参数）。

import time

def timer(func):
    def wrapper():
        start_time=time.time()
        func()
        stop_time = time.time()
        print("running time:%s"%(stop_time-start_time))
    return wrapper
@timer
def index():
    time.sleep(3)
    print("welcome to oldboy")
index()

执行结果：

welcome to oldboy
running time:3.000171422958374

2、无参装饰器（有位置参数）

#为定义可以传值的函数，添加计时的装饰器，此时要注意函数和装饰器中的位置参数传值的问题！！！
import time

def timer(func):
    def wrapper(*args,**kwargs):  #有权接管，无权为空！
        start_time = time.time()
        func(*args,**kwargs)  #home(name)
        stop_time = time.time()
        print("run time is %s"%(stop_time-start_time))
    return wrapper

@timer   #index = timer(index)
def index():
    time.sleep(1.5)
    print("welcome to beijing!")

@timer
def home(name):  #wrapper('alex')
    time.sleep(1)
    print("welcome to %s home page"%name)

@timer
def auth(name,passwd):
    time.sleep(0.5)
    print(name,passwd)

@timer
def tell():
    print("---***---")

index()
home("alex")
auth("alex","alex3714")
tell()

执行结果如下：

welcome to beijing!
run time is 1.5000858306884766
welcome to alex home page
run time is 1.0000569820404053
alex alex3714
run time is 0.5010290145874023
---***---
run time is 0.0

3、无参装饰器（主函数带有返回值的情况！）

#给带有返回值的函数添加统计时间的装饰器!

import time

def timer(func):
    def wrapper(*args,**kwargs):
        start_time = time.time()
        res = func(*args,**kwargs) #my_max(1,2)
        stop_time = time.time()
        print("run time is %s"%(stop_time-start_time))
        return res  #先将所有的功能接收出来，然后在返回
    return wrapper

@timer
def my_max(x,y):
    time.sleep(2)
    print("max:")
    res = x if x>y else y
    return res   #函数的返回值！
res = my_max(1,2)   # res = wrapper(1,2)
print(res)

执行结果如下：

max:
run time is 2.0001144409179688
2

4、装饰器的实质既然是函数的调用，那么上面介绍完无参的情况，现在举例介绍有参的情况。

有参装饰器（登陆认证）

#为主函数添加认证功能。主函数不变，在主函数上@*** 一个装饰器，装饰器是靠函数来实现。
#认证功能是在真正的函数调用之前添加上认证的功能，所以认证是在装饰器中添加。
#有参装饰器是在装饰器上添加区分，认证的时候，需要区分不同类型信息的来源（如：本地文件，数据库），
#所以在装饰器上就需要添加认证参数，以区分开认证的来源，执行认证的时候，会在认证主函数中进行逻辑操作。
def auth2(auth_type):  #参数认证函数
    def auth(func):
        def wrapper(*args,**kwargs):  #认证函数
            if auth_type == "file":
                name = input("username:")
                passwd = input("passwd：")
                if name == "song" and passwd =="123":
                    print("auth successful")
                    res = func(*args,**kwargs)
                    return res   #认证成功之后源代码再运行。
                else:
                    print("auth error")
            elif auth_type == "sql":
                print("不会！")
        return wrapper
    return auth

#@auth2(auth_type="sql")    #调用  不同认证类型的参数
#def index():  #index = auth(index)
#    print("welcome to index page")

@auth2(auth_type="file")  #调用  不同认证类型的参数
def my_index():
    print("welcome to inexd page")
#index()
my_index()

执行结果如下：

username:song
passwd123
auth successful
welcome to inexd page

5、多个装饰器调用！实现主函数的登录认证功能

#多个装饰器调用！
#1、加上了统计函数的执行时间
#2、加上了认证功能，为避免重复登录认证，执行功能完成之后再判断登录状态。已登录则跳过，第一次登录会有登录信息的认证。

import time
current_login={'name':None,'login':False}

def timmer(func):
    def wrapper(*args,**kwargs):
        start_time=time.time()
        res=func(*args,**kwargs)  #my_max(1,2)
        stop_time=time.time()
        print('run time is %s' %(stop_time-start_time))
        return res
    return wrapper

def auth2(auth_type='file'):
    def auth(func):
        # print(auth_type)
        def wrapper(*args,**kwargs):
            if current_login['name'] and current_login['login']:
                res=func(*args,**kwargs)
                return res
            if auth_type == 'file':
                name=input('username: ')
                password=input('password: ')
                if name == 'zhejiangF4' and password == 'sb945':
                    print('auth successfull')
                    res=func(*args,**kwargs)
                    current_login['name']=name
                    current_login['login']=True
                    return res
                else:
                    print('auth error')
            elif auth_type == 'sql':
                print('还他妈不会玩')
        return wrapper
    return auth

@timmer
@auth2(auth_type='file') #@auth  #index=auth(index)
def index():
    print('welcome to inex page')

@auth2()
def home():
    print('welcome to home page')



#调用阶段
index()
home()

执行结果如下：

username: zhejiangF4
password: sb945
auth successfull
welcome to inex page
run time is 10.760615348815918
welcome to home page