如何为方法链创建类型提示？[重复]

Question

我正在编写一个包含一个对象的库，该对象允许对其某些方法进行方法链接。我想为这个库的用户提供一个简单的方法来知道是否有任何特定的方法支持链接（并且“用户”包括提供自动完成的IDE）。

根据这个问题和这个问题，通过在支持链接的方法调用结束时html" target="_blank">返回自己可以很容易地完成链接。

然而，这些问题都没有解决如何向用户表明一个方法支持链接(除了明显的docstring注释)。如下面的示例代码所示，我试图添加一个类型提示，但是由于当我试图在类型提示中使用类< code>A时，它还没有完全定义，Python无法实例化该类的一个实例。

class A():

    def chain(self) -> A:  # `self` is of type A, so `-> A` is an appropriate return type
        print("Chaining!")
        return self        # Enable method chaining

>>> A().chain().chain().chain()       # Should print "Chaining!" three times.
Traceback (most recent call last):
  File "scratch.py", line 1, in <module>
    class A():
  File "scratch.py", line 3, in A
    def chain(self) -> A:
NameError: name 'A' is not defined

指示方法支持Python中的链接的有效方法是什么？

理想的答案是使用类型提示或其他允许IDE支持自动完成的构造。

Answer 1

class A():
    def chain(self) -> 'A':  # `self` is of type A, so `-> A` is an appropriate return type
        print("Chaining!")
        return self

A().chain().chain().chain()

在您的代码中，A没有用引号定义，因此它被计算为变量

Answer 2

根据这篇优秀的文章，您可以在类型提示中使用您的类作为字符串。

在这种情况下，您需要按如下方式修改函数签名：

def chain(self) -> "A":