Python класс данных наследует часть атрибутов от родительского класса данных - PullRequest
0 голосов
/ 09 марта 2020

Я построил несколько баз dataclass. Теперь я хочу создать дочерний элемент dataclass, наследующий от этих базовых классов, но может наследовать часть атрибутов, определенных в некоторых базовых классах. Примеры могут быть:

import dataclasses

@dataclasses.dataclass
class A:
    a: int = None
    b: float = None
    c: str = None

@dataclasses.dataclass
class B:
    d: int = None
    b: float = 3.5

@dataclasses.dataclass
class C:
    e: int = None
    f: float = 3.5
    g: int = None

@dataclasses.dataclass
class D(A, B):
    def __post_init__(self):
        for _field, _field_property in C.__dataclass_fields__.items():
            if _field != "g":
                setattr(self, _field, _field_property.default)

А именно, я хочу создать дочерний класс D, наследующий A и B, и атрибуты в C, за исключением g. Проверка дочернего класса D

>>> D.__dataclass_fields__.keys() # got dict_keys(['d', 'b', 'a', 'c'])
>>> d = D(a=4, b=2, c=5, d=3.4, e=2.1, g=55)
Traceback (most recent call last):
  File "<pyshell#77>", line 1, in <module>
    d = D(a=4, b=2, c=5, d=3.4, e=2.1, g=55)
TypeError: __init__() got an unexpected keyword argument 'e'

И

>>> D.__dict__.keys()
dict_keys(['__module__', '__post_init__', '__doc__', '__dataclass_params__', '__dataclass_fields__', '__init__', '__repr__', '__eq__', '__hash__'])

Когда я изменил __post_init__ на __init__ и использовал super().__init__() для наследования, все равно не могу атрибуты из класса C и теряют преимущество dataclass, т. е.

>>> @dataclasses.dataclass
class D(A, B):
    def __init__(self):
        super().__init__()
        for _field, _field_property in C.__dataclass_fields__.items():
            if _field != "g":
                setattr(self, _field, _field_property.default)

И запускают

>>> d = D(a=4, b=2, c=5, d=3.4, e=2.1, g=55)
Traceback (most recent call last):
  File "<pyshell#81>", line 1, in <module>
    d = D(a=4, b=2, c=5, d=3.4, e=2.1, g=55)
TypeError: __init__() got an unexpected keyword argument 'a'

Что мне делать?

1 Ответ

1 голос
/ 09 марта 2020

Подобно тому, что было предложено @GiacomoAlzetta, я неожиданно выдвинул эту идею, используя dataclasses.make_dataclass, т.е. сгенерировал копию C, но исключая атрибут g, то есть

<<< C_part = dataclasses.make_dataclass("C_part", [(_field, _field_property.type, _field_property.default) for _field, _field_property in C.__dataclass_fields__.items() if _field != "g"])

Таким образом У меня есть

>>> C_part.__dataclass_fields__.keys()  # dict_keys(['e', 'f'])

Тогда D можно получить

>>> @dataclasses.dataclass
class D(A, B, C_part):
    pass

>>> d = D(a=4, b=2, c=5, d=3.4, e=2.1, f=55)
>>> d
D(e=2.1, f=55, d=3.4, b=2, a=4, c=5)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...