Python如何添加类相关类型提示 实现参数类与返回实例类型绑定

要实现「传入哪个连接类，就返回对应类实例」的类型推断，同时保留IDE代码补全能力，不需要使用Any，通过**绑定基类的泛型TypeVar配合类类型标注Type**即可实现。

实现逻辑 #

• 定义泛型类型变量，通过bound参数将泛型范围限定为BaseConnection的所有子类，既保证类型安全，又支持所有自定义连接类
• 参数标注为Type[T]：代表传入的参数是类对象本身（而非类的实例）
• 返回值直接标注为泛型T，类型检查器和IDE会根据传入的类自动推断返回实例的具体类型

修正后可直接运行的代码 #

from typing import TypeVar, Type
from abc import abstractmethod


class BaseConnection:
    @abstractmethod
    def connect(self):
        pass


class MysqlConnection(BaseConnection):
    def connect(self):
        # Mysql连接实现逻辑
        pass


class PGsqlConnection(BaseConnection):
    def connect(self):
        # PGsql连接实现逻辑
        pass


class SqlLiteConnection(BaseConnection):
    def connect(self):
        # SqlLite连接实现逻辑
        pass


# 定义绑定到BaseConnection的泛型类型变量
T = TypeVar("T", bound=BaseConnection)


def create_connnection(conn_constructor: Type[T]) -> T:
    return conn_constructor()


if __name__ == '__main__':
    # 修正原代码调用时参数名不匹配的问题：形参为conn_constructor，不是conn
    connection = create_connnection(conn_constructor=SqlLiteConnection)
    # 此处IDE会自动识别connection为SqlLiteConnection类型，可正常补全所有方法
    connection.connect()

额外说明 #

• 兼容所有符合BaseConnection接口的自定义连接类，后续新增连接类不需要修改类型标注
• 如果使用Python 3.9及以上版本，不需要导入typing.Type，直接使用内置的type标注即可，函数签名可写为def create_connnection(conn_constructor: type[T]) -> T:
• 该方案是类型安全的：如果传入非BaseConnection子类的参数，类型检查器会直接抛出错误，避免非法调用

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