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的回答，我意识到自我只是第一个参数，因此根本不需要使用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模型和Object之间在这个问题上的区别是什么？

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）是因为当Dummy装饰器以validate_input作为参数时，尚未定义Dummy类。我想这可能是以不同的方式实现的，但是现在这就是Python的工作方式。我看到的最简单的解决方案是坚持使用getattr，这将起作用，因为它在运行时查找方法。

Decorator调用实例方法

1 个答案: