如何在argparse执行命令行参数逻辑后自动输出完成提示消息

argparse.add_argument() 仅负责参数的定义、解析、类型校验、默认值赋值环节，没有提供「参数对应业务逻辑执行完成后自动触发动作」的钩子能力，无法直接通过参数配置实现自动输出提示的需求，但可以通过低侵入的代码结构调整实现，不需要在每个判断分支里重复写print语句。

方案1：统一逻辑出口（最简洁，适合当前场景） #

不要把不同参数对应的执行逻辑散落在if/elif分支中，建立参数和处理函数的映射，所有分支执行完成后走统一收尾逻辑打日志，后续新增solver也不需要重复写日志代码。
改造后的示例代码：

import argparse

def compo1_handler():
    print('This is solver version1')

def compo2_handler():
    print('This is solver version2')

# 参数值与处理函数的映射表
solver_handlers = {
    'compo1': compo1_handler,
    'compo2': compo2_handler
}

def main():
    parser = argparse.ArgumentParser() 
    parser.add_argument('-C','--code', type=str, help='Choose solver', required=True)
    args = parser.parse_args()
    
    # 校验参数合法性
    handler = solver_handlers.get(args.code)
    if not handler:
        raise ValueError(f"Invalid solver code: {args.code}")
    # 执行业务逻辑
    handler()
    # 统一出口输出通用提示，所有分支自动生效
    print(f"INFO [solver {args.code}.py] completed success")

if __name__ == '__main__':
    main()

额外说明：你之前调用时写的python3 main.py -code compo1属于不规范写法，单横杠后跟的短参数要求是单个字符，你定义的短参数是-C、长参数是--code，规范调用命令应该是：

python3 main.py -C compo1
# 或者
python3 main.py --code compo1

之前能正常运行是argparse的前缀兼容特性在生效，参数多的时候很容易出现解析冲突，不建议这么用。

方案2：装饰器解耦（适合逻辑复杂的场景） #

如果后续通用提示逻辑需要扩展（比如加执行耗时统计、异常捕获），可以用装饰器把日志逻辑和业务逻辑完全分离，不需要改动业务函数和主流程代码：

import argparse
from functools import wraps
import time

# 通用收尾装饰器
def solver_finish_log(func):
    @wraps(func)
    def wrapper(solver_name):
        start = time.time()
        try:
            res = func()
            cost = time.time() - start
            print(f"INFO [solver {solver_name}.py] completed success, cost: {cost:.2f}s")
            return res
        except Exception as e:
            print(f"ERROR [solver {solver_name}.py] failed, reason: {str(e)}")
            raise
    return wrapper

# 业务逻辑只需要加装饰器即可，不需要写日志代码
@solver_finish_log
def compo1_handler():
    print('This is solver version1')

@solver_finish_log
def compo2_handler():
    print('This is solver version2')

solver_handlers = {
    'compo1': compo1_handler,
    'compo2': compo2_handler
}

def main():
    parser = argparse.ArgumentParser() 
    parser.add_argument('-C','--code', type=str, help='Choose solver', required=True)
    args = parser.parse_args()
    handler = solver_handlers.get(args.code)
    if not handler:
        raise ValueError(f"Invalid solver code: {args.code}")
    handler(args.code)

if __name__ == '__main__':
    main()

补充说明 #

add_argument里的default参数仅能实现「用户未传该参数时，给参数赋预设值」的效果，和执行完成后的自动动作没有任何关系，无法满足自动输出需求。

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