Python 类型提示，输入类型取决于另一个函数的输出类型

Question

考虑以下功能

def f1(a:int, b:int):
    return (a,b)

def f2(x):
    return x[1]

如何自动提示“f2”的参数“x”的类型提示是“f1”的输出类型，即“tuple[int,int]”。

我目前使用以下简单代码

from typing import Tuple

def f1(a:int, b:int):
    return (a,b)

def f2(x:Tuple[int,int]):
    return x[1]

但是，当我修改“f1”时，我还必须修改“f2”。

有更方便的方法吗？

Answer 1

您可以使用类型别名来注释

f1

的返回类型和

f2

的参数类型。

from typing import Tuple

F1Result = Tuple[int, int]

def f1(a: int, b: int) -> F1Result:
    return (a, b)

def f2(x: F1Result):
    return x[1]