本来想写一个登录验证装饰器的,然后卡在了,装饰器里面如何拿到被装饰方法里的参数,也就是请求过来的token,找了好久也没有找到,可能是搜索的方式不对,后面在书上无意中看到了我想要的,记录一下;
我就直接上代码了,简单明了,代码只是简单的思路,剩下的自己发散思维去写吧
from functools import wraps
def check_login(func):
"""
登录认证装饰器
param: func:是被装饰的方法
"""
# wraps的作用就是不会修改被装饰的方法名(防止多个装饰器重叠报错)
@wraps(func)
def wrapper(*args, **kwargs):
"""
*args和*kwargs不懂的可以自己网上查下
"""
user_name = kwargs.get('user', '')
passwd = kwargs.get('passwd', '')
if user_name == 'admin' and passwd == '1234':
print('验证通过')
# 这一行如果不写,他是不会跑到test_func方法里面的
func(*args, **kwargs)
else:
# 直接抛异常
raise ValueError('验证失败')
return wrapper
@check_login
def test_func(user, passwd):
print('登录成功')
if __name__ == '__main__':
test_func(user='admin', passwd='1234')