Python根据命令行参数数量选择配置文件的实现问题

你的问题出在静态import无法通过变量指定模块路径，Python的import语句是在代码加载时解析的，不能直接用字符串路径导入模块，而且你定义的config_file变量不会自动关联到import操作。下面是具体的解决方法：

错误点分析 #

1. config_file只是字符串路径，Python的import机制不识别这种方式，所以import config会报错找不到模块（因为你没有名为config.py的文件）。
2. main函数里的*args参数没用到，input1 = sys.argv[0]是脚本自身的路径，对配置逻辑没有帮助。
3. 没有把命令行参数传递给config_custom.py的config()函数，无法实现自定义参数的赋值。

解决方案：动态导入模块 + 统一配置入口 #

推荐创建一个统一的配置入口模块（比如config.py），让所有脚本都从这里导入配置，在这个模块里根据命令行参数动态选择加载config_default或config_custom：

步骤1：创建config.py作为统一入口

import sys
import importlib

# 根据命令行参数数量选择配置模块
if len(sys.argv) == 1:
    # 无额外参数，加载默认配置
    _target_module = importlib.import_module("config_default")
elif len(sys.argv) == 3:
    # 有2个自定义参数，加载自定义配置并传递参数
    _target_module = importlib.import_module("config_custom")
    # 假设config_custom的config函数需要接收这两个参数，根据实际函数定义调整
    _target_module.config(sys.argv[1], sys.argv[2])
else:
    print("Error: Check the arguments")
    sys.exit()

# 将目标模块的config函数导出，其他脚本直接调用即可
config = _target_module.config

步骤2：修改config_custom.py的config()函数

确保它能接收命令行传入的参数，示例如下：

# config_custom.py
def config(arg1, arg2):
    # 可将参数保存为全局变量或返回配置字典
    global custom_config
    custom_config = {
        "param1": arg1,
        "param2": arg2
    }
    return custom_config

步骤3：其他脚本导入配置

所有需要用配置的脚本，直接导入统一的config模块即可：

# 其他脚本示例
import config

# 获取配置
current_config = config.config()
print(current_config)

替代方案：在主脚本中动态导入并全局导出 #

如果不想新增config.py，也可以在主脚本中处理动态导入，并把配置模块全局化：

# 主脚本 main.py
import sys
import importlib

def main():
    if len(sys.argv) == 1:
        config_module = importlib.import_module("config_default")
    elif len(sys.argv) == 3:
        config_module = importlib.import_module("config_custom")
        config_module.config(sys.argv[1], sys.argv[2])
    else:
        print("Error: Check the arguments")
        sys.exit()
    
    # 全局导出配置模块，供其他脚本导入
    global config
    config = config_module

if __name__ == "__main__":
    main()

其他脚本导入方式：

from main import config
current_config = config.config()

这样就能实现根据命令行参数数量自动切换配置模块的需求，同时解决ModuleNotFoundError的问题。

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