Python装饰器

一.装饰器的概念

装饰器它不是一个新的知识点，它是有之前我们学习的名称空间、函数嵌套、闭包函数等知识点汇总而来

什么是装饰器？

           器：指工具
              装饰：添加功能

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

def index():
    pass

index()

# 给index函数添加统计执行时间的功能.
补充：如何计算时间
import time

import time # 内置的时间模块，拿来直接用

print(time.time()) # 1694139138.4503667 称为是时间戳：1694139138秒数，从1970-1-1到当前时间的秒数

# 1694139138.4503667-------------》格式化之后的时间

time.sleep(3) # 睡眠
print(123)

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))

二.装饰器的简易版本

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()

三.装饰器进阶版本（解决参数问题、解决返回值问题）

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 outer(func):
    # func:index
    # func = home
    def get_time(*args, **kwargs):
        # func:index
        start_time = time.time()
        # index() # 只能够统计index函数的时间
        # func()---->index()
        # func:home
        res=func(*args, **kwargs)  # 只能够统计index函数的时间
        end_time = time.time()
        print('执行时间：%s' % (end_time - start_time))
        # return res # 其实返回的是上面index的返回值
        return res
    return get_time

home = outer(home)
res=home('jerry')  # get_time('jerry')
print(res)

五.装饰器练习题(认证登录功能)

# 必须登录之后才能够访问函数
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()

"""
我们现在需要做：只要登录成功一次，后续的函数就不用在登录了，直接可以访问?
"""

六.装饰器的固定模板

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. 语法糖的原理：
    它会把下面的被装饰对象的名字当成参数传递给装饰器
"""