def wrapper(f):
def inner(*args,**kwargs):
'被装饰之前的函数'
1.开发封闭原则
软件面世时，不可能把所有的功能都设计好，当前的未来一两年功能给你上线，定期更新迭代。对于软件之前的写的源代码一般都不会修改，对函数里面的代码以及函数的调用方式。
开放原则：在源码不改变的情况下，增加一些额外的功能。
封闭原则：不要改变源码
python中装饰器：完美的诠释了的开放封闭原则。
装饰器就是一个函数： 他要装饰一个函数，在不改变原函数的源码以及调用方式的前提下，给其增加一个额外的功能。
27.1 初识装饰器
1 李业，在一家xx科技有限公司工作，主管安排了一个任务，写一个代码测试怼怼哥写的函数的执行效率
import time
def index():
time.sleep(2)
print('欢迎访问博客园首页')
print(time.time())
start_time = time.time()
index()
end_time = time.time()
print(f'此函数的执行效率{end_time-start_time}')
2. 主管让你测试小邓，李大象,重复代码太多。
def func1():
time.sleep(2)
print('欢迎访问日记首页')
def func2():
time.sleep(1)
print('欢迎访问评论首页')
start_time = time.time()
func1()
end_time = time.time()
print(f'此函数的执行效率{end_time-start_time}')
start_time = time.time()
func2()
end_time = time.time()
print(f'此函数的执行效率{end_time-start_time}')
3. 整合到函数中
def func1():
time.sleep(2)
print('欢迎访问日记首页')
def func2():
time.sleep(1)
print('欢迎访问评论首页')
def test_time(x):
start_time = time.time()
x()
end_time = time.time()
print(f'此函数的执行效率{end_time-start_time}')
test_time(func1)
test_time(func2)
4. 怼怼哥这个函数在实际项目中被500执行，主管要求：在被执行此函数时，
同时要测试一下被执行函数的效率。
def index():
time.sleep(2)
print('欢迎访问博客园首页')
# index()
def test_time(x):
start_time = time.time()
x()
end_time = time.time()
print(f'此函数的执行效率{end_time-start_time}')
test_time(index)
版本4的问题： 开放原则满足了，封闭原则：不改变原函数的源码，以及调用方式。
违反了封闭原则：改变了函数的调用方式。
版本5： 不能改变原函数的调用方式（闭包）：
def index():
time.sleep(2)
print('欢迎访问博客园首页')
index()
def func1():
time.sleep(2)
print('欢迎访问日记首页')
def test_time(x): # x = index
def inner():
start_time = time.time()
x()
end_time = time.time()
print(f'此函数的执行效率{end_time-start_time}')
return inner
index = test_time(index)
index()
语法糖@加上装饰器函数的名
def test_time(x): # x = index
def inner():
start_time = time.time()
x()
end_time = time.time()
print(f'此函数的执行效率{end_time-start_time}')
return inner
# @test_time # index = test_time(index)
def index():
time.sleep(2)
print('欢迎访问博客园首页')
index = test_time(index)
index()
def func1():
time.sleep(2)
print('欢迎访问日记首页')
@test_time
def func2():
time.sleep(1)
print('欢迎访问评论首页')
func2 = test_time(func2)
func3 = test_time(func3)
func2()
| z |
27.2 被装饰函数带返回值
def test_time(x): # x = index
def inner():
start_time = time.time()
ret = x()
# print(F'ret: {ret}')
end_time = time.time()
print(f'此函数的执行效率{end_time-start_time}')
return ret
return inner
@test_time # index = test_time(index)
def index():
time.sleep(0.5)
print('欢迎访问博客园首页')
return True
print(index()) # inner()
27.3 被装饰函数带参数
无论加不加装饰器，你的实参'太白金星'应该传给形参n,。
但版本6不能实现传参，index('太白金星') == inner('太白金星')
import time
struct_time = time.localtime()
# print(time.strftime("%Y-%m-%d %H:%M:%S",struct_time))
def test_time(x):# x = index
def inner(*args,**kwargs):
#函数的定义：* **聚合
# args = ('苹果')
# args =(1,3)
start_time = time.time()
ret = x(*args,**kwargs)
#函数的执行：* **打散
# ret = x(*('苹果')) ==x('苹果',)
# ret = x(*(1,3)) == x(1,3)
#print(f'ret:{ret}')
end_time = time.time()
print(f'此函数的效率{end_time-start_time}')
return ret
return inner
@test_time # index = test_time(index)
def index(n):
time.sleep(0.5)
print(f'欢迎{n}访问博客园首页')
return True
@test_time # index = test_time(index)
def func2(a,b):
time.sleep(0.5)
print(f'最终结果：{a+b}')
return a + b
print(index('苹果')) # inner('苹果')
print(func2(1,3)) # == inner(1,3)
27.4 标准版装饰器
def warpper(f):
def inner(*args,**kwargs):
'''被装饰函数之前的操作'''
# print(657)
ret = f(*args,**kwargs)
'''被装饰函数之后的操作'''
# print('执行完毕了')
return ret
return inner
@warpper
def func():
print(111)
func()
装饰器的应用：在不改变原函数的源码以及调用方式前提下，为其增加额外的功能。登陆认证，打印日志等。
def wrapper(f):
def inner(*args,**kwargs):
ret = f(*args,**kwargs)
return ret
return inner
27.5 模拟博客园登录
dic_status = {
'username':None,
'status':False
}
#方法一：
# def get_pwd():
# dic = {}
# with open('qq',encoding='utf-8')as f1:
# for i in f1:
# line = i.strip().split("|")
# dic[line[0]] = line[1]
# return dic
# 方法二：
def get_pwd():
dic = {}
with open('qq',encoding='utf-8')as f1:
for i in f1:
user,pwd = i.strip().split("|")
dic[user] = pwd
return dic
# 方法三：
# def get_pwd():
# with open('qq', encoding='utf-8')as f1:
# return {i.strip().split('|')[0]:i.strip().split('|')[1] for i in f1}#字典推导式
def login(dic_pwd):
dic1 = dic_pwd()
count = 0
while count < 3:
username = input('请输入用户名').strip()
password = input('请输入密码').strip()
# if username in dic1 and password == dic1[username]:
if dic1.get(username) == password:
print('登录成功')
dic_status['username'] = username
dic_status['status'] = True
return True
else:
print('用户名或密码错误')
count += 1
return False
def auth(f):
def inner(*args,**kwargs):
'''要进行登陆认证：
有两个分支：
1，如果之前登陆成功了，直接访问。
2，如果这是之前没有登陆过，先进行三次登陆。
'''
if dic_status['status']:
ret = f(*args,**kwargs)
return ret
else:
if login(get_pwd):
ret = f(*args, **kwargs)
return ret
return inner
@auth
def article():
print('欢迎访问文章页面')
@auth
def diary():
print('欢迎访问日记页面')
@auth
def comment():
print('欢迎访问评论页面')
article()
diary()
comment()
27.6 装饰器的执行步骤
def wrapper(f): # 1,执行wrapper函数，并且将func函数名 传给 变量f
def inner(*args,**kwargs):
print(111) # 5，打印111
ret = f(*args,**kwargs) # 6，执行真正的func函数
print(222) # 8, 打印222
return ret
return inner
@wrapper # func = wrapper(func) # 2,将inner函数名给了重新定义的变量 func = inner
def func():
print(333) # 7，打印333
print(444) # 3, 打印 444
func() # 4，inner() 执行inner函数
print(555) # 9，打印555
def func():
print(666)
func = inner
print(func)
27.7 带参数的装饰器
def wrapper(f):
def inner(*args,**kwargs):
if f.__name__ == 'qq': #通过函数名
ret = f(*args,**kwargs)
return ret
else:
pass
return inner
def wrapper_out(n,*args,sex='男',):
def wrapper(f): # f
def inner(*args,**kwargs):
ret = f(*args,**kwargs) # func1()
return ret
return inner
return wrapper
def func1():
print('in func1')
func = wrapper_out(1) # wrapper函数名
ly = func(func1) # wrapper(func1) = inner
ly() # inner()
从两个文件读取账号密码，登录
方法一：代码多，重复
def wrapper_out(n):
def wrapper(f):
def inner(*args,**kwargs):
if n == 'qq':
username = input('请输入用户名：').strip()
password = input('请输入密码：').strip()
with open('qq',encoding='utf-8') as f1:
for line in f1:
user,pwd = line.strip().split('|')
if username == user and password == pwd:
print('登陆成功')
ret = f(*args,**kwargs)
return ret
return False
elif n == 'tiktok':
username = input('请输入用户名：').strip()
password = input('请输入密码：').strip()
with open('tiktok', encoding='utf-8') as f1:
for line in f1:
user, pwd = line.strip().split('|')
if username == user and password == pwd:
print('登陆成功')
ret = f(*args, **kwargs)
return ret
return False
return inner
return wrapper
@wrapper_out('qq')
def qq():
print('成功访问qq')
@wrapper_out('tiktok')
def tiktok():
print('成功访问抖音')
qq()
tiktok()
方法二：直接判断n 传进来的路径。
def wrapper_out(n):
def wrapper(f):
def inner(*args,**kwargs):
username = input('请输入用户名：').strip()
password = input('请输入密码：').strip()
with open(n,encoding='utf-8') as f1:
for line in f1:
user,pwd = line.strip().split('|')
if username == user and password == pwd:
print('登陆成功')
ret = f(*args,**kwargs)
return ret
return False
return inner
return wrapper
@wrapper_out('qq')
def qq():
print('成功访问qq')
@wrapper_out('tiktok')
def tiktok():
print('成功访问抖音')
qq()
tiktok()
@wrapper_out('qq')
def qq():
print('成功访问qq')
qq()
看到带参数的装饰器分两步执行：
'''
@wrapper_out('腾讯')
1. 执行wrapper_out('腾讯') 这个函数，把相应的参数'腾讯' 传给 n,并且得到返回值 wrapper函数名。
2. 将@与wrapper结合，得到我们之前熟悉的标准版的装饰器按照装饰器的执行流程执行。
'''
"""
方法三：qq 抖音两个文件取账号密码,带参数的装饰器
status = {'flag':False,'username':None}
def get_pwd(path):
dic = {}
with open(path, encoding='utf-8')as f1:
for i in f1:
line = i.strip().split('|')
dic[line[0]] = line[1]
return dic
def login(dic_pwd):
dic1 = dic_pwd
username = input('请输入用户名:')
pwd = input('请输入密码:')
if username in dic1 and dic1[username]==pwd:
print('登录成功')
status['flag'] = True
status['username'] = username
return True
else:
return False
def wrapper_out(n):
def wrapper(f):
def inner(*args,**kwargs):
if login(get_pwd(n)):
ret = f(*args,**kwargs)
return ret
return inner
return wrapper
@wrapper_out('qq')
def qq():
print('成功访问qq')
@wrapper_out('douyin')
def ti():
print('成功访问抖音')
qq()
ti()
27.8 多个装饰器装饰一个函数
def wrapper1(func1): # func1 = f原函数
def inner1():
print('wrapper1 ,before func') # 2
func1()
print('wrapper1 ,after func') # 4
return inner1
def wrapper2(func2): # func2 == inner1
def inner2():
print('wrapper2 ,before func') # 1
func2() # inner1
print('wrapper2 ,after func') # 5
return inner2
@wrapper2 # f = wrapper2(f) 里面的f == inner1 外面的f == inner2
@wrapper1 # f = wrapper1(f) 里面的f == func1 外面的 f == inner1
def f():
print('in f') # 3
f() # inner2()
def wrapper1(func):
def inner1():
print('wrapper1 ,before func')
ret = func()
print('wrapper1 ,after func')
return ret
return inner1
def wrapper2(func):
def inner2():
print('wrapper2 ,before func')
ret = func()
print('wrapper2 ,after func')
return ret
return inner2
def wrapper3(func):
def inner3():
print('wrapper3 ,before func')
ret = func()
print('wrapper3 ,after func')
return ret
return inner3
@wrapper3
@wrapper2
@wrapper1
def f():
print('in f')
return '哈哈哈'
print(f())