如何在运行时创建适用于Pydantic的递归TypeAlias？

静态递归类型别名可以直接通过type关键字定义，但运行时用TypeAliasType结合ForwardRef时，会出现ForwardRef无法自动解析为类型别名本身的问题，导致类型结构中保留未解析的引用。以下是具体解决方案：

核心思路 #

要让递归类型别名在运行时正确生效，需要在创建TypeAliasType后，手动将其中的ForwardRef替换为类型别名本身，或者利用类型解析工具完成引用绑定。

方法一：手动递归解析ForwardRef #

编写解析函数遍历类型结构，将指向自身的ForwardRef替换为已创建的TypeAliasType：

from typing import ForwardRef, TypeAliasType, Union, get_args, get_origin
from collections.abc import Sequence

def resolve_recursive_alias(alias_name: str, alias_value):
    """解析递归类型别名中的ForwardRef引用"""
    def _resolve(type_obj):
        # 替换指向目标别名的ForwardRef
        if isinstance(type_obj, ForwardRef):
            if type_obj.__forward_arg__ == alias_name:
                return ForwardRef(alias_name)
            return type_obj
        # 处理泛型类型（如Sequence）
        origin = get_origin(type_obj)
        if origin is not None:
            resolved_args = tuple(_resolve(arg) for arg in get_args(type_obj))
            return origin[resolved_args]
        return type_obj

    resolved_value = _resolve(alias_value)
    alias = TypeAliasType(alias_name, resolved_value)
    # 二次解析，将占位的ForwardRef替换为别名本身
    final_value = _resolve(resolved_value)
    return TypeAliasType(alias_name, final_value)

# 使用示例
dynamic_types = float | int
ref = ForwardRef("recursive")
initial_type = dynamic_types | Sequence[ref]
recursive = resolve_recursive_alias("recursive", initial_type)

print(recursive.__value__)
# 输出：float | int | collections.abc.Sequence[recursive]

方法二：利用Pydantic的引用解析工具 #

如果使用Pydantic v2，可以直接用pydantic.resolve_refs自动解析类型引用：

from typing import ForwardRef, TypeAliasType
from collections.abc import Sequence
from pydantic import resolve_refs

dynamic_types = float | int
ref = ForwardRef("recursive")
# 创建未解析的类型别名
recursive = TypeAliasType("recursive", dynamic_types | Sequence[ref])
# 传入命名空间完成引用解析
recursive = resolve_refs(recursive, globalns=globals(), localns=locals())

print(recursive.__value__)
# 输出：float | int | collections.abc.Sequence[recursive]

插件扩展场景的实践 #

针对你提到的插件动态扩展Pydantic模型字段类型的场景，可按以下步骤实现：

1. 维护动态允许类型集合，供插件添加新类型：

# 初始允许类型
allowed_types = {int, float}
# 插件添加新类型（示例添加str）
allowed_types.add(str)

2. 生成并解析递归类型别名：

from typing import Union

union_type = Union[tuple(allowed_types)]
ref = ForwardRef("MyTypeAlias")
initial_type = union_type | Sequence[ref]
MyTypeAlias = resolve_recursive_alias("MyTypeAlias", initial_type)

3. 更新Pydantic模型并重建：

from pydantic import BaseModel

class MyModel(BaseModel):
    mylist: Sequence["MyTypeAlias"]

# 重建模型以解析新类型别名
MyModel.model_rebuild(force=True)

# 验证生效
test_data = [1, "hello", [3.14, [True]]]
model = MyModel(mylist=test_data)
print(model)
# 输出：mylist=[1, 'hello', [3.14, [True]]]

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