“with”语句是否支持类型提示？

Question

您可以为使用with语法定义的变量定义类型提示吗？

with example() as x:
    print(x)

我想在上面输入提示说这x是一个str（作为一个例子）。

我发现的唯一解决方法是使用中间变量，但这感觉很糟糕。

with example() as x:
    y: str = x
    print(y)

我在打字文档中找不到示例。

Answer 1

PEP 526 已在 Python 3.6 中实现，允许您注释变量。例如，您可以使用

x: str
with example() as x:
    [...]

或者

with example() as x:
    x: str
    [...]

Answer 2

通常类型注释放置在 API 边界处。在这种情况下，应该从 推断类型example.__enter__。如果该函数未声明任何类型，则解决方案是创建相应的存根文件以帮助类型检查器推断该类型。

具体来说，这意味着创建一个.pyi与从中Example导入的模块具有相同词干的文件。然后可以添加以下代码：

class Example:
    def __enter__(self) -> str: ...
    def __exit__(self, exc_type, exc_value, exc_traceback) -> None: ...