Блог

Я пытаюсь создать два очень похожих декоратора, каждый из которых принимает общие args и kwargs , и оба из которых внутренне разделяют код. Единственное различие между их двумя определениями заключается в значении определенного флага (который не предоставляется пользователю). Полный контекст содержится в этом вопросе.

В настоящее время у меня есть первый декоратор, определенный (без ссылки на флаг) как

 import functools


def decorator_factory(*args_units, **kwargs_units):

    def _decorator(func):
        @functools.wraps(func)
        def _wrapper(*args, **kwargs):
            return f(*args, **kwargs, *args_units, **kwargs_units)

        return _wrapper
    return _decorator
 

который можно использовать следующим образом

 @decorator_factory(*args_units, **kwargs_units)
def func_to_be_decorated(*args, **kwargs):
    ...
 

Вместо этого я хочу создать двух очень похожих декораторов, что-то вроде этого

 def decorator_factory1(*args_units, **kwargs_units):
    return functools.partial(_decorator, flag=True)


def decorator_factory2(*args_units, **kwargs_units):
    return functools.partial(_decorator, flag=False)


def _decorator(func, flag):
    @functools.wraps(func)
    def _wrapper(*args, **kwargs):
        if flag:
            return f1(*args, **kwargs, *args_units, **kwargs_units)
        else:
            return f2(*args, **kwargs, *args_units, **kwargs_units)

    return _wrapper
 

который затем можно было бы использовать следующим образом

 @decorator_factory1(*args_units, **kwargs_units)
def func1_to_be_decorated(*args, **kwargs):
    ...


@decorator_factory2(*args_units, **kwargs_units)
def func2_to_be_decorated(*args, **kwargs):
    ...
 

Проблема в том , что этот шаблон не передается *args_units и **kwargs_units , поскольку они больше недоступны в локальном пространстве имен внутри _decorator функции.