以前用装饰器, 都是定义好了装饰器后, 使用 @装饰器名的方法写入被装饰函数的正上方
在这里, 定义的装饰器都是没有参数的
在定义装饰器的函数的时候, 没有在括号里定义参数, 这就叫做无参装饰器
既然有无参装饰器, 那么当然也就会有有参装饰器
有参装饰器的定义和使用
定义一个普通的装饰器
- db_path = "db.txt"
- login_dic = {
- "user": None,
- "status": None,
- }
- def auth(func):
- def wrapper(*args, **kwargs):
- if login_dic['user'] and login_dic['status']:
- res = func(*args, **kwargs)
- return res
- name = input("your name:").strip()
- password = input("your password:").strip()
- with open(db_path, "r", encoding="utf-8") as f:
- user_dic = eval(f.read())
- if name in user_dic and password == user_dic[name]:
- print("login ok")
- login_dic['user'] = name
- login_dic['status'] = True
- res = func(*args, **kwargs)
- return res
- else:
- print("login error")
- return wrapper
- @auth
- def index():
- print("welcome to index page")
- @auth
- def home(name):
- print("welcome to %s home page" % name)
- index()
- home()
上面的例子是一个典型的基于文件的认证方式的装饰器
在生产环境中, 用户的认证方式可能有很多种, 比如还有 LDAP 和数据库的认证方式
如果想把上面的代码修改, 添加基于不同认证方式的用户认证
- db_path = "db.txt"
- login_dic = {
- "user": None,
- "status": None,
- }
- def deco(auth_type):
- def auth(func):
- def wrapper(*args, **kwargs):
- if auth_type == "file":
- if login_dic['user'] and login_dic['status']:
- res = func(*args, **kwargs)
- return res
- name = input("your name:").strip()
- password = input("your password:").strip()
- with open(db_path, "r", encoding="utf-8") as f:
- user_dic = eval(f.read())
- if name in user_dic and password == user_dic[name]:
- print("login ok")
- login_dic['user'] = name
- login_dic['status'] = True
- res = func(*args, **kwargs)
- return res
- else:
- print("login error")
- elif auth_type == 'ldap':
- print("ldap 认证方式")
- elif auth_type == "sql":
- print("数据库的认证方式")
- else:
- print("不知道的认证方式")
- return wrapper
- return auth
有参装饰器的调用
- @deco(auth_type="file")
- def index():
- print("welcome to index page")
- @deco(auth_type="abc")
- def home(name):
- print("welcome to %s home page" % name)
- index()
- home()
在上面的例子里,
deco(auth_type="file")
这个函数的执行以后就得到了 auth 函数, 但是
现在得到的 auth 函数跟以前的 auth 函数是不一样的
,
现在的 auth 函数内部添加了认证方式这样一个参数
,
这样
添加 auth_type 参数后的结果就相当于加在 index 函数和 home 函数正上方的是 auth 函数
, 这样就为最开始时的 auth 函数添加了一个参数
函数中的路由功能
先定义 3 个函数, 现在想做的是只需要用户输入字符串, 就执行对应的函数
可以通过为每个函数添加装饰器的功能,
把普通函数添加到函数字典中, 然后到函数字典中找到输入字符串对应的函数地址, 加括号就可以运行
了
类似于
- func_dic = {"f1": f1, "f2": f2, "f3": f3}
- def f1():
- print("f1 func")
- def f2():
- print("f2 func")
- def f3():
- print("f3 func")
这里就要使用到有参装饰器
定义有参装饰器
- def make_func_dic(key):
- def deco(func):
- func_dic[key] = func
- return deco
用装饰器 make_func_dic 装饰函数
- func_dic = {}
- def make_func_dic(key):
- def deco(func):
- func_dic[key] = func
- return deco
- @make_func_dic("f1") # 等同于 deco(f1)
- def f1():
- print("f1 func")
- @make_func_dic("f2") # 等同于 deco(f2)
- def f2():
- print("f2 func")
- @make_func_dic("f3") # 等同于 deco(f3)
- def f3():
- print("f3 func")
打印 func_dic, 得到结果
{'f1': <function f1 at 0x0000000002D2A510>, 'f2': <function f2 at 0x0000000002D2A598>, 'f3': <function f3 at 0x0000000002D2A620>}
可以看到 func_dic 字典中,
每个字符串对应的是相应字符串的函数的内存地址
这样想运行 f1 函数, 只需要把 func_dic 中的 "f1" 的值加括号运行就可以了
然后使用户输入字符串, 运行字符串对应的函数
- while True:
- cmd=input(">>:").strip()
- if cmd in func_dic:
- func_dic[cmd]()
这样就可以得到想要的结果了
这样做的好处: 只需要输入对应的字符串, 就可以执行对应的函数,
不需要使用 if 和 elif 对输入的字符串进行一条一条的判断
了
这就是使用装饰器在代码级别达到函数的路由功能
来源: https://www.cnblogs.com/renpingsheng/p/8516306.html