Mypy typehint可能是一种采用`cls'参数而不是`self'的方法吗？

Question

我很好奇是否有可能让Mypy了解类中的某些方法 cls 参数而不是 self 参数。我想到的具体用例是在键入sqlalchemy hybrid_property 's .expression decorator，但更简单的示例将尝试实现自定义/非构建 - 在 classMethod 的版本中。

将显式类型注释应用于 cls 不起作用，例如

from typing import Type

class Example:
  @my_custom_classmethod
  def foo(cls: Type["Example"]) -> str:
    return "doesn't work"

mypy抱怨删除的自我“ type [ackemens]”不是其类“示例” ，我猜这通常是有道理的，但是在这里没有意义：/

我设法提出的最好的解决方案就是＃＃type：imagore 方法的签名：

class Example:
  @my_custom_classmethod
  def foo(cls: Type["Example"]) -> str: # type: ignore
    return "this actually kind of does work!"

实际上，这实际上确实可以正常工作，因为 cls 在方法的主体中具有正确的类型，并且装饰器接收到正确类型的方法，但感觉有些骇人听闻。

Answer 1

bind_self 是一种检查和分配 self （ cls ）类型的方法。它依赖于 is_classmethod node flag（实现在这里）。

flag is_classmethod 在语义分析期间设置在这里。它是通过直接比较装饰器和 incears.classmethod 来完成的。因此，您的 custom_classmethod 不被视为 classMethod 。

因此，您无法生产将相当于 ClassMethod 内置的装饰器。但是，您可以使用自定义描述符克服此问题：

from __future__ import annotations
from typing import Callable, Concatenate, Generic, ParamSpec, TypeVar, TYPE_CHECKING

_T = TypeVar('_T')
_R = TypeVar('_R')
_P = ParamSpec('_P')


if TYPE_CHECKING:
    base = classmethod
else:
    class base(Generic[_R], classmethod):
        pass
    
class my_classmethod(base[_R], Generic[_P, _R]):
    @property
    def __func__(self) -> Callable[Concatenate[_T, _P], _R]:
        return super().__func__
    def __init__(self: my_classmethod[_P, _R], f: Callable[Concatenate[_T, _P], _R], /) -> None:
        super().__init__(f)
    def __get__(self, obj: _T, type: type[_T] | None = None) -> Callable[_P, _R]:
        return super().__get__(obj, type)
        
    
class A:
    @my_classmethod
    def foo(cls, x: int) -> str:
        return str(x)

reveal_type(A.foo(1))  # N: Revealed type is "builtins.str"
reveal_type(A().foo(1))  # N: Revealed type is "builtins.str"
A.foo('a')  # E: Argument 1 has incompatible type "str"; expected "int"

这在运行时起作用并通过类型检查。描述符很有趣。我假设 python＆gt; = 3.10 ，最近的足够 mypy ：较旧的Python版本可能需要一些更改（至少从 typing_extensions 导入），旧的 mypy 在完整的描述符支持的情况下可能会遇到麻烦。是的，您可以从 ClassMethod 继承。如果您不想要，则必须重新实施 __获取__ 和 __ INIT __ 无论如何（以及 __ func __ 属性，如果您喜欢一致性）。请记住，在3​​.10及以后的 classMethod 还具有包装属性。

这是