Python：不子类化时如何为返回类的可变属性做类型提示？

可以用泛型dataclass实现，不需要子类化就能让mypy等类型检查器识别不同函数返回的response具体类型。Python 3.11原生支持泛型dataclass，直接通过typing.Generic和TypeVar来定义即可。

步骤1：定义泛型版的OperationResult #

先导入必要的模块，然后用TypeVar声明一个类型变量，让OperationResult继承Generic并绑定这个类型变量：

from dataclasses import dataclass
from typing import Generic, TypeVar, Optional, Union, Literal

# 声明类型变量，可绑定任意类型（List/Dict/str等）
T = TypeVar('T')

@dataclass
class OperationResult(Generic[T]):
    status_code: int
    response: T
    id: Optional[int] = None

步骤2：在函数中指定具体的response类型 #

不同操作函数返回时，直接通过OperationResult[具体类型]来标注返回值，类型检查器会自动识别response的类型：

# 返回字典类型的response
def fetch_user_details(user_id: int) -> OperationResult[dict[str, str]]:
    return OperationResult(
        status_code=200,
        response={"username": "johndoe", "email": "john@example.com"},
        id=user_id
    )

# 返回列表类型的response
def fetch_user_list() -> OperationResult[list[int]]:
    return OperationResult(
        status_code=200,
        response=[1, 2, 3, 4],
        id=None
    )

# 可能返回字典（成功）或字符串（失败）的函数
def create_user(username: str) -> Union[OperationResult[dict[str, int]], OperationResult[str]]:
    if len(username) < 3:
        return OperationResult(status_code=400, response="用户名长度不能小于3")
    return OperationResult(status_code=201, response={"user_id": 1001})

步骤3：类型窄化处理成功/失败分支 #

如果函数返回的是多种可能的OperationResult，可以通过检查status_code来让类型检查器自动窄化response的类型：

result = create_user("jo")
if result.status_code == 201:
    # mypy会识别这里的response是dict[str, int]，访问键不会报错
    print(f"创建成功，用户ID：{result.response['user_id']}")
else:
    # 这里的response会被识别为str
    print(f"创建失败：{result.response}")

进阶：用Literal绑定状态码和类型（可选） #

如果想让状态码和response类型强关联，可以用Literal限定状态码范围，让类型检查器更精准：

SuccessCode = Literal[200, 201]
ErrorCode = Literal[400, 404, 500]

@dataclass
class OperationResult(Generic[T]):
    status_code: Union[SuccessCode, ErrorCode]
    response: T
    id: Optional[int] = None

def get_user(user_id: int) -> Union[OperationResult[dict[str, str]], OperationResult[str]]:
    if user_id == 1001:
        return OperationResult[dict[str, str]](
            status_code=200,
            response={"username": "johndoe"},
            id=user_id
        )
    return OperationResult[str](status_code=404, response="用户不存在")

这样实现后，类型检查器能准确识别每个函数返回的response具体类型，访问字典键时不会再报错，同时保持了统一的返回类型结构，不需要子类化。

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