装饰器

应用场景：在原有函数的基础上，增加一点功能

例如：下面代码在返回值前面增加字符串‘当前时间：’

import time
def getXXXTime()
    return time.strftime('%Y_%m_%d %H:%M:%S',time.localtime())

装饰代码：

import time

# 定义一个装饰器函数
def sayLocal(func):
    def wrapper():
        print('当地时间：', end='') 
 7         return func()
    return wrapper

def getXXXTime():
    return time.strftime('%Y_%m_%d %H:%M:%S',time.localtime())


print(getXXXTime())
# 装饰 getXXXTime
getXXXTime = sayLocal(getXXXTime)

print (getXXXTime()) 

# 结果
2020_04_14 20:55:49
当地时间：2020_04_14 20:55:49

便捷语法：

@sayLocal
def getXXXTime():
    return time.strftime('%Y_%m_%d %H:%M:%S',time.localtime())

被装饰的函数有参数：

　　例如：

import time
def getXXXTimeFormat1(name):
    curTime = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
    return f'{curTime} ，数据采集者：{name} '


def getXXXTimeFormat2(name,place):
    curTime = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
    return f'{curTime} ，数据采集者：{name} , 采集地：{place}'

　　可以使用函数的可变参数：

import time

def sayLocal(func):
    def wrapper(*args, **kargs):
        curTime = func(*args, **kargs)
        return f'当地时间： {curTime}'
    return wrapper

@sayLocal
def getXXXTimeFormat1(name):
    curTime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())
    return f'{curTime} ，数据采集者：{name} '

@sayLocal
def getXXXTimeFormat2(name, place):
    curTime = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())
    return f'{curTime} ，数据采集者：{name} , 采集地：{place}'

print(getXXXTimeFormat1('张三'))    
print(getXXXTimeFormat2('张三',place='北京'))    

# 运行结果
当地时间： 2020-04-14 21:10:49 ，数据采集者：张三 
当地时间： 2020-04-14 21:10:49 ，数据采集者：张三 , 采集地：北京

多个装饰器装饰同一个函数：

def add_auth(func):
    print("---开始进行装饰权限1的功能---")
    def call_func(*args, **kwargs):
        print("---这是权限验证1---")
        return func(*args, **kwargs)
    return call_func

def set_func(func):
    print("---开始进行装饰权限2的功能---")
    def call_func(*args, **kwargs):
        print("---这是权限验证2---")
        return func(*args, **kwargs)
    return call_func
    # return call_func()

@add_auth
@set_func
def demo():
    print("----test1----")

demo()

　　运行结果：

---开始进行装饰权限2的功能---
---开始进行装饰权限1的功能---
---这是权限验证1---
---这是权限验证2---
----test1----

从运行结果可以看出：

• 越靠近原函数的装饰器（代码从上往下，原函数上最后一个@符后的名称）越先装饰
• 越远离原函数的装饰器越先执行，装饰器都执行完成之后再执行原函数
• 多个装饰器装饰同一个函数相当于给原函数包多层，先执行的包住后执行的，最后包住原函数执行的代码。

类装饰器：

class Test(object):

    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print("这是装饰器添加的功能......")
        return self.func(*args, **kwargs)


@Test
def get_str():
    return '哈哈哈'

print(get_str())

　　运行结果：

这是装饰器添加的功能......
哈哈哈