如何通过argparse将Python可调用对象传入脚本执行？

控制台没法直接实例化Python对象传给脚本，给你几个新手友好的实用方案：

方法1：预定义字符串映射到可调用对象 #

适合你只有几种固定的可调用对象需要使用的场景，提前在脚本里把可调用对象和字符串绑定，控制台只传对应的字符串就行。

脚本示例 #

import argparse

# 你需要的可调用类
class TextCallable:
    def __init__(self, file_path):
        self.file_path = file_path
    
    def __call__(self, text):
        # 这里写你的实际逻辑，比如读取文件处理文本
        with open(self.file_path, 'r') as f:
            content = f.read()
        return f"{text} - 文件内容预览：{content[:20]}"

# 预定义字符串和可调用对象的映射
callable_options = {
    "file_reader": TextCallable("some_file.txt"),
    "log_reader": TextCallable("app.log")
}

def main():
    parser = argparse.ArgumentParser(description="处理可调用对象参数的脚本")
    # 让用户从预定义的字符串里选
    parser.add_argument("--text-callable", choices=callable_options.keys(), 
                        required=True, help="选择要使用的可调用对象")
    args = parser.parse_args()
    
    # 根据用户传入的字符串获取对应的可调用对象
    target_callable = callable_options[args.text_callable]
    # 测试调用（替换成你的实际业务逻辑）
    result = target_callable("测试输入文本")
    print(result)

if __name__ == "__main__":
    main()

控制台执行命令 #

python my_script.py --text-callable file_reader

方法2：通过配置文件加载可调用对象参数 #

适合需要灵活配置可调用对象参数的场景，把创建对象需要的信息存在配置文件里，控制台传配置文件路径，脚本读取后实例化对象。

步骤1：创建配置文件 #

比如用JSON格式的config.json：

{
    "callable_class": "TextCallable",
    "args": ["some_file.txt"]
}

步骤2：修改脚本读取配置 #

import argparse
import json

class TextCallable:
    def __init__(self, file_path):
        self.file_path = file_path
    
    def __call__(self, text):
        with open(self.file_path, 'r') as f:
            content = f.read()
        return f"{text} - 文件内容预览：{content[:20]}"

def main():
    parser = argparse.ArgumentParser(description="从配置文件加载可调用对象")
    parser.add_argument("--config", required=True, help="配置文件的路径")
    args = parser.parse_args()
    
    # 读取配置文件
    with open(args.config, 'r') as f:
        config_data = json.load(f)
    
    # 根据配置实例化对象
    if config_data["callable_class"] == "TextCallable":
        target_callable = TextCallable(*config_data["args"])
    else:
        raise ValueError(f"不支持的可调用类：{config_data['callable_class']}")
    
    # 测试调用
    result = target_callable("测试输入文本")
    print(result)

if __name__ == "__main__":
    main()

控制台执行命令 #

python my_script.py --config config.json

方法3：动态加载模块中的可调用对象（谨慎使用） #

如果需要非常灵活的调用（比如从其他模块加载类），可以通过控制台传类的路径和参数，脚本动态导入并实例化。注意：这个方法有安全风险，不要在生产环境或处理未知输入时使用。

脚本示例 #

import argparse
import importlib

class TextCallable:
    def __init__(self, file_path):
        self.file_path = file_path
    
    def __call__(self, text):
        with open(self.file_path, 'r') as f:
            content = f.read()
        return f"{text} - 文件内容预览：{content[:20]}"

def load_target_callable(identifier):
    # 格式规则：[模块名.]类名:参数1,参数2
    if ':' in identifier:
        class_info, args_str = identifier.split(':', 1)
        args = args_str.split(',')
    else:
        class_info = identifier
        args = []
    
    # 拆分模块和类名
    if '.' in class_info:
        module_name, class_name = class_info.rsplit('.', 1)
        module = importlib.import_module(module_name)
        callable_class = getattr(module, class_name)
    else:
        # 类在当前脚本里，直接从全局变量取
        callable_class = globals()[class_info]
    
    return callable_class(*args)

def main():
    parser = argparse.ArgumentParser(description="动态加载可调用对象")
    parser.add_argument("--text-callable", required=True, 
                        help="可调用对象标识，格式：[module.]Class:arg1,arg2")
    args = parser.parse_args()
    
    try:
        target_callable = load_target_callable(args.text_callable)
        result = target_callable("测试输入文本")
        print(result)
    except Exception as e:
        print(f"加载失败：{str(e)}")

if __name__ == "__main__":
    main()

控制台执行命令 #

# 加载当前脚本里的TextCallable，参数是some_file.txt
python my_script.py --text-callable TextCallable:some_file.txt

# 如果TextCallable在其他模块，比如utils.py里
python my_script.py --text-callable utils.TextCallable:some_file.txt

内容的提问来源于stack exchange，提问作者Lukas W.