2

在核心，我想要做的是采用一些看起来像这个 未经修饰的验证函数的函数：

def f(k: bool):
    def g(n):
        # check that n is valid
        return n
    return g

并使它们看起来像这个修饰的验证函数：

@k
def f():
    def g(n):
        # check that n is valid
        return n
    return g

这里的想法k是在所有实现功能中描述相同的功能。

具体来说，这些函数都返回“验证”函数，以与voluptuous 验证框架一起使用。因此，所有类型的函数f()都返回一个稍后由Schema(). kis 实际上allow_none，也就是说一个标志，确定一个None值是否正常。一个非常简单的例子可能是这个示例使用代码：

x = "Some input value."
y = None
input_validator = Schema(f(allow_none=True))
x = input_validator(x)  # succeeds, returning x
y = input_validator(y)  # succeeds, returning None
input_validator_no_none = Schema(f(allow_none=False))
x = input_validator(x)  # succeeds, returning x
y = input_validator(y)  # raises an Invalid

在不更改示例使用代码的情况下，我试图通过将未修饰的验证函数更改为修饰的验证函数来实现相同的结果。举一个具体的例子，改变这个：

def valid_identifier(allow_none: bool=True):
    min_range = Range(min=1)
    validator = Any(All(int, min_range), All(Coerce(int), min_range))
    return Any(validator, None) if allow_none else validator

对此：

@allow_none(default=True)
def valid_identifier():
    min_range = Range(min=1)
    return Any(All(int, min_range), All(Coerce(int), min_range))

这两个返回的函数应该是等价的。

我试图写的是这个，利用decorator图书馆：

from decorator import decorator

@decorator
def allow_none(default: bool=True):
    def decorate_validator(wrapped_validator, allow_none: bool=default):
        @wraps(wrapped_validator)
        def validator_allowing_none(*args, **kwargs):
            if allow_none:
                return Any(None, wrapped_validator)
            else:
                return wrapped_validator(*args, **kwargs)
        return validator_allowing_none
    return decorate_validator

我有一个unittest.TestCase为了测试这是否按预期工作：

@allow_none()
def test_wrapped_func():
    return Schema(str)

class TestAllowNone(unittest.TestCase):

    def test_allow_none__success(self):
        test_string = "blah"

        validation_function = test_wrapped_func(allow_none=False)
        self.assertEqual(test_string, validation_function(test_string))
        self.assertEqual(None, validation_function(None))

但我的测试返回以下失败：

    def validate_callable(path, data):
        try:
>           return schema(data)
E           TypeError: test_wrapped_func() takes 0 positional arguments but 1 was given

我尝试对此进行调试，但无法让调试器实际进入装饰。我怀疑由于命名问题，例如在这个（非常长的）博客文章系列中提出的问题，test_wrapped_func没有正确设置它的参数列表，因此装饰器甚至从未被执行，但它也可能完全是另外一回事。

我尝试了其他一些变化。通过从 中删除函数括号@allow_none：

@allow_none
def test_wrapped_func():
    return Schema(str)

我得到一个不同的错误：

>       validation_function = test_wrapped_func(allow_none=False)
E       TypeError: test_wrapped_func() got an unexpected keyword argument 'allow_none'

删除@decorator失败：

>       validation_function = test_wrapped_func(allow_none=False)
E       TypeError: decorate_validator() missing 1 required positional argument: 'wrapped_validator'

这是有道理的，因为@allow_none需要一个参数，因此逻辑上需要括号。替换它们会给出原始错误。

装饰器很微妙，我显然在这里遗漏了一些东西。这类似于对函数进行柯里化，但它并不完全有效。关于如何实施，我缺少什么？

4

1 回答 1

2

我认为您将allow_none=default论点置于错误的嵌套级别。它应该在最里面的函数（包装器）上，而不是装饰器（中间层）。

尝试这样的事情：

def allow_none(default=True):    # this is the decorator factory
    def decorator(validator):    # this is the decorator
        @wraps(validator)
        def wrapper(*args, allow_none=default, **kwargs):    # this is the wrapper
            if allow_none:
                return Any(None, validator)
            else:
                return validator(*args, **kwargs)
        return wrapper
    return decorator

如果您不需要可设置默认值，则可以摆脱最外层的嵌套，只需将默认值设置为包装函数中的常量（或者如果您的调用者总是传递一个值，则省略它）。请注意，正如我在上面所写的，allow_none包装器的参数是仅关键字参数。如果您想将它作为位置参数传递，您可以将它移到 前面*args，但这要求它是第一个位置参数，从 API 的角度来看，这可能是不可取的。更复杂的解决方案可能是可能的，但对于这个答案来说太过分了。

于 2015-10-30T01:30:27.253 回答