当属性是另一个 attrs 对象时进行属性验证检查

Question

我有两个 attrs 类，其中一个是另一个的属性。我想确保如果一个特定的值被赋予，另一个特定的值。

我查看了验证器，但认为没有任何东西可以捕捉到这一点。

这是我写的：

@attr.s
class C:
    a = attr.ib(validator=attr.validators.instance_of(str))

@attr.s
class D:
    x = attr.ib(validator=attr.validators.instance_of(str))
    z = attr.ib(validator=attr.validators.instance_of(C))

    def __attrs_post_init__(self):
        if self.x == "a":
            assert self.z.a == "a12"
        elif self.x == "b":
            assert self.z.a == "b12"
        elif self.x == "c":
            assert self.z.a == "c12"


d = D(x="a", z=C("a12"))

我认为这行得通，但我感觉我可能遗漏了一些明显的东西。

Answer 1

你的方法很好，除了你可能不应该在生产代码中使用assert（当你运行python -O时它们会被优化）。

如果 z 出现在 x 之后，您还可以使用 decorator validator 获取部分初始化的实例作为参数，但我发现您的解决方案更稳健，因为它不依赖于属性定义的顺序。