python - 创建自己的可选版本

Question

我的代码中有这样的内容:

Undefined = Literal['__undefined_attr__']
undefined: Undefined = '__undefined_attr__'

def funkc(
    foo: Union[str, Undefined] = undefined,
    bar: Union[int, Undefined] = undefined,
    baz: Union[str, Undefined] = undefined,
    boo: Union[float, Undefined] = undefined,
    # ... lots more args like that with many different types
):
    if foo is not undefined:
        ...
    if bar is not undefined:
        ...
    ...  # etc.

现在，如果我可以使用 None 作为默认值，这一切都会变得更加简单，例如:

def funkc(
    foo: Optional[str] = None,
    bar: Optional[int] = None,
    baz: Optional[str] = None,
    boo: Optional[float] = None,
    # ... lots more args like that with many different types
):
    if foo is not None:
        ...
    if bar is not None:
        ...
    ...  # etc.

所以我想我可以创建自己的类型快捷方式，可以像这样使用:

def funkc(
    foo: OptionallyDefined[str] = undefined,
    bar: OptionallyDefined[int] = undefined,
    baz: OptionallyDefined[str] = undefined,
    boo: OptionallyDefined[float] = undefined,
    # ... lots more args like that with many different types
):
    if foo is not undefined:
        ...
    if bar is not undefined:
        ...
    ...  # etc.

但实际上创建这个 OptionalylDefined 的东西却让我无法理解。 mypy 可以做到这一点吗？

<小时/>

解决方案:

感谢user2357112 supports Monica's answer ，这就是我最终使用的:

class Undefined:
    instance = None

    def __new__(cls, *args, **kwargs) -> "Undefined":
        """Singleton, just in case..."""
        if not cls.instance:
            cls.instance = super().__new__(cls, *args, **kwargs)  # type: ignore
        return cls.instance


undefined = Undefined()
ArgType = TypeVar("ArgType")
OptionallyDefined = Union[ArgType, Undefined]


def funkc(...):
    if foo is not undefined:
        ...

Answer 1

我这里没有 mypy，所以我无法测试它，但以下应该可以工作。 (确保使用 mypy 对其进行测试，而不仅仅是运行它):

T = typing.TypeVar('T')

OptionallyDefined = typing.Union[T, Undefined]

def funkc(
    foo: OptionallyDefined[str] = undefined,
    ...
):
    ...

您自己的 TypeVar 尝试失败，因为您同时以与 typing.Literal 不兼容的方式更改了 undefined 的定义。

<小时/>

顺便说一句，这种定义和使用 undefined 的方式并不安全。 MyPy 会将任何 '__undefined_attr__' 字符串视为 Undefined 类型的有效值，但您的 is 比较将拒绝某些字符串并允许其他字符串，取决于实现细节以及任何特定字符串的来源。我会编写一个类并使用该类的实例来表示 undefined，而不是使用字符串和 typing.Literal。