python - 绕过 "typing.Union is not a generic class"

Question

在 Python 的 typing 模块中，似乎任何不是 generic 的东西类型不支持下标:

>>> import typing
>>> from yarl import URL

>>> StrOrURL = typing.Union[URL, str]
>>> typing.List[StrOrURL]                # OK; List is generic
typing.List[typing.Union[yarl.URL, str]]

>>> SetOrDict = typing.Union[set, dict]
>>> SetOrDict[StrOrURL]                  # Raises
TypeError: typing.Union[set, dict] is not a generic class

无论 SetOrDict 是用 typing.Union 还是 typing.TypeVar 定义的，都会出现这种情况。 (至少在 Python 3.7 中。)

看起来有一种方法可以将 Generic 子类化，大概是这样的:

class SetOrDict(typing.Generic[StrOrURL]):
    # ...

但这会立即引发 TypeError: Parameters to Generic[...] must all be type variables。

有没有办法适应上述情况？即，下标用户定义(非通用)typedef 的推荐做法是什么？

Answer 1

您不能使用 Union[...]、别名或其他方式作为通用类型，不。您的 union contents 也不是通用的，您无法使用 dict< 说明 dict 或 set 对象可以包含哪些类型 或 set 直接，你会使用 Set或 Dict .

你有一个新工会:

URLorStrSetOrDict = typing.Union[Set[StrOrUrl], Dict[StrOrUrl, Any]]

请注意，字典有键和值，您需要提供这两者的信息。如果这用作 API 的输入，则考虑使用更通用和不可变的 AbstractSet 和 Mapping 类型；这清楚表明您的 API 只需要读取信息。

就我个人而言，我会认真考虑重构一个代码库，该代码库对于它可以接受的对象种类如此困惑。要求 API 只接受集合并且现有代码必须传入 set(dictionary) 而不是 dictionary 并不是一个艰巨的要求，或者 API 可能真的可以与任何StrOrUrl 的可迭代。