2

我有几个使用 attrs 定义的类,如下所示:

from attr import attrs, attrib, validators

@attrs
class MyClass:
    name = attrib(])
    @name.validator
    def check_length(self, attribute, value):
        if not (3 <= len(value) <= 30):
            raise ValueError("Name must be between 3 and 30 characters")

    description = attrib()
    @description.validator
    def check_length(self, attribute, value):
        if not (10 <= len(value) <= 400):
            raise ValueError("Description must be between 10 and 400 characters")

对于几个属性,我需要创建一个验证器来检查它是否在某个范围内的数据。我想避免重复,所以我可以创建一个自定义验证器,在其中我为 min 和 max 传递一些额外的参数,比如这个:

def range_validator(instance, attribute, value, min_value, max_value):
    if  min_value >= len(value) >= max_value:
        raise ValueError("Must be between {} and {}".format(min_value, max_value))

但是我不知道如何从 attrib() 内部调用这个验证器,传递额外的参数(min_value 和 max_value),我的意思是做一些类似的事情:

name = attrib(validator=[range_validator(self, 10, 30)])
4

1 回答 1

4

你可以使用functools.partial

def range_validator(instance, attribute, value, min_value, max_value):
    lv = len(value)
    if min_value > lv or lv > max_value:
        raise ValueError("Must be between {} and {}".format(min_value, max_value))

@attrs
class C:
    x = attrib(validator=partial(range_validator, min_value=10, max_value=30))

或者,您可以使用闭包:

def make_range_validator(min_value, max_value):
    def range_validator(instance, attribute, value):
        lv = len(value)
        if min_value > lv or lv > max_value:
            raise ValueError("Must be between {} and {}".format(min_value, max_value))

    return range_validator

@attrs
class C:
    x = attrib(validator=make_range_validator(10, 30))

我个人更喜欢闭包工厂方法,因为它们更明确地说明了你在做什么。Partials 对我来说总是有点特别,但那可能只是我。

(请注意,我冒昧地修复了您的验证器中的逻辑错误——您可能也想应用它。:))

于 2018-08-23T10:49:19.520 回答