Python之装饰器

一、什么装饰器
        '装饰'代指为被装饰对象添加新的功能，’器’代指器具/工具，装饰器与被装饰的对象均可以是任意可调用对象。概括地讲，装饰器的作用就是在不修改被装饰对象源代码和调用方式的前提下为被装饰对象添加额外的功能。装饰器经常用于有切面需求的场景，比如：插入日志、性能测试、事务处理、缓存、权限校验等应用场景，装饰器是解决这类问题的绝佳设计，有了装饰器，就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。

装饰器的核心思想：
        在'不改变原来的调用方式'和'内部代码'的基础之上给函数'添加额外的功能'

二、装饰器的实现
函数装饰器分为：无参装饰器和有参装饰两种

两者原理：都是’函数嵌套+闭包+函数对象’的组合使用的产物

2.1 无参装饰器的实现
以记录代码运行时间为例，进行装饰器的实现

def index():
    time.sleep(3)
    print('from index')
 
 
# 1. 在执行函数之前加一个时间点
start_time = time.time() # 此时的时间戳
index()
 
# 2. 等index函数执行之后在打个时间点
end_time = time.time()
 
# 3. 获取index 的总执行时间
print('index一共执行了%s秒' % (end_time - start_time))
2.2.1 装饰器的简易版本（无参修饰器）
import time
 
 
def index():
    time.sleep(3)
    print('from index')
 
 
def home():
    time.sleep(1)
    print('from home')
 
 
def outer(func):
    # func:index
    # func = home
    def get_time():
        # func:index
        start_time = time.time()
        # index() # 只能够统计index函数的时间
        # func()---->index()
        func()  # 只能够统计index函数的时间
        end_time = time.time()
        print('执行时间：%s' % (end_time - start_time))
 
    return get_time
 
 
# get_time(index()) # get_time(None)
# get_time(index) #
 
index = outer(index)  # res:get_time的函数名(其实也是get_time的内存地址)
index()  # get_time()
# res = outer(home)
# res()

2.2.2装饰器进阶版本（有参修饰器）
（解决参数问题、解决返回值问题）

import time
 
 
def index():
    time.sleep(3)
    print('from index')
 
 
def home(name):
    time.sleep(1)
    print('from home:', name)
 
 
def outer(func):
    # func:index
    # func = home
    def get_time(*args, **kwargs):
        # func:index
        start_time = time.time()
        # index() # 只能够统计index函数的时间
        # func()---->index()
        # func:home
        func(*args, **kwargs)  # 只能够统计index函数的时间
        end_time = time.time()
        print('执行时间：%s' % (end_time - start_time))
 
    return get_time
 
 
# get_time(index()) # get_time(None)
# get_time(index) #
 
index = outer(index)  # res:get_time的函数名(其实也是get_time的内存地址)
index('aaa')  # get_time()
# home = outer(home)
# home('jerry')  # get_time('jerry')
 
'''
我们现在统计的函数有时候需要参数，有时候不需要参数，所以，参数到底应该传还是不传？
    我们确定不了什么时候传，或者是传几个?
        如何将解决?
            *args和**kwargs
            
'''

解决返回值问题

import time
 
 
def index():
    time.sleep(3)
    print('from index')
 
 
def home(name):
    time.sleep(1)
    print('from home:', name)
 
 
def outer(func):
    # func:index
    # func = home
    def get_time(*args, **kwargs):
        # func:index
        start_time = time.time()
        # index() # 只能够统计index函数的时间
        # func()---->index()
        # func:home
        func(*args, **kwargs)  # 只能够统计index函数的时间
        end_time = time.time()
        print('执行时间：%s' % (end_time - start_time))
 
    return get_time
 
 
# get_time(index()) # get_time(None)
# get_time(index) #
 
index = outer(index)  # res:get_time的函数名(其实也是get_time的内存地址)
index('aaa')  # get_time()
# home = outer(home)
# home('jerry')  # get_time('jerry')
 
'''
我们现在统计的函数有时候需要参数，有时候不需要参数，所以，参数到底应该传还是不传？
    我们确定不了什么时候传，或者是传几个?
        如何将解决?
            *args和**kwargs
            
'''

三、装饰器练习题(认证登录功能)
# 必须登录之后才能够访问函数
def index():
    print('from index')
 
 
def home():
    print('from home')
 
def home1():
    print('from home1')
 
# index()
# 用户必须是登录之后才能访问，否则不让访问
 
is_login={'is_login':False}
 
# 1. 写一个登录程序
def outer(func):
    # func = index
    def auth():
        if is_login.get('is_login'):
            ## 如果条件成立，直接执行函数
            res=func()
            return res # 解决返回值问题的
 
        username = input('username:>>>').strip()
        password = input('password:>>>').strip()
 
        # 2. 比较用户名和密码
        if username == 'jerry' and password == '123':
            # 执行函数
            print('登录成功')
            func()
            is_login['is_login'] = True #
        else:
            print('用户名或者密码错误')
    return auth
 
# auth(index)
# auth(home)
 
index=outer(index)
index()
 
home=outer(home)
home()
 
home1=outer(home1)
home1()

四、修饰器的固定模板+语法糖
#什么是语法糖：

        一种附加在计算机语言中的语法。这种语法对语言的功能没有影响，但更便于程序员使用。一般来说，使用语法糖可以提高程序的可读性，从而减少出错的几率。一种附加在计算机语言中的语法。这种语法对语言的功能没有影响，但更便于程序员使用。一般来说，使用语法糖可以提高程序的可读性，从而减少出错的几率。

1. 语法糖的书写规范：
    @装饰器名字
    必须把上面的写在被装饰对象的头上(紧贴着被装饰对象写)
    
2. 语法糖的原理：
    它会把下面的被装饰对象的名字当成参数传递给装饰器

修饰器固定模板
 
def outer(func):
    def inner(*args, **kwargs):
        print('函数被调用之前需要添加的功能')
        res=func(*args, **kwargs) # 真正的函数执行
        print('函数被调用之后需要添加的功能')
        return res
 
    return inner
 
def outer(func):
    def inner(*args, **kwargs):
        print('函数被调用之前需要添加的功能')
        res = func(*args, **kwargs)  # 真正的函数执行
        print('函数被调用之后需要添加的功能')
        return res
 
    return inner
 
语法糖
 
# @outer # index = outer(index)
# def index():
#     print('from index')
#     return 'index'
 
# @outer # home=outer(home)
# def home():
#     print('from home')
 
# # index = outer(index)
# 
# index()
# 
# home() # inner()
 
@outer # func=outer(func)
def func():
    print('from func')
    
func()
 
"""
1. 语法糖的书写规范：
    @装饰器名字
    必须把上面的写在被装饰对象的头上(紧贴着被装饰对象写)
    
2. 语法糖的原理：
    它会把下面的被装饰对象的名字当成参数传递给装饰器