Как я могу использовать список [customClass] как тип с @dataclass в Python 3.7.x - PullRequest
Новая
       11

Как я могу использовать список [customClass] как тип с @dataclass в Python 3.7.x

0 голосов
/ 11 января 2019

У меня есть следующие классы данных. 

@dataclass
class Package:
    '''Class for keeping track of one destination.'''
    _address: []

@dataclass
class Destination:
'''Class for keeping track of a destination.'''
_start: str
_end: str
_distance: float

def __init__(self, param):
    self._start = param[0]
    self._end = param[1]
    self._distance = param[2]

и следующий класс данных, который вызывает вышеуказанный класс. 

@dataclass
class DestinationContainer:
    '''Class for keeping track of a package destination.
       and all the possible combinations of potential next destination '''
    _package: Package
    _destinations: List[Destination]

    def __init__(self):
        pass

    def addPkg(self,param):
        self._package = param

При попытке запустить программу я получаю следующую ошибку

TypeError: Параметры для универсальных типов должны быть типами.

Я также пытался вызвать _destinations член таким образом. 

_destinations: List[Destination] = field(default_factory=list)

Тогда я получаю следующую ошибку

TypeError: Параметры для универсальных типов должны быть типами.

Я также пытался установить ученика как 

    _destinations: []

И после проверки объекта экземпляра список внутри класса недоступен.

Я тоже пытался. 

_destinations: List = field(default_factory=lambda: [])

и я получаю следующую ошибку при попытке добавить в список

AttributeError: у объекта 'DestinationContainer' нет атрибута '_destination'

1 Ответ

0 голосов
/ 11 марта 2019

Как сказал Патрик в комментариях, ваша главная проблема в том, что вы определяете свои собственные функции __init__ при использовании @dataclass. Если вы удалите и немного реструктуризируете свой код, он должен работать как положено: 

from dataclasses import dataclass
from typing import List

@dataclass
class Package:
    _address: []

@dataclass
class Destination:
    _start: str
    _end: str
    _distance: float

@dataclass
class DestinationContainer:
    _package: Package
    _destinations: List[Destination]

    def addPkg(self, param):
        # sure this shouldn't be "self._package.append(param)"? 
        self._package = param


# works
dc = DestinationContainer(
    Package(['some address']),
    [Destination('s', 'e', 1.0)]
)
# also works
dc.addPkg(Package(['some other address']))
...