如何解决如何根据条件选择要应用的装饰器?
可以根据条件应用装饰器吗?下面是一个简单的示例:
import os
def bold_decorator(fn):
def wrapper(*args,**kwargs):
return '**' + fn(*args,**kwargs) + '**'
return wrapper
def italic_decorator(fn):
def wrapper(*args,**kwargs):
return '_' + fn(*args,**kwargs) + '_'
return wrapper
if os.environ.get('STYLE'):
@bold_decorator
else:
@italic_decorator
def create_text(text=''):
return text
if __name__ == '__main__':
print(create_text('Decorator decisions'))
我想要的是,当设置了环境变量时,应用bold_decorator
。如果未设置,请使用italic_decorator
。我正在使用带有装饰器jwt_required
和jwt_optional
的JWT的Flask框架,在其中不能修改这些装饰器的源代码。我正在尝试找到解决此问题的方法。任何帮助将不胜感激
解决方法
对于flask-jwt-extended的特定情况,您可以在自己的自定义装饰器中执行if/else
逻辑,然后将其应用于所有视图函数。您可以在此处(https://flask-jwt-extended.readthedocs.io/en/stable/custom_decorators/)查看该文档的完整文档,但看起来可能像这样:
from flask_jwt_extended import (
verify_jwt_in_request,verify_jwt_in_request_optional
)
def custom_jwt_required(fn):
@wraps(fn)
def wrapper(*args,**kwargs):
if os.environ.get('ALLOW_OPTIONAL_JWT'):
verify_jwt_in_request_optional()
else:
verify_jwt_in_request()
return fn(*args,**kwargs)
return wrapper
对于更一般的情况,您可以在导入级别执行此操作。像这样:
if os.environ.get('STYLE'):
from decorators import bold_decorator as decorator
else:
from decorators import italic_decorator as decorator
@decorator
def create_text(test=''):
return text
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。