45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
import dataclasses
|
|
import typing
|
|
|
|
|
|
def nested_replace(obj, new_values: typing.Dict[str, typing.Any]):
|
|
"""Replace an attribute on a dataclass, much like dataclasses.replace,
|
|
except it supports nested replacement definitions. For example:
|
|
|
|
>>> new_obj = nested_replace(obj, {'attr1.sub_attr2.sub_sub_attr3': 4})
|
|
>>> new_obj.attr1.sub_attr2.sub_sub_attr3
|
|
4
|
|
|
|
"""
|
|
new_inst = obj
|
|
for name, value in new_values.items():
|
|
if '.' in name:
|
|
# Recurse into the desired name and come out with a top-level
|
|
# dataclass which has been updated appropriately.
|
|
name, remainder = name.split('.', 1)
|
|
value = nested_replace(
|
|
getattr(new_inst, name),
|
|
{remainder: value}
|
|
)
|
|
# We have a plain old name. So set it.
|
|
new_inst = dataclasses.replace(new_inst, **{name: value})
|
|
return new_inst
|
|
|
|
|
|
def replace(obj, **changes):
|
|
"""
|
|
A version of dataclasses.replace that handles ClassVar declarations.
|
|
|
|
See https://bugs.python.org/issue33796.
|
|
|
|
"""
|
|
|
|
orig = obj.__dataclass_fields__
|
|
object.__setattr__(
|
|
obj, '__dataclass_fields__',
|
|
{name: field for name, field in orig.items()
|
|
if field._field_type is not dataclasses._FIELD_CLASSVAR}
|
|
)
|
|
new = dataclasses.replace(obj, **changes)
|
|
object.__setattr__(obj, '__dataclass_fields__', orig)
|
|
return new
|