首页 > 解决方案 > 尽管 mypy 很高兴,但通用超类“未定义”

问题描述

我一直在与 mypy 很高兴的奇怪情况作斗争,但是当我运行代码时 python 崩溃了。这是我在使用泛型时遇到的一个问题。我在同一个目录中有两个文件。当我在 中运行 python3 时sample_parser.py,出现以下错误:

Traceback (most recent call last):
  File "/Users/joe/repos/ut-norm-james/so_generics/sample_parser.py", line 8, in <module>
    class SampleParser(LookaheadParser[str]):
NameError: name 'LookaheadParser' is not defined

这是lookahead_parser.py

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from typing import TypeVar, Generic, Optional, Callable


_T = TypeVar("_T")
StateFunc = Callable[[_T], Callable]


class LookaheadParser(Generic[_T]):
    def __init__(self):
        self._token_index: int = 0
        self._tokens: list[_T] = []
        self._state: StateFunc

    def _parse(self) -> None:
        while self._token_index < len(self._tokens):
            self._state = self._state(self._tokens[self._token_index])
            self._token_index += 1

    def _lookahead(self, offset: int) -> Optional[_T]:
        prospective_index = self._token_index + offset
        if prospective_index < len(self._tokens):
            return self._tokens[prospective_index]
        return None

    def _advance(self, offset: int) -> None:
        self._token_index += offset

这是sample_parser.py

from __future__ import annotations
from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from .lookahead_parser import LookaheadParser, StateFunc


class SampleParser(LookaheadParser[str]):
    def __init__(self, raw_text: str):
        self._raw_text = raw_text
        self._state = self._state_first

    def parse(self) -> None:
        self._parse()

    def _state_first(self, token: str) -> StateFunc:
        pass  # stub


if __name__ == "__main__":
    pass  # stub

我正在运行 Python 3.9.6,但它也在 3.9.5 中发生。无论我使用_T还是都会出现问题T,以防您对此有所怀疑。mypy 本身没有报告任何问题。

标签: pythongenericsmypy

解决方案


lookahead_parser仅当TYPE_CHECKING为真时才导入。该变量在运行时为假,因此模块永远不会被加载。只需删除if TYPE_CHECKING:零件并无条件导入即可。

from .lookahead_parser import LookaheadParser, StateFunc

推荐阅读