如何获取不可变对象的修改后的副本?
即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
背景:我想在我的配置中使用不可变对象。但是我也想在处理非常相似的配置时避免代码重复,例如用于单元测试
答案 0 :(得分: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)
将所有传入该方法的值覆盖。然后,更新后的值将传递到新对象的构造函数/初始值,并返回。
希望有帮助,祝您编码愉快!
答案 1 :(得分:1)
attrs
具有专门为您提供的功能:https://www.attrs.org/en/stable/api.html#attr.evolve
由于将实例作为第一个参数,因此您也可以使用它来定义方法:
@attr.s
class C:
some_magical_method = attr.evolve