方法装饰器的 Pythonic 应用？

Question

我试图了解如何编写一个装饰器来检查每个方法的 main_val 属性，尤其是当 main_val 属性的位置可能发生变化，甚至有默认值时。另外，有没有更 Pythonic 的方式来应用这个条件？

我在下面写了一个例子来形象地描述这个问题，希望有更多经验的人能解释一下。

对于下面的每个方法，main_val 应该是我的装饰器应该检查的唯一属性。我不确定如何强制装饰器始终检查 main_val，除非它作为 key=value 属性传递或引用 *args 中的硬编码位置。

class TestCase:
    def __init__(self):
        self.allowable_list = [1,2,3,4,5]

    def decorator_check(func):
        # I want to apply condition to "main_val" attribute of any method this decorator is applied to
        def wrapper(self, val, *args, **kwargs):
            if val not in self.allowable_list:
                raise AttributeError("value {} not in allowable list {}".format(val, self.allowable_list))
        return func(self, val, *args, **kwargs)
    return wrapper

    @decorator_check
    def calc(self, main_val, val_num_two, val_num_three):
        print("(calc) - main_val: {} val_num_two: {} val_num_three: {}".format(main_val, val_num_two, val_num_three))
        return main_val * val_num_two * val_num_three

    @decorator_check
    def calc_two(self, another_val, main_val,  val_num_two):
        print("(calc_two) - another_val: {} main_val: {} val_num_two: {}".format(another_val, main_val, val_num_two))
        return another_val * main_val * val_num_two

    @decorator_check
    def calc_three(self, another_val, val_num_two, main_val=3):
        print("(calc_two) - another_val: {} main_val: {} val_num_two: {}".format(another_val, main_val, val_num_two))
        return another_val * val_num_two * main_val

test_obj = TestCase()

---

test_obj.calc(100, 2 ,3)

按预期返回错误

test_obj.calc_two(2, 5, 3)

返回 (calc_two) - another_val: 2 main_val: 5 val_num_two: 3

test_obj.calc_three(2,5,100)

返回 (calc_two) - another_val: 2 val_num_two: 5 main_val: 100

有什么方法可以解决这个问题？

Answer 1

使用inspect.signature 确定签名、绑定参数、应用默认值并计算main_val 的最终值：

import functools
import inspect

def main_val_decorator(f):
    f_sig = inspect.signature(f)
    @functools.wraps(f)
    def wrapper(*args, **kwargs):
        bound = f_sig.bind(*args, **kwargs)
        bound.apply_defaults()
        main_val = bound.arguments['main_val']

        do_whatever_with(main_val)
        return f(*args, **kwargs)
    return wrapper

这比您需要做的工作更多，因为它也确定所有其他参数的绑定，但它比手动执行自省要方便得多。