pydantic dataclasses intelliSense在vscode中

Question

我在继承的Pydantic Dataclass上收到Intellisense错误。我能够将其复制到两个操作系统，一个在Vscode中，另一个在Pycharm中。

您可以使用此片段复制IntelliSense错误：

from abc import ABC, abstractmethod
from pydantic import BaseModel, dataclasses
from typing import List

# python --version -> Python 3.10.2
# pydantic=1.9.0
# Running on MacOS Monterey 12.3.1


@dataclasses.dataclass(frozen=True, eq=True)
class PersonABC(ABC):
    name: str

    @abstractmethod
    def print_bio(self):
        pass

@dataclasses.dataclass(frozen=True, eq=True)
class Jeff(PersonABC):
    age: int

    def print_bio(self):
        print(f"{self.name}: {self.age}")     # In VScode, self.name is not registered as a known variable:
                                              # `Cannot access member "name" for type "Jeff"
                                              # Member "name" is unknownPylancereportGeneralTypeIssues`

class Family(BaseModel):
    person: List[PersonABC]


jeff = Jeff(name="Jeff Gruenbaum", age=93)
print(Family(person=[jeff]))

该错误发生在实现的 print_bio 函数中。完整的错误是上面的评论。 有没有办法解决此问题并恢复Intellisense？

Answer 1

我能够通过从basemodel继承而不是使用 pydantic.dataclasses.dataclass 来解决此问题。我相信这是Pydantic Dataclasses或Pylance的错误，但是使用基本模型是一个可行的解决方案。

from abc import ABC, abstractmethod
from typing import List
from pydantic import BaseModel

# python --version -> Python 3.10.2
# pydantic=1.9.0
# Running on MacOS Monterey 12.3.1


class PersonABC(ABC, BaseModel):
    name: str

    @abstractmethod
    def print_bio(self):
        pass

class Jeff(PersonABC):
    age: int

    def print_bio(self):
        print(f"{self.name}: {self.age}")     # Now the intellisense error does not appear.

class Family(BaseModel):
    person: List[PersonABC]


jeff = Jeff(name="Jeff Gruenbaum", age=93)
print(Family(person=[jeff]))