使用 db-api 进行 Python 类型提示

Question

我想添加 db-api 类型提示，例如:

def test_connect() -> Connection :
...

知道我正在动态加载模块驱动程序(意思是，没有像“pyodbc.Connection”这样的硬编码)，并且 Python 中没有正式的接口(interface)契约。

有什么想法吗？

Answer 1

您可能想要使用 protocols在这里。

简而言之，您在代码库中定义了一个自定义协议(protocol)，其中包含任何“连接”对象必须具有的方法签名。然后，您可以自由返回任何任意对象，只要它包含那些具有指定方法的方法。

最后说明:我知道 mypy 支持协议(protocol)，但我不确定其他类型检查器是否支持。有一个 open PEP正式将协议(protocol)引入 Python 类型生态系统——假设其他类型检查器将在 PEP 被接受后添加对协议(protocol)的支持，如果它们还没有的话。