Python泛型类型提示问题：mypy报错原因及优化方案咨询

1. 报错原因 #

这是代码写法导致的类型推断不精确，并非mypy不支持的特性：

• 尽管你明确知道this_type对应泛型参数T的类，但mypy无法从type[T]精确推断this_type(**asdict(a))的返回类型——asdict(a)返回的是dict[str, Any]，mypy无法验证该字典的键值对完全匹配T的字段；
• 加上你原本用TradeBar | QuoteBar作为泛型上界，mypy会把实例化结果推断为TradeBar | QuoteBar，而非具体的泛型参数T，因此和返回值要求的list[T]类型不匹配。

2. 优化实现方式 #

将BarBase作为泛型类型上界是更合理的选择：TradeBar和QuoteBar都继承自BarBase，这样的约束更符合面向对象设计，后续新增BarBase子类时也能直接复用MyUsefulClass。

同时需要帮助mypy精确推断实例化结果的类型，以下是修改后的代码：

from datetime import datetime
from dataclasses import dataclass, asdict
from typing import TypeVar, cast

@dataclass
class BarBase:
    sym: str
    timestamp: datetime | None

@dataclass
class TradeBar(BarBase):
    open: float
    high: float
    low: float
    close: float

@dataclass
class QuoteBar(BarBase):
    bid: float
    ask: float

# 定义带约束的TypeVar，明确T必须是BarBase的子类
T = TypeVar('T', bound=BarBase)

class MyUsefulClass[T]:
    def __init__(self, this_type: type[T]):
        self.this_type = this_type

    def silly_fn(self, x: T, y: T) -> list[T]:
        # 用cast告知mypy，实例化的结果就是泛型参数T
        return [cast(T, self.this_type(**asdict(a))) for a in (x, y)]

util = MyUsefulClass[QuoteBar](QuoteBar)

util.silly_fn(
    QuoteBar(sym="abc", timestamp=None, bid=1.1, ask=2.2),
    QuoteBar(sym="def", timestamp=None, bid=99.9, ask=100.1),
)

无cast的替代方案 #

如果不想使用cast，可以给BarBase添加类型安全的转换类方法，让mypy直接识别返回类型：

@dataclass
class BarBase:
    sym: str
    timestamp: datetime | None

    @classmethod
    def from_dict(cls: type[T], data: dict[str, Any]) -> T:
        return cls(**data)

之后在silly_fn中调用a.from_dict(asdict(a))，mypy就能自动推断出返回类型为T。

内容的提问来源于stack exchange，提问作者user1276048