如何定义自定义的无特殊形式类型

Question

我需要以下的行为：

NoOp = ...  # don't know how to define this

def foo(bar: NoOp[int]) -> int:
    return bar + 1

foo(1)

我希望Mypy像bar完全一样，就像我只是将其注释为bar：int。我还需要它与int本身不同，以便编写代码，以检查是否用Eg noop [str] vs Just str noop >。只是Mypy应该对待他们。

---

我试图与 newtype得到类似的东西，但是由于它将结果定义视为一个子类，因此无法用于我的目的，而我必须使用它的方式也创建了非常非常的代码很难阅读。

Answer 1

如果我正确理解，noop [t]和t在运行时和类型检查器的眼中都是完全相同的类型。唯一的区别是您要编写检查注释的自定义代码，并且能够告诉t除noop [t]外。这是正确的吗？

如果是这样，则可以查看 注释 类型，在Python 3.9中引入，并通过typing_extensions进行了回波。它采用注释的[t，x]的形式，其中t是一种类型注释，x是任意的元数据。它允许您将元数据x与注释t相关联，而不更改类型（IE 注释[T，x]等于<代码> t ）。

对于您的情况，您可以将noop定义为通用类型别名：

T = TypeVar("T")
NoOp = Annotated[T, "no-op"]  # use "no-op" as metadata, but you could just choose any value

def foo(bar: NoOp[int]) -> int:
    return bar + 1

然后，您可以使用 typing.get_type_hints （替换typing用typing_extensions在3.9之前的版本）。注释对象将具有__元数据__属性，该属性容纳了元数据的元组：

>>> typing.get_type_hints(foo)  # by default it strips the metadata
{'x': int, 'return': int}

>>> typing.get_type_hints(foo, include_extras=True)
{'x': typing.Annotated[int, 'no-op'], 'return': int}

>>> x_annotation = typing.get_type_hints(foo, include_extras=True)["x"]
>>> x_annotation.__metadata__
('no-op',)

原因是元组，因为您可以将多个值与之相关联，例如注释[INT，“ A” ，“ B”，1.234]。

Answer 2

有些疑问，因为我不确定我会得到您的用户酶 - 对不起，如果这不是您的需求。

from typing import Generic, TypeVar

T = TypeVar('T')   # you can better specify this if needed

class Wrapper(Generic[T]):
    def __init__(content: T):
    self.content = content


def f(x: Wrapper[int] | int) -> int:
    if isinstance(x, Wrapper): 
        return x.content + 1
    return x + 1

另一方面，如果您想制作包装器以便在f中进行实例检查，则可以做类似的

    def __add__(self, other: Wrapper[T] | T):
        if isinstance(other, Wrapper):
            return Wrapper(self.content + other.content)
        return Wrapper(self.content + other)

操作需要小心，因为您最终会出现一个非交易__添加__（不一定是错误的，但要注意的是什么）。