如何根据参数指定模块实现类继承并解决MRO与IDE提示问题？

问题背景 #

存在多版本API模块（如api1.py、api2.py），每个模块内的API类接口完全一致（属性、方法名称相同），需实现一个Handler类，根据传入的版本参数动态继承对应版本的API，同时保留IDE的代码提示支持，避免出现方法解析顺序（MRO）冲突。

API版本示例 #

# api1.py
class API:
    version='1.0'
    
    def OnCallback(self, msg):
        pass

# api2.py
class API:
    version='2.0'

    def OnCallback(self, msg):
        pass

原方案存在的问题：传入api1版本时，因_Handler继承了api1.API，动态创建类时又重复继承该类，触发MRO错误；若移除_Handler的父类，则丢失IDE代码提示，增加开发难度。

优化方案 #

方案一：抽象基类做类型提示，动态类仅继承目标API与逻辑类 #

通过定义抽象基类统一API接口，为IDE提供类型提示；自定义逻辑类不依赖具体API，动态创建类时仅继承目标API与逻辑类，彻底避免重复继承。

from importlib import import_module
from abc import ABC, abstractmethod

# 定义抽象基类，为IDE提供API接口提示
class BaseAPI(ABC):
    version: str
    
    @abstractmethod
    def OnCallback(self, msg):
        pass

# 独立的自定义逻辑类，通过类型标注让IDE识别API属性
class _HandlerLogic:
    def __init__(self, account):
        self.account = account

    def OnCallback(self, msg):
        self: BaseAPI  # 类型标注，让IDE识别self包含BaseAPI的属性
        print(self.version, self.account, msg)

def get_handler(version, account):
    api_module = import_module(version)
    # 动态创建Handler类，仅继承目标API和逻辑类
    Handler = type('Handler', (api_module.API, _HandlerLogic), {})
    return Handler(account)

测试验证

# 适配api1版本
handler1 = get_handler('api1', 'account1')
print(handler1.version)  # 输出：1.0
handler1.OnCallback('test1')  # 输出：1.0 account1 test1

# 适配api2版本
handler2 = get_handler('api2', 'account2')
print(handler2.version)  # 输出：2.0
handler2.OnCallback('test2')  # 输出：2.0 account2 test2

方案二：判断目标API，避免重复继承 #

保留_Handler对基准API的继承以获取IDE提示，动态创建类时判断目标API是否与基准API为同一类，避免重复继承导致的MRO冲突。

from importlib import import_module
from api1 import API as BaseAPI

# 继承基准API，保留IDE代码提示
class _Handler(BaseAPI):
    def __init__(self, account):
        self.account = account

    def OnCallback(self, msg):
        print(self.version, self.account, msg)

def get_handler(version, account):
    api_module = import_module(version)
    target_api = api_module.API
    # 若目标API与基准API为同一类，仅继承目标API；否则同时继承目标API与_Handler
    bases = (target_api,) if target_api is BaseAPI else (target_api, _Handler)
    Handler = type('Handler', bases, {})
    return Handler(account)

测试验证

handler1 = get_handler('api1', 'account1')
handler1.OnCallback('test')  # 输出：1.0 account1 test

handler2 = get_handler('api2', 'account2')
handler2.OnCallback('test')  # 输出：2.0 account2 test

方案三：工厂函数内动态定义Handler类 #

直接在工厂函数中定义Handler类，使其继承动态导入的目标API，每个版本对应独立的Handler类，无继承冲突且IDE可正常识别接口。

from importlib import import_module
from typing import Type

def get_handler(version: str, account: str):
    api_module = import_module(version)
    API: Type = api_module.API

    # 动态定义Handler类，直接继承目标API
    class Handler(API):
        def __init__(self, account):
            super().__init__()
            self.account = account

        def OnCallback(self, msg):
            print(self.version, self.account, msg)

    return Handler(account)

测试验证

handler1 = get_handler('api1', 'account1')
handler1.OnCallback('hello')  # 输出：1.0 account1 hello

handler2 = get_handler('api2', 'account2')
handler2.OnCallback('world')  # 输出：2.0 account2 world

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