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

Question

可以根据条件应用装饰器吗？下面是一个简单的例子：

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 框架，在其中我无法修改这些装饰器的源代码。我正在尝试找到解决此问题的方法。任何帮助将不胜感激

Answer 1

对于 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

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

How to choose which decorator to apply based on condition?

if-statement

decorator

python-decorators

flask-jwt-extended