调用实例方法的装饰器

Question

我有一个类A 和方法do_something(self,a,b,c) 和另一个实例方法验证输入并检查名为can_do_something(self,a,b,c) 的权限。

这是我代码中的常见模式，我想编写一个接受验证函数名称并执行测试的装饰器。

def validate_input(validation_fn_name):
        def validation_decorator(func):
            def validate_input_action(self,*args):
                error = getattr(self,validation_fn_name)(*args)
                if not error == True:
                    raise error
                else:
                    return func(*args)
            return validate_input_action
        return validation_decorator

调用函数如下

@validate_input('can_do_something')
def do_something(self,a,b,c):
   return a + b + c

问题是我不确定如何通过验证功能维护self。我已将验证 fn 名称与 getattr 一起使用，因此 fn 可以在实例的上下文中运行，但我无法为 func(*args) 执行此操作。

那么实现这一目标的正确方法是什么？

谢谢。

编辑

所以按照@André Laszlo 的回答，我意识到 self 只是第一个参数，所以根本不需要使用getattr，只需传递*args。

def validate_input(validation_fn):
    def validation_decorator(func):
        def validate_input_action(*args):
            error = validation_fn(*args)
            if not error == True:
                raise error
            else:
                return func(*args)
        return validate_input_action
    return validation_decorator

更优雅，它也支持静态方法。

向@André Laszlo 示例添加静态方法证明装饰器正在工作：

 class Foo(object):
    @staticmethod
    def validate_baz(a,b,c):
       if a > b:
          return ValueError('a gt b')

    @staticmethod
    @validate_input(Foo.validate_baz)
    def baz(a,b,c):
       print a,b,c

    >>> Foo.baz(1,2,3)
    1 2 3
    >>> Foo.baz(2,1,3)
    Traceback (most recent call last):
      File "<stdin>", line 1, in <module>
      File "<stdin>", line 6, in validate_input_action
    ValueError: a gt b

但是，当我尝试在 django 模型中做同样的事情时：

from django.db import models
from django.conf import settings

settings.configure()

class Dummy(models.Model):
    id = models.AutoField(primary_key=True)
    name = models.CharField(max_length=10)

    def can_say_name(self):
        if name is None:
            return Exception('Does not have a name')

    @validate_input(can_say_name)
    def say_name(self):
        print self.name

    @staticmethod
    def can_create_dummy(name):
        if name == 'noname':
            return Exception('No name is not a name !')

    @staticmethod
    @validate_input(Dummy.can_create_dummy)
    def create_dummy(name):
        return Dummy.objects.create(name=name)

我得到以下信息：

NameError: name 'Dummy' is not defined

那么在这个问题上，django 模型和对象有什么不同呢？

Answer 1

我认为这是你想要的：

def validate_input(validation_fn_name):
    def validation_decorator(func):
        def validate_input_action(self, *args):
            error = getattr(self, validation_fn_name)(*args)
            if error is not None:
                raise error
            else:
                arglist = [self] + list(args)
                return func(*arglist)
        return validate_input_action
    return validation_decorator

class Foo(object):

    def validate_length(self, arg1):
        if len(arg1) < 3:
            return ValueError('%r is too short' % arg1)

    @validate_input('validate_length')
    def bar(self, arg1):
        print "Arg1 is %r" % arg1


if __name__ == "__main__":
    f = Foo()
    f.bar('hello')
    f.bar('')

输出是：

Arg1 is 'hello'
Traceback (most recent call last):
  File "validator.py", line 27, in <module>
    f.bar('')
  File "validator.py", line 6, in validate_input_action
    raise error
ValueError: '' is too short

更新答案

出现错误 (NameError: name 'Dummy' is not defined) 是因为当 validate_input 装饰器将 Dummy 作为参数时，尚未定义 Dummy 类。我想这可能以不同的方式实现，但目前这就是 Python 的工作方式。我看到的最简单的解决方案是坚持使用getattr，因为它会在运行时查找方法。