import os,sys
from auto_sliceing_operate import main_begin_sliceing
from download_batch_data import main_download_batch_data_and_trans
from auto_sliceing_operate import main_download_zip
if __name__ == '__main__':
    # 根据参数决定执行哪个模块
    # 命令映射表：命令名 -> 处理函数
    command_handlers = {
        'batch_dwonload': main_download_batch_data_and_trans.main,
        'begin_sliceing': main_begin_sliceing.main,
        'download_zip': main_download_zip.main,
    }
    
    # 检查参数数量
    if len(sys.argv) < 2:
        print('Usage: python main.py <command> <work_dir>')
        print('可用命令:', ' | '.join(command_handlers.keys()))
        sys.exit(1)
    
    command = sys.argv[1]
    
    # 检查命令是否存在
    if command not in command_handlers:
        print(f'错误: 未知命令 "{command}"')
        print('可用命令:', ' | '.join(command_handlers.keys()))
        sys.exit(1)
    
    # 检查是否提供了 work_dir 参数
    if len(sys.argv) < 3:
        print(f'Usage: python main.py {command} <work_dir>')
        sys.exit(1)
    
    work_dir = sys.argv[2]
    
    # 执行对应的处理函数
    command_handlers[command](work_dir=work_dir)