python - 如何创建其中一种类型作为参数提供的类型组合？

Question

我需要类型组合的简写，其中一种类型作为参数提供。

示例:

class CustomType:
  pass

# Shorthand
  OptionalCustomType = Union[Optional[T], CustomType]

# Usage
  def fun(x: OptionalCustomType[str]) -> str:
    # Type of x should be equivalent to Union[None, CustomType, str]
    if x is None:
      return "None"
    if x is CustomType:
      return "Custom Type"
    return "Some string"

Answer 1

您的代码示例基本上几乎按原样运行。您只需将 T 设为 typevar:

from typing import Optional, Union, TypeVar

class CustomType:
    pass

T = TypeVar('T')
OptionalCustomType = Union[Optional[T], CustomType]

# This type-checks without an issue
def fun(x: OptionalCustomType[str]) -> str:
    # Type of x should be equivalent to Union[None, CustomType, str]
    if x is None:
        return "None"
    if x is CustomType:
        return "Custom Type"
    return "Some string"

y: OptionalCustomType[int]

# In mypy, you'll get the following output:
# Revealed type is 'Union[builtins.int, None, test.CustomType]'
reveal_type(y)

这种特殊技术被称为 generic type aliases .