如何为带多继承的动态类装饰器添加Python类型注解？

问题描述 #

尝试为可构建包含新方法的动态数据类的转换装饰器添加类型注解，计划使用dataclass_transform函数，但不确定返回值的类型提示设计。简化示例代码如下：

import types
from typing import *

class Field:
    """Custom Dataclass Field"""
    ...

class Base:
    """Custom Dataclass base for additional functions"""
    __fields__: List[Field]

    def encode(self) -> bytes:
        ...

    @classmethod
    def decode(cls, data: bytes) -> Self:
        ... 

T = TypeVar('T')

@dataclass_transform()
def transform(newclass: Type[T]) -> ?:
    return types.new_class('test', (newclass, Base), {})

需要确定transform函数返回值的类型注解（即代码中的?部分），希望通过Type、NewType或Generic的组合实现动态多继承的类型提示。

可行方案 #

1. 用交集类型精准标注（Python 3.10+） #

Python 3.10及以上支持交集类型&，可以直接标注返回类型为原类T和Base的交集类型，让类型检查器识别返回类同时具备两者的属性和方法：

from typing import TypeVar, Type, dataclass_transform

T = TypeVar('T')

@dataclass_transform()
def transform(newclass: Type[T]) -> Type[T & Base]:
    return types.new_class('test', (newclass, Base), {})

2. 基于Protocol定义合并接口 #

如果需要兼容更低版本Python，或者想更清晰地定义合并后的类接口，可以用Protocol来描述原类和Base的组合类型：

from typing import Protocol, TypeVar, Type, dataclass_transform

class Field:
    """Custom Dataclass Field"""
    ...

class BaseProtocol(Protocol):
    __fields__: list[Field]
    def encode(self) -> bytes: ...
    @classmethod
    def decode(cls, data: bytes) -> Self: ...

T = TypeVar('T')

class MergedType(T, BaseProtocol, Protocol):
    pass

@dataclass_transform()
def transform(newclass: Type[T]) -> Type[MergedType[T]]:
    return types.new_class('test', (newclass, Base), {})

3. 简化兼容方案 #

如果不需要极致精准的类型标注，结合dataclass_transform的特性，可以直接标注返回Type[T]，同时通过文档注释说明新增的成员：

from typing import TypeVar, Type, dataclass_transform

T = TypeVar('T')

@dataclass_transform()
def transform(newclass: Type[T]) -> Type[T]:
    """将类转换为继承自Base的数据类，新增encode、decode方法及__fields__属性"""
    return types.new_class('test', (newclass, Base), {})

这种方案下，主流类型检查器（如mypy、pyright）会结合dataclass_transform的标识，自动识别新增的成员。

总结 #

优先推荐交集类型方案（Python 3.10+），它最简洁且精准；若需兼容旧版本，Protocol方案是更严谨的选择；简化方案则适合对类型标注精度要求不高的场景，兼顾兼容性和可读性。

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