使用装饰器提供新的 args、kwargs

Question

我做了一些关于装饰器的研究，主要是关于典型的用例。 事实证明，在大多数情况下，它们充当验证器、计时器、权限检查器——非常透明的东西。 我想知道使用它们来预先计算并向装饰函数提供一些变量是否会是 Pythonic。

我脑子里的情况是这样的：

装饰器：

def foo(fn):
    @functools.wraps(fn)
    def wrapper(*args, **kwargs):
        fn_name = fn.__name__
        var = Test._do_something(Test.store_dict[fn_name])
        kw_args = {**kwargs, **dict(var=var)}
        return fn(*args, **kw_args)
    return wrapper

班级：

class Test:
    store_dict = dict(fn1='fn1_operator',
                      fn2='fn2_operator')

    @staticmethod
    @foo
    def fn1(dummy_data, **kwargs):
        return Test.__handle_data(dummy_data=dummy_data,
                                  **kwargs)
    @staticmethod
    @foo
    def fn2(dummy_data, **kwargs):
        return Test.__handle_data(dummy_data=dummy_data,
                                  **kwargs)

    @staticmethod
    def _do_something(var):
        return var

    @staticmethod
    def __handle_data(dummy_data, var):
        return (dummy_data, var)

用法：

test_instance = Test()
test_instance.fn1(dummy_data='test')

正如您所看到的， fn1和fn2方法在做几乎相同的事情（调用__handle_data方法），但具有不同的var 。 变量var取决于被调用的函数名。 使用var=fn1_operator等调用fn1结果。 在我心中有 Python 之禅：

简单胜于复杂。

我怀疑那是pythonic。

另一方面，如果没有那个装饰器，我的代码会有很多重复：

@staticmethod
def fn1(dummy_data):
    var = _do_something('fn1')
    return Test.__handle_data(dummy_data=dummy_data,
                              var=var)
@staticmethod
def fn2(dummy_data):
    var = _do_something('fn2')
    return Test.__handle_data(dummy_data=dummy_data,
                              var=var)

这是正确的吗？ 我应该改变什么？

Answer 1

我不会在这里使用装饰器； 我会写一个函数来返回特定数据的闭包。

def make_test(arg):
    @staticmethod
    def _(dummy_data):
        var = _do_something(arg)
        return Test.__handle_data(dummy_data, var=var)
    return _

fn1 = make_test('fn1')
fn2 = make_test('fn2')

（我没有测试，但我很确定如果你装饰闭包，或者简单地返回一个未装饰的函数并编写fn1 = staticmethod(make_test('fn1'))等，它的工作原理是一样的。）

---

使用专注于常见内容的装饰器，它可能看起来像

def foo(f):
    def _(dummy_data):
        var = _do_something(f())
        return Test.__handle_data(dummy_data, var=var)
    return _

@foo
def fn1(self):
    return 'fn1'


@foo
def fn2(self):
    return 'fn2'

它具有相同的行为，但意图不太清楚。 装饰器中隐藏了太多东西。