Question

如何获取不可变对象的修改后的副本？

即some_magical_method看起来像这样的片段吗？

import attr

@attr.s(frozen=True, slots=True)
class Config:
    param1: int = attr.ib()
    param2: str = attr.ib()

my_base_config = Config(param1=1, param2="2")
my_derived_config = my_base_config.some_magical_method(param2="two")

print(my_derived_config.param1) # output: 1
print(my_derived_config.param2) # output: two

背景：我想在我的配置中使用不可变对象。但是我也想在处理非常相似的配置时避免代码重复，例如用于单元测试

Answer 1

这是一种满足您需求的方法！将其添加到您的对象中，这是我的代码和输出：

@attr.s(frozen=True, slots=True)
class Config:
    param1: int = attr.ib()
    param2: str = attr.ib()

    def some_magical_method(self, **kwargs):
        params_to_use = {"param1": self.param1, "param2": self.param2}
        params_to_use.update(kwargs)
        new_obj = Config(**params_to_use)
        return new_obj

my_base_config = Config(param1=1, param2="2")
my_derived_config = my_base_config.some_magical_method(param2="two")

>>> print(my_derived_config.param1) # output: 1
1
>>>print(my_derived_config.param2) # output: two
two

此处的关键是该方法首先使用其自己的值创建新的Config对象，然后使用params_to_use.update(kwargs)将所有传入该方法的值覆盖。然后，更新后的值将传递到新对象的构造函数/初始值，并返回。

希望有帮助，祝您编码愉快！

Answer 2

attrs具有专门为您提供的功能：https://www.attrs.org/en/stable/api.html#attr.evolve

由于将实例作为第一个参数，因此您也可以使用它来定义方法：

@attr.s
class C:
    some_magical_method = attr.evolve

如何获得不可变对象的修改后的副本？

2 个答案: