一.装饰器的概念
装饰器它不是一个新的知识点,它是有之前我们学习的名称空间、函数嵌套、闭包函数等知识点汇总而来
什么是装饰器?
器:指工具
装饰:添加功能
装饰器的核心思想:
在'不改变原来的调用方式'和'内部代码'的基础之上给函数'添加额外的功能'.
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. 语法糖的原理:
它会把下面的被装饰对象的名字当成参数传递给装饰器
"""