如何通过argparse导入指定Python文件中的指定函数？

我正在编写一个小型工具函数，它接收Python文件路径以及该文件内待调用的函数名作为输入参数。例如，存在文件src/path/to/file_a.py，其中定义了如下函数：

def foo():
    ...

在工具函数中，我通过以下命令解析参数：

python ./util.py --path src/path/to/file_a.py --function foo

并且需要将foo函数用于util.py中另一库的compile()逻辑：

def compile():
    compiler.compile(
        function=foo,
        # 其他参数
    )

请问通过argparse实现该函数导入的最佳方式是什么？

我有一些初步思路，但效果不佳：

def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument("--path", type=str)
    parser.add_argument("--function", type=str)
    return parser.parse_args()

def compile(path, function):
    import path  # 直接导入路径字符串会报错
    compiler.compile(
        function=path.function,
        # 其他参数
    )

if __name__ == "__main__":
    args = get_args()
    compile(
        path=args.path,
        function=args.function
    )

我还考虑过使用sys.path.append，但不确定如何从中导入foo函数：

def compile(path, function):
    import sys
    sys.path.append(path)

要动态导入指定路径下的Python文件中的函数，可结合os.path处理路径，再使用importlib模块完成动态导入——这是Python官方推荐的动态导入方式，比直接拼接sys.path更可靠。

具体实现步骤：

1. 解析命令行参数，获取文件路径和函数名
2. 将文件路径拆分为目录路径和模块文件名（去掉.py后缀）
3. 将目录路径添加到sys.path，让Python能定位到目标模块
4. 使用importlib.import_module导入模块
5. 通过getattr从模块中提取目标函数

完整代码示例：

import argparse
import sys
import os
import importlib

def get_args():
    parser = argparse.ArgumentParser(description="动态导入指定文件中的函数并编译")
    parser.add_argument("--path", required=True, type=str, help="目标Python文件的路径")
    parser.add_argument("--function", required=True, type=str, help="要调用的函数名")
    return parser.parse_args()

def compile_function(module_path, func_name):
    # 拆分文件路径为目录和模块名
    dir_path, file_name = os.path.split(module_path)
    module_name = os.path.splitext(file_name)[0]
    
    # 将目录添加到sys.path，优先从该目录导入
    if dir_path not in sys.path:
        sys.path.insert(0, dir_path)
    
    # 动态导入模块
    try:
        module = importlib.import_module(module_name)
    except ModuleNotFoundError:
        raise ValueError(f"无法找到模块: {module_name}，路径: {dir_path}")
    
    # 获取目标函数
    try:
        target_func = getattr(module, func_name)
    except AttributeError:
        raise ValueError(f"模块 {module_name} 中不存在函数 {func_name}")
    
    # 执行编译逻辑
    compiler.compile(
        function=target_func,
        # 其他必要参数
    )

if __name__ == "__main__":
    args = get_args()
    compile_function(args.path, args.function)

关键细节说明 #

• 用os.path.split和os.path.splitext拆分路径，避免手动拼接字符串出错
• sys.path.insert(0, dir_path)确保优先从指定目录导入，避免与已有同名模块冲突
• 加入异常处理，可清晰反馈导入失败原因（模块不存在、函数不存在）
• importlib是Python3.1+内置标准库，用于动态导入比__import__更清晰安全

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