Python中如何实现静态检查冻结数据类，运行时用普通数据类？

你想要在类型检查阶段使用冻结dataclass，运行时切换为普通dataclass以避免实例化开销，但当前代码中类型检查器报错参数数量过多，且无法捕获属性修改的错误。以下是两种可行的解决方法：

方法一：显式定义类型正确的装饰器函数 #

避免使用partial，直接定义带明确类型签名的装饰器函数，让类型检查器能准确推断其行为：

from dataclasses import dataclass
from typing import TYPE_CHECKING, TypeVar

T = TypeVar('T')

if TYPE_CHECKING:
    def frozen(cls: type[T]) -> type[T]:
        return dataclass(cls, frozen=True)
else:
    frozen = dataclass

@frozen
class Foo:
    x: int
    y: int

foo = Foo(1, 2)  # 类型检查器不再报参数错误
foo.x = 3        # mypy/Pyright会正确报错：无法给冻结类的属性赋值

原理：显式定义的装饰器拥有清晰的类型签名，类型检查器能识别它会将类转换为dataclass（类型检查阶段为冻结状态），因此可以正确处理构造函数参数校验和属性可变性检查。

方法二：使用cast强制标注装饰器类型 #

如果坚持使用partial，可以通过cast告诉类型检查器frozen的类型符合dataclass装饰器的要求：

from dataclasses import dataclass
from typing import TYPE_CHECKING, Callable, TypeVar, cast
from functools import partial

T = TypeVar('T')
# 定义dataclass装饰器的标准类型
DataclassDecorator = Callable[[type[T]], type[T]]

if TYPE_CHECKING:
    frozen = cast(DataclassDecorator, partial(dataclass, frozen=True))
else:
    frozen = dataclass

@frozen
class Foo:
    x: int
    y: int

foo = Foo(1, 2)
foo.x = 3  # 类型检查器会正确捕获修改错误

原理：partial返回的函数类型无法被类型检查器自动识别为合法的dataclass装饰器，通过cast强制标注类型后，检查器会按照dataclass装饰器的逻辑处理类，解决参数错误和可变性检查问题。

原代码出错原因 #

类型检查器无法自动推断partial(dataclass, frozen=True)返回的函数是合法的dataclass装饰器，因此：

• 无法识别装饰后的类会生成带位置参数的构造函数，导致报"参数数量过多"错误
• 无法识别类型检查阶段类是冻结状态，因此不会捕获foo.x = 3的赋值错误

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