一天一个Python知识点——Day 131:Python静态类型检查深度解析

# 变量注解name: str = "Alice"count: int = 0ratio: float = 3.14is_active: bool = True# 容器类型（Python 3.9+ 新语法）from typing import List, Dict, Optional, Union# 传统方式items: List[str] = ["apple", "banana"]scores: Dict[str, float] = {"math": 95.5, "english": 88.0}# Python 3.9+ 简化语法items: list[str] = ["apple", "banana"]scores: dict[str, float] = {"math": 95.5, "english": 88.0}# 可选值middle_name: Optional[str] = None  # 等同于 str | None# 联合类型identifier: Union[int, str] = 100  # 传统identifier: int | str = 100         # Python 3.10+ 新语法

from typing import Tuple, Callable, Anydef process_data(    data: list[float],    threshold: float = 0.5,    callback: Optional[Callable[[float], bool]] = None) -> tuple[list[float], int]:    """    处理数据并返回结果    Args:        data: 待处理的浮点数列表        threshold: 过滤阈值，默认0.5        callback: 可选的回调函数    Returns:        处理后的数据和有效计数    """    filtered = [x for x in data if x > threshold]    if callback:        filtered = [x for x in filtered if callback(x)]    return filtered, len(filtered)# 使用类型别名提高可读性Vector = list[float]Matrix = list[list[float]]def matrix_multiply(a: Matrix, b: Matrix) -> Matrix:    """矩阵乘法"""    return [        [            sum(a[i][k] * b[k][j] for k in range(len(b)))            for j in range(len(b[0]))        ]        for i in range(len(a))    ]

from typing import TypeVar, Generic, Sequencefrom dataclasses import dataclassT = TypeVar('T')  # 任意类型U = TypeVar('U', bound=int | float)  # 受限类型变量class Stack(Generic[T]):    """泛型栈实现"""    def __init__(self) -> None:        self._items: list[T] = []    def push(self, item: T) -> None:        self._items.append(item)    def pop(self) -> T:        return self._items.pop()    def is_empty(self) -> bool:        return len(self._items) == 0# 使用示例string_stack: Stack[str] = Stack()string_stack.push("hello")  # 正确# string_stack.push(123)    # mypy会报错@dataclassclass Pair(Generic[T, U]):    """泛型数据类"""    first: T    second: U# 使用示例pair1: Pair[str, int] = Pair("age", 25)pair2: Pair[int, list[str]] = Pair(1, ["a", "b"])

from typing import Protocol, runtime_checkablefrom abc import abstractmethod# 定义协议（结构子类型）class Renderable(Protocol):    @abstractmethod    def render(self) -> str:        ...    @property    @abstractmethod    def width(self) -> int:        ...    @property    @abstractmethod    def height(self) -> int:        ...# 任何具有render、width、height的对象都符合Renderableclass Button:    def __init__(self, text: str, w: int, h: int):        self.text = text        self._width = w        self._height = h    def render(self) -> str:        return f"<button>{self.text}</button>"    @property    def width(self) -> int:        return self._width    @property    def height(self) -> int:        return self._heightdef render_all(items: list[Renderable]) -> str:    """渲染所有可渲染对象"""    return "\n".join(item.render() for item in items)# 即使Button没有显式继承Renderable，也可以使用button = Button("Click me", 100, 50)print(render_all([button]))  # 类型检查通过

# mypy.ini 或 pyproject.toml[mypy]python_version = 3.11warn_return_any = truewarn_unused_configs = truedisallow_untyped_defs = truedisallow_incomplete_defs = truecheck_untyped_defs = truedisallow_untyped_decorators = trueno_implicit_optional = truewarn_redundant_casts = truewarn_unused_ignores = truewarn_no_return = truewarn_unreachable = true# 严格模式（逐步启用）strict = false  # 开始时设为false，逐步开启# 排除目录exclude = [    'build/',    'dist/',    '\.venv/',    'tests/fixtures/',]# 每模块配置[mypy-tests.*]disallow_untyped_defs = false  # 测试文件可以宽松些

