如何在Python中根据输入加载同一包的不同版本并卸载？

先修正核心语法问题 #

你示例中的导入写法不符合Python语法，正确格式应为 from 包名 import 成员，或通过动态导入实现更灵活的版本适配。

实现思路 #

1. 动态生成包名：将输入的版本号（如1.0.0）转换为对应包名格式（如package_1_0_0），避免编写大量if-elif分支。
2. 安全动态导入：使用importlib模块加载指定包，替代硬编码导入，扩展性更强。
3. 用完彻底清理：从sys.modules中移除已加载的包（清除Python模块缓存），同时清理当前命名空间的相关变量，确保下一次导入不受旧版本影响。

完整实现代码 #

基础版本

import importlib
import sys

while True:
    version = input().strip()
    package_name = f"package_{version.replace('.', '_')}"
    
    try:
        # 动态导入目标包
        pkg_module = importlib.import_module(package_name)
        # 获取需要的函数和类
        a = getattr(pkg_module, 'a')
        C = getattr(pkg_module, 'C')
        
        # 执行业务逻辑
        c = C()
        a(c)
        
    except ModuleNotFoundError:
        print(f"不存在版本 {version} 对应的包")
    except AttributeError as e:
        print(f"目标包缺少必要成员：{str(e)}")
    finally:
        # 卸载模块，清除缓存
        if package_name in sys.modules:
            del sys.modules[package_name]
        # 清理当前命名空间中的变量
        for var in ['pkg_module', 'a', 'C', 'c']:
            if var in locals():
                del locals()[var]

更简洁的函数封装版本

把版本处理逻辑封装到函数中，函数执行完毕后内部变量会自动回收，无需手动清理局部变量：

import importlib
import sys

def handle_version(version):
    package_name = f"package_{version.replace('.', '_')}"
    try:
        pkg_module = importlib.import_module(package_name)
        a = getattr(pkg_module, 'a')
        C = getattr(pkg_module, 'C')
        
        c = C()
        a(c)
    except ModuleNotFoundError:
        print(f"不存在版本 {version} 对应的包")
    except AttributeError as e:
        print(f"目标包缺少必要成员：{str(e)}")
    finally:
        # 仅需清理模块缓存
        if package_name in sys.modules:
            del sys.modules[package_name]

while True:
    version = input().strip()
    handle_version(version)

关键说明 #

• 动态包名生成：通过version.replace('.', '_')自动转换版本号为合法Python包名，新增版本时无需修改分支判断逻辑。
• importlib的优势：相比exec等动态执行代码的方式，importlib更安全，能有效避免注入风险。
• 模块缓存清理：Python会将已导入的模块缓存到sys.modules中，若不删除，下次导入同包名模块会直接使用缓存，无法加载新版本。

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