如何根据条件选择应用哪个装饰器

Question

是否可以根据条件应用decorator？下面是一个琐碎的例子：

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使用Flask框架，它有装饰器jwt_required和jwt_optional，在这些装饰器中我不能修改这些装饰器的来源。我正在设法找到解决这个问题的办法。如有任何帮助，将不胜感激

Answer 1

对于flask jwt扩展的特定情况，您可以在自己的自定义装饰器中执行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