247 lines
5.8 KiB
Markdown
247 lines
5.8 KiB
Markdown
# 装饰器
|
|
|
|
## 什么是装饰器
|
|
|
|
让其他函数在不需要做任何代码变动的前提下，增加额外的功能，装饰器的返回值也是一个函数对象。
|
|
|
|
装饰器的应用场景：比如插入日志，性能测试，事务处理，缓存等等场景。
|
|
|
|
## 装饰器的形成过程
|
|
|
|
如果我想测试某个函数的执行时间
|
|
|
|
```python
|
|
import time
|
|
def func1():
|
|
print('in func1')
|
|
def timer(func):
|
|
def inner():
|
|
start = time.time()
|
|
func()
|
|
print(time.time() - start)
|
|
return inner
|
|
func1 = timer(func1)
|
|
func1()
|
|
```
|
|
|
|
但是如果有多个函数，我都想让你测试他们的执行时间，你每次是不是都得func1 = timer(func1)?这样还是有点麻烦，因为这些函数的函数名可能是不相同，有func1，func2,graph,等等，所以更简单的方法，python给你提供了，那就是语法糖。
|
|
|
|
```python
|
|
import time
|
|
def timer(func):
|
|
def inner():
|
|
start = time.time()
|
|
func()
|
|
print(time.time() - start)
|
|
return inner
|
|
@timer
|
|
def func1():
|
|
time.sleep(1)
|
|
print('in func1')
|
|
func1()
|
|
```
|
|
|
|
装饰一个带参数的函数
|
|
|
|
```python
|
|
import time
|
|
def timer(func):
|
|
def inner(a):
|
|
start = time.time()
|
|
func(a)
|
|
print(time.time() - start)
|
|
return inner
|
|
@timer
|
|
def func1(a):
|
|
time.sleep(1)
|
|
print(a)
|
|
func1('hello world')
|
|
```
|
|
|
|
装饰一个带各种参数的函数
|
|
|
|
```python
|
|
import time
|
|
def timer(func):
|
|
def inner(*args,**kwargs):
|
|
start = time.time()
|
|
func(*args,**kwargs)
|
|
print(time.time() - start)
|
|
return inner
|
|
@timer
|
|
def func1(*args,**kwargs):
|
|
print(*args,**kwargs)
|
|
func1('hello world','abc',123,432)
|
|
```
|
|
|
|
查看函数的相关信息，在加上装饰器后就失效了
|
|
|
|
```python
|
|
def index():
|
|
'''这是一条注释信息'''
|
|
print('from index')
|
|
print(index.__doc__) # 查看函数注释
|
|
print(index.__name__) # 查看函数名称
|
|
```
|
|
|
|
导入wraps装饰器
|
|
|
|
```python
|
|
from functools import wraps
|
|
def deco(func):
|
|
@wraps(func)
|
|
def wrapper(*args,**kwargs):
|
|
return func(*args,**kwargs)
|
|
return wrapper
|
|
@deco
|
|
def index():
|
|
'''这是一条注释信息'''
|
|
print('from index')
|
|
print(index.__doc__) # 查看函数注释
|
|
print(index.__name__) # 查看函数名称
|
|
```
|
|
|
|
## 开放封闭原则
|
|
|
|
一句话，软件实体应该是可扩展但是不可修改的。
|
|
|
|
* 对于扩展是开放的
|
|
* 对于修改是封闭的
|
|
|
|
装饰器完美的遵循了这个开放封闭原则
|
|
|
|
## 装饰器的主要功能和固定结构
|
|
|
|
```python
|
|
def timer(func):
|
|
def inner(*args,**kwargs):
|
|
'''执行函数之前要做的'''
|
|
re = func(*args,**kwargs)
|
|
'''执行函数之后要做的'''
|
|
return re
|
|
return inner
|
|
# 下面是加上wraps的固定结构
|
|
from functools import wraps
|
|
def timer(func):
|
|
@wraps(func)
|
|
def wrapper(*args,**kwargs)
|
|
return func(*args,**kwargs)
|
|
return wrapper
|
|
```
|
|
|
|
## 带参数的装饰器
|
|
|
|
加上一个outer函数，可以携带一个flag的值，然后控制装饰器是否生效
|
|
|
|
```python
|
|
def outer(flag):
|
|
def timer(func):
|
|
def inner(*args,**kwargs):
|
|
if flag:
|
|
print('函数开始执行')
|
|
re = func(*args,**kwargs)
|
|
if flag:
|
|
print('函数执行完毕')
|
|
return re
|
|
return inner
|
|
return timer
|
|
@outer(True)
|
|
def func():
|
|
print('test')
|
|
func()
|
|
```
|
|
|
|
## 多个装饰器装饰一个函数
|
|
|
|
```python
|
|
def wrapper1(func):
|
|
def inner():
|
|
print('第一个装饰器，在程序运行之前')
|
|
func()
|
|
print('第一个装饰器，在程序运行之后')
|
|
return inner
|
|
def wrapper2(func):
|
|
def inner():
|
|
print('第二个装饰器，在程序运行之前')
|
|
func()
|
|
print('第二个装饰器，在程序运行之后')
|
|
return inner
|
|
@wrapper1
|
|
@wrapper2
|
|
def f():
|
|
print('Hello')
|
|
f()
|
|
```
|
|
|
|
# 小练习
|
|
|
|
- 给前面的基于文件数据库的登录注册加上一个打游戏功能
|
|
- 使用装饰器验证用户是否登录
|
|
|
|
```python
|
|
def check_auth(func):
|
|
def inner(name):
|
|
if len(flag) == 1:
|
|
func(name)
|
|
else:
|
|
print("没登录就别打游戏了！")
|
|
return inner
|
|
|
|
@check_auth
|
|
def play_game(name):
|
|
print(f"{name[0]}开始游戏")
|
|
|
|
flag = []
|
|
while True:
|
|
# 获取文件中保存的用户名和密码信息
|
|
dic = {}
|
|
with open("db", "a+", encoding="utf-8") as f:
|
|
f.seek(0)
|
|
temp_data = f.readlines()
|
|
for i in temp_data:
|
|
data = i.strip()
|
|
if len(data) == 0:
|
|
continue
|
|
temp_db = data.split("✨")
|
|
dic[temp_db[0]] = temp_db[-1]
|
|
|
|
# 登录注册的逻辑
|
|
print("登录".center(30,"="))
|
|
print("请选择：\n1.登录\n2.注册\n3.退出登录\n4.打游戏")
|
|
choice = input(">>>")
|
|
if choice == "1":
|
|
username = input("输入用户名：")
|
|
if username in dic.keys():
|
|
password = input("输入密码：")
|
|
if dic[username] == password:
|
|
print("\033[42;30m 登录成功！ \033[0m")
|
|
flag.append(username)
|
|
else:
|
|
print("\033[41;37m 密码错误 \033[0m")
|
|
else:
|
|
print("\033[41;37m用户名不存在！\033[0m")
|
|
elif choice == "2":
|
|
username = input("输入用户名")
|
|
if username in dic.keys():
|
|
print("\033[34;41m 用户名已存在！\033[0m")
|
|
continue
|
|
password = input("请输入密码")
|
|
dic[username] = password
|
|
print("\033[42;30m 注册成功！ \033[0m")
|
|
elif choice == "3":
|
|
flag.clear()
|
|
print("退出成功！")
|
|
elif choice == "4":
|
|
play_game(flag)
|
|
else:
|
|
print("\033[41;37m输入错误！\033[0m")
|
|
|
|
# 将数据写入文件
|
|
with open("db", "w+", encoding="utf-8") as f:
|
|
for k,v in dic.items():
|
|
data = f"{k}✨{v}\n"
|
|
f.write(data)
|
|
|
|
```
|
|
|