如何查找psycopg2函数的类型提示？以connect返回值标注为例

以你提到的psycopg2.connect为例，它的返回值类型是psycopg2.extensions.Connection，你的代码可以修改为：

import psycopg2
from psycopg2.extensions import Connection

def sql_connect(sql_config: dict = None) -> Connection:
    db = psycopg2.connect(
        host=sql_config["host"],
        port=sql_config["port"],
        dbname=sql_config["database"]
    )
    return db

下面是几种通用的类型提示查找方法：

• 查阅官方文档：靠谱的第三方包基本都会在文档里明确标注函数的参数、返回类型，直接定位到对应函数的说明页就能找到信息。
• 利用IDE即时提示：PyCharm、VS Code这类IDE，把鼠标悬停在函数名上，会弹出包含返回类型的签名信息。如果没显示，可能需要安装对应包的类型补充库。
• 安装类型补充包：不少包本身没自带类型提示，但社区会维护types-xxx格式的补充包，比如psycopg2可安装types-psycopg2，安装后IDE就能识别正确类型，也能直接导入对应的类型类。
• 运行时查看类型：临时在代码中添加print(type(psycopg2.connect(...)))，运行后就能看到返回对象的类型。这是运行时类型，静态检查时可能需要微调，但应急场景很实用。
• 查看源码或类型stub文件：如果包是开源的，直接看函数源码找返回对象的类型；或者去包的安装目录找.pyi后缀的类型stub文件，里面有专门的类型定义。

内容的提问来源于stack exchange，提问作者mCs