如何从数据类中动态删除字段?

Question

我想继承我的数据类，但删除它的一些字段。我如何在运行时做到这一点，这样我就不需要一个一个地复制所有成员?

的例子:

from dataclasses import dataclass
@dataclass
class A:
a: int
b: int
c: int
d: int
@remove("c", "d")
class B(A):
pass

使得A定义了a, b, c, d，而B只定义了a和b。

Answer 1

我们可以从__annotations__字典和__dataclass_fields__字典中删除特定字段，然后使用dataclasses.make_dataclass重建我们的类:

def remove(*fields):
def _(cls):
fields_copy = copy.copy(cls.__dataclass_fields__)
annotations_copy = copy.deepcopy(cls.__annotations__)
for field in fields:
del fields_copy[field]
del annotations_copy[field]
d_cls = dataclasses.make_dataclass(cls.__name__, annotations_copy)
d_cls.__dataclass_fields__ = fields_copy
return d_cls
return _

注意我们复制注释和字段是为了不影响A，否则它也会从A中删除这些字段，任何其他继承A并再次删除我们已经删除的字段的尝试都会导致错误。即:

from dataclasses import dataclass
@dataclass
class A:
a: int
b: int
c: int
d: int
@remove("c", "d")
class B(A):
pass
@remove("b", "c")
class C(A):
pass

这将给我们一个KeyError，因为我们已经删除了"而这在A的字典中已经不存在了。

Answer 2

这是另一种实现。

from dataclasses import make_dataclass, fields, dataclass
@dataclass
class A:
a: int
b: int
c: int
d: int
def remove(*exclusions):
def wrapper(cls):
new_fields = [(i.name, i.type, i) for i in fields(cls) if i.name not in exclusions]
return make_dataclass(cls.__name__, new_fields)
return wrapper
@remove('b', 'a')
class B(A):
pass
foo = B(1, 2)
baz = A(1, 2, 3, 4)
print(foo)
print(baz)

虽然在写出来的时候，我意识到它可能更适合作为一个类工厂函数，因为实例检查(如isinstance(foo, A))会失败，代码的用户可能会发现它令人惊讶。

from dataclasses import make_dataclass, fields, dataclass
@dataclass
class A:
a: int
b: int
c: int
d: int
def factory(base, name, exclusions):
new_fields = [(i.name, i.type, i) for i in fields(base) if i.name not in exclusions]
return make_dataclass(name, new_fields)
B = factory(base=A, name='B', exclusions=('b', 'c'))
foo = B(1, 2)
baz = A(1, 2, 3, 4)
print(foo)
print(baz)