# 基础检查mypy my_project/# 常用选项mypy --strict my_project/                 # 严格模式mypy --ignore-missing-imports .           # 忽略缺失导入mypy --no-implicit-optional .             # 不允许隐式可选mypy --warn-unused-ignores .              # 警告未使用的类型忽略mypy --disallow-incomplete-defs .         # 禁止不完整定义mypy --check-untyped-defs .               # 检查无类型定义的函数体# 输出格式mypy --pretty .                           # 彩色输出mypy --no-error-summary .                 # 不显示错误摘要mypy --any-exprs-report .                 # 显示Any类型使用报告# 集成到开发流程mypy --fast-module-lookup .               # 快速模块查找mypy --cache-dir .mypy_cache .            # 指定缓存目录

# 为没有类型提示的库添加类型存根[mypy-numpy]ignore_missing_imports = false[mypy-pandas.*]ignore_missing_imports = false# 使用类型存根[mypy]mypy_path = stubs/  # 自定义存根目录# 安装社区类型存根# pip install types-requests types-PyYAML types-python-dateutil

# 阶段1：关键模块开始# user_service.py - 首先添加类型def get_user(user_id: int) -> dict[str, Any]:  # 开始时用Any    ...# 阶段2：逐步替换Anyfrom typing import TypedDictclass User(TypedDict):    id: int    name: str    email: strdef get_user(user_id: int) -> User:    ...# 阶段3：启用严格检查# mypy.ini 中设置 disallow_any = true

# test_types.py - 类型测试from typing import get_type_hintsimport pytestdef test_function_signatures():    """验证函数类型提示是否正确"""    from my_module import process_data    hints = get_type_hints(process_data)    assert hints['data'] == list[float]    assert hints['return'] == tuple[list[float], int]    # 检查默认值类型    import inspect    sig = inspect.signature(process_data)    assert isinstance(sig.parameters['threshold'].default, float)# pytest插件辅助# pip install pytest-mypy"""# pytest.ini[pytest]mypy_ini_file = mypy.iniaddopts = --mypy"""

# .github/workflows/type-check.ymlname: Type Checkingon: [push, pull_request]jobs:  type-check:    runs-on: ubuntu-latest    strategy:      matrix:        python-version: ["3.9", "3.10", "3.11"]    steps:    - uses: actions/checkout@v4    - name: Set up Python      uses: actions/setup-python@v4      with:        python-version: ${{ matrix.python-version }}    - name: Install dependencies      run: |        pip install mypy        pip install -e .[dev]        # 安装类型存根        pip install types-requests types-PyYAML || true    - name: Run mypy      run: |        mypy --python-version ${{ matrix.python-version }} \             --config-file mypy.ini \             --html-report mypy-report \             --any-exprs-report \             .    - name: Upload type check report      if: always()      uses: actions/upload-artifact@v3      with:        name: mypy-report-${{ matrix.python-version }}        path: mypy-report/

# 使用cast处理类型断言from typing import cast, Anydef get_config() -> dict[str, Any]:    raw = load_raw_config()    return cast(dict[str, Any], raw)# 类型忽略（谨慎使用）def legacy_code(x):    # type: (int) -> str  # Python 2风格注释    return str(x)def modern_code(x: int) -> str:    result: Any = some_untyped_function(x)    return cast(str, result)  # 或使用 result: str = cast(str, ...)# 使用# type: ignore的准则data = json.loads(raw)  # type: ignore  # 缺少类型存根# 应该添加注释说明原因

# 避免在热路径上使用运行时类型检查from typing import TYPE_CHECKINGif TYPE_CHECKING:    # 这些导入只在类型检查时使用    from expensive_module import HeavyTypeelse:    HeavyType = Any  # 运行时使用轻量级替代def process_data(data: list['HeavyType']) -> list[str]:  # 前向引用    # 实际运行时不导入HeavyType    return [str(item) for item in data]

# 使用pre-commit自动类型检查# .pre-commit-config.yamlrepos:  - repo: https://github.com/pre-commit/mirrors-mypy    rev: v1.5.1    hooks:      - id: mypy        args: [--config-file=mypy.ini]        additional_dependencies:           - types-requests          - types-PyYAML# 与IDE集成"""VS Code: 使用Pylance/PyrightPyCharm: 内置mypy支持设置: 1. 启用类型检查2. 配置mypy路径3. 设置严格程度"""