如何声明Python中两个JSON库的类型符合PEP 483一致性？

问题场景 #

你正在使用两个第三方JSON库，各自定义了专属的JsonValue类型，且已确认这些类型符合PEP 483的一致性要求。代码场景如下：

from A import create_some_json
from B import consume_some_json

# 类型定义
create_some_json: Callable[[...], A.JsonValue]
consume_some_json: Callable[[B.JsonValue], ...]

你已知可以用typing.cast做类型转换，但面对hypothesis.SearchStrategy这类组合子API时会很繁琐，因此有以下疑问：

1. 是否必须用装饰器实现类型转换，进而污染调用栈？ #

不需要。可以用更简洁且不污染调用栈的方式处理：

• 封装类型断言函数：写一个极简的封装函数，把cast逻辑集中起来，运行时无额外开销：

  from typing import cast
  def to_b_json(json_val: A.JsonValue) -> B.JsonValue:
      return cast(B.JsonValue, json_val)
  

  调用时直接用consume_some_json(to_b_json(create_some_json()))，比重复写cast更清爽。
• 一次性转换Hypothesis策略：针对SearchStrategy，只需对策略本身做一次类型断言，不用在每个组合子调用里重复处理：

  from hypothesis import strategies as st
  from typing import cast, SearchStrategy
  
  a_json_strategy = st.from_regex(r"^.*$", fullmatch=True).map(create_some_json)
  b_json_strategy: SearchStrategy[B.JsonValue] = cast(SearchStrategy[B.JsonValue], a_json_strategy)
  

• 用协议（Protocol）统一类型：定义一个符合JSON结构的协议，让两个库的JsonValue都适配该协议，类型检查器会自动认可兼容性，无需手动转换：

  from typing import Protocol, TypeVar
  
  T = TypeVar("T", bound="JsonValueProtocol")
  
  class JsonValueProtocol(Protocol):
      @classmethod
      def __class_getitem__(cls, item: type[T]) -> type: ...
  
  # 类型检查器会认为A.JsonValue和B.JsonValue都符合该协议
  consume_some_json(create_some_json())  # 无需cast即可通过检查
  

2. JSON容器类型转换时，内部元素的类型会如何变化？ #

由于两个库的JsonValue类型符合PEP 483一致性，它们的容器（比如列表、字典）在运行时本质就是Python原生的list和dict，只是类型标注用了各自的泛型包装。

当你对整个A.JsonValue做cast转换为B.JsonValue时，类型检查器会信任你的断言，自动将嵌套的内部元素也视为B.JsonValue类型——比如A.JsonValue中的list[A.JsonValue]会被推断为list[B.JsonValue]，完全匹配目标库的类型定义。运行时不会有任何实际转换操作，只是类型层面的调整。

3. 对泛型执行cast操作会对其协变性产生什么影响？ #

Python泛型的协变性是通过TypeVar的covariant参数声明的。当你对泛型类型执行cast时，相当于手动绕过了类型系统的协变规则检查：

• 类型检查器会直接接受你指定的目标类型，不会验证原泛型和目标泛型之间的协变关系是否合法。
• 比如如果A.JsonValue是协变的，而B.JsonValue不是，cast后类型检查器不会报错——但这完全基于你对两个库类型一致性的判断，若判断错误，可能引发运行时问题，但类型检查器不会拦截。

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