orc-order-v2/run.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
OCR订单处理系统 - 主入口
---------------------
提供命令行接口，整合OCR识别、Excel处理和订单合并功能。
"""

import os
import sys
import argparse
from typing import List, Optional

from app.config.settings import ConfigManager
from app.core.utils.log_utils import get_logger, close_all_loggers, set_log_level
from app.services.ocr_service import OCRService
from app.services.order_service import OrderService
from app.services.tobacco_service import TobaccoService

logger = get_logger(__name__)

def parse_args():
    """
    解析命令行参数
    
    Returns:
        解析后的参数
    """
    parser = argparse.ArgumentParser(description='OCR订单处理系统')
    
    # 通用选项
    parser.add_argument('--config', type=str, help='配置文件路径')
    parser.add_argument('--log-level', type=str, choices=['debug', 'info', 'warning', 'error', 'critical'], help='日志级别')
    
    # 子命令
    subparsers = parser.add_subparsers(dest='command', help='子命令')
    
    # OCR识别命令
    ocr_parser = subparsers.add_parser('ocr', help='OCR识别')
    ocr_parser.add_argument('--input', type=str, help='输入图片路径')
    ocr_parser.add_argument('--batch', action='store_true', help='批量处理')
    ocr_parser.add_argument('--batch-size', type=int, default=5, help='批处理大小')
    ocr_parser.add_argument('--max-workers', type=int, default=4, help='最大线程数')
    
    # Excel处理命令
    excel_parser = subparsers.add_parser('excel', help='Excel处理')
    excel_parser.add_argument('--input', type=str, help='输入Excel文件路径')
    
    # 合并命令
    merge_parser = subparsers.add_parser('merge', help='合并采购单')
    merge_parser.add_argument('--input', type=str, help='输入采购单文件路径（逗号分隔）')
    
    # 完整流程命令
    pipeline_parser = subparsers.add_parser('pipeline', help='完整处理流程')
    pipeline_parser.add_argument('--input', type=str, help='输入图片路径')
    pipeline_parser.add_argument('--merge', action='store_true', help='是否合并采购单')
    
    # 烟草订单处理
    tobacco_parser = subparsers.add_parser('tobacco', help='处理烟草订单')
    tobacco_parser.add_argument('--input', type=str, help='输入订单明细文件路径')
    
    # 解析参数
    parsed_args = parser.parse_args()
    
    return parsed_args

def main():
    """
    主函数入口
    
    Returns:
        退出码
    """
    # 解析命令行参数
    args = parse_args()
    
    if not args.command:
        argparse.ArgumentParser().print_help()
        return 1
    
    # 加载配置
    config_path = args.config
    config_manager = ConfigManager(config_path)
    config = config_manager.config
    
    # 设置日志级别
    log_level = getattr(args, 'log_level', None)
    if log_level:
        set_log_level(log_level)
    
    try:
        if args.command == 'ocr':
            # OCR识别处理
            ocr_service = OCRService(config)
            if args.batch:
                # 批量处理
                total, success = ocr_service.batch_process(
                    batch_size=args.batch_size, 
                    max_workers=args.max_workers
                )
                return 0 if success > 0 else 1
            else:
                # 处理单个文件
                result = ocr_service.process_image(args.input)
                return 0 if result else 1
                
        elif args.command == 'excel':
            # Excel处理
            order_service = OrderService(config)
            if args.input:
                # 处理指定文件
                result = order_service.process_excel(args.input)
            else:
                # 处理最新文件
                result = order_service.process_excel()
            return 0 if result else 1
            
        elif args.command == 'merge':
            # 合并采购单
            order_service = OrderService(config)
            if args.input:
                # 合并指定文件
                file_list = args.input.split(',')
                result = order_service.merge_purchase_orders(file_list)
            else:
                # 合并所有采购单
                result = order_service.merge_all_purchase_orders()
            return 0 if result else 1
            
        elif args.command == 'pipeline':
            # 完整流程
            ocr_service = OCRService(config)
            order_service = OrderService(config)
            
            # 1. OCR处理
            if args.input:
                # 处理单个文件
                excel_file = ocr_service.process_image(args.input)
            else:
                # 批量处理
                total, success = ocr_service.batch_process()
                if total == 0:
                    logger.warning("没有找到需要处理的图片")
                elif success == 0:
                    logger.warning("OCR处理没有成功处理任何新文件")
                excel_file = None  # 批量处理不返回具体文件
            
            # 2. Excel处理
            if excel_file:
                # 处理指定的Excel文件
                result = order_service.process_excel(excel_file)
            else:
                # 处理最新的Excel文件
                result = order_service.process_excel()
                
            if not result:
                logger.error("Excel处理失败")
                return 1
                
            # 3. 合并采购单（可选）
            if args.merge:
                result = order_service.merge_all_purchase_orders()
                if not result:
                    logger.warning("合并采购单失败")
                    # 不影响整体流程，继续执行
            
            return 0
            
        elif args.command == 'tobacco':
            # 烟草订单处理
            tobacco_service = TobaccoService(config)
            if args.input:
                # 处理指定文件
                logger.info(f"开始处理烟草订单，输入文件: {args.input}")
                result = tobacco_service.process_tobacco_order(args.input)
            else:
                # 处理最新文件
                logger.info("开始烟草公司订单处理")
                result = tobacco_service.process_tobacco_order()
            
            # 检查结果是否为None
            if result is None:
                logger.error("烟草订单处理失败")
                return 1
            else:
                logger.info(f"烟草订单处理成功，输出文件: {result}")
                # 确保result是绝对路径
                if not os.path.isabs(result):
                    result = os.path.abspath(result)
                logger.info(f"烟草订单处理完成，绝对路径: {result}")
                return 0
            
        else:
            logger.error(f"未知命令: {args.command}")
            return 1
    
    except Exception as e:
        logger.error(f"执行过程中发生错误: {e}", exc_info=True)
        return 1
    finally:
        # 关闭所有日志记录器
        close_all_loggers()

if __name__ == "__main__":
    sys.exit(main())