新增条码映射编辑功能图形化界面

2025-05-10 11:39:11 +08:00
parent 7b7d491663
commit 5c0b709528
46 changed files with 2510 additions and 499 deletions
@@ -75,6 +75,21 @@ class OCRService:
        logger.info(f"OCRService开始批量处理图片, batch_size={batch_size}, max_workers={max_workers}")
        return self.ocr_processor.process_images_batch(batch_size, max_workers)
    
+    # 添加batch_process作为process_images_batch的别名，确保兼容性
+    def batch_process(self, batch_size: int = None, max_workers: int = None) -> Tuple[int, int]:
+        """
+        批量处理图片（别名方法，与process_images_batch功能相同）
+        
+        Args:
+            batch_size: 批处理大小
+            max_workers: 最大线程数
+            
+        Returns:
+            (总处理数, 成功处理数)元组
+        """
+        logger.info(f"OCRService.batch_process被调用，转发到process_images_batch")
+        return self.process_images_batch(batch_size, max_workers)
+    
    def validate_image(self, image_path: str) -> bool:
        """
        验证图片是否有效
@@ -0,0 +1,255 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+烟草公司订单处理服务
+----------------
+处理烟草公司特定格式的订单明细文件，生成银豹采购单
+"""
+
+import os
+import glob
+import datetime
+import pandas as pd
+import xlrd
+import xlwt
+import re
+from xlutils.copy import copy
+from openpyxl import load_workbook
+from typing import Optional, Dict, Any, List, Tuple
+from app.core.utils.log_utils import get_logger
+from app.core.utils.dialog_utils import show_custom_dialog  # 导入自定义弹窗工具
+from ..config.settings import ConfigManager
+
+logger = get_logger(__name__)
+
+class TobaccoService:
+    """烟草公司订单处理服务"""
+    
+    def __init__(self, config: Dict[str, Any]):
+        """
+        初始化服务
+        
+        Args:
+            config: 配置信息
+        """
+        self.config = config
+        # 修复配置获取方式，使用fallback机制
+        self.output_dir = config.get('Paths', 'output_folder', fallback='data/output')
+        self.template_file = config.get('Paths', 'template_file', fallback='templates/银豹-采购单模板.xls')
+        self.output_file = os.path.join(self.output_dir, '银豹采购单_烟草公司.xls')
+    
+    def get_latest_tobacco_order(self) -> Optional[str]:
+        """
+        获取最新的烟草订单明细文件
+        
+        Returns:
+            文件路径或None
+        """
+        # 获取今日开始时间戳
+        today = datetime.date.today()
+        today_start = datetime.datetime.combine(today, datetime.time.min).timestamp()
+        
+        # 查找订单明细文件
+        file_pattern = os.path.join(self.output_dir, "订单明细*.xlsx")
+        candidates = glob.glob(file_pattern)
+        
+        if not candidates:
+            logger.warning("未找到烟草公司订单明细文件")
+            return None
+        
+        # 按创建时间排序
+        candidates.sort(key=os.path.getctime, reverse=True)
+        latest_file = candidates[0]
+        
+        # 检查是否是今天的文件
+        if os.path.getctime(latest_file) >= today_start:
+            logger.info(f"找到最新烟草订单明细文件: {latest_file}")
+            return latest_file
+        else:
+            logger.warning(f"找到的烟草订单明细文件不是今天创建的: {latest_file}")
+            return latest_file  # 仍然返回最新文件，但给出警告
+    
+    def process_tobacco_order(self, input_file=None):
+        """
+        处理烟草订单
+        
+        Args:
+            input_file: 输入文件路径，如果为None则自动查找最新文件
+            
+        Returns:
+            输出文件路径或None（如果处理失败）
+        """
+        try:
+            # 如果没有指定输入文件，查找最新的文件
+            if input_file is None:
+                input_file = self.get_latest_tobacco_order()
+                
+            if input_file is None:
+                logger.warning("未找到烟草公司订单明细文件")
+                logger.error("未找到可处理的烟草订单明细文件")
+                return None
+                
+            logger.info(f"开始处理烟草公司订单: {input_file}")
+            
+            # 读取订单时间和总金额
+            order_info = self._read_order_info(input_file)
+            if not order_info:
+                logger.error(f"读取订单信息失败: {input_file}")
+                return None
+                
+            order_time, total_amount = order_info
+            
+            # 读取订单数据
+            order_data = self._read_order_data(input_file)
+            if not order_data:
+                logger.error(f"读取订单数据失败: {input_file}")
+                return None
+                
+            # 生成银豹采购单
+            output_file = self._generate_pospal_order(order_data, order_time)
+            if not output_file:
+                logger.error("生成银豹采购单失败")
+                return None
+                
+            # 获取处理条目数
+            total_count = len(order_data)
+            
+            # 输出处理结果
+            logger.info(f"烟草公司订单处理成功，订单时间: {order_time}, 总金额: {total_amount}, 处理条目: {total_count}")
+            logger.info(f"采购单已生成: {output_file}")
+            
+            # 显示处理结果对话框
+            self.show_result_dialog(output_file, order_time, total_count, total_amount)
+            
+            return output_file
+            
+        except Exception as e:
+            logger.error(f"处理烟草公司订单时发生错误: {e}", exc_info=True)
+            return None
+    
+    def _read_order_info(self, file_path: str) -> Optional[Tuple[str, float]]:
+        """
+        读取订单信息（时间和总金额）
+        
+        Args:
+            file_path: 文件路径
+            
+        Returns:
+            包含订单时间和总金额的元组或None
+        """
+        try:
+            wb_info = load_workbook(file_path, data_only=True)
+            ws_info = wb_info.active
+            order_time = ws_info["H1"].value or "（空）"
+            total_amount = ws_info["H3"].value or 0
+            
+            return (order_time, total_amount)
+        except Exception as e:
+            logger.error(f"读取订单信息出错: {e}")
+            return None
+    
+    def _read_order_data(self, file_path: str) -> Optional[pd.DataFrame]:
+        """
+        读取订单数据
+        
+        Args:
+            file_path: 文件路径
+            
+        Returns:
+            订单数据DataFrame或None
+        """
+        columns = ['商品', '盒码', '条码', '建议零售价', '批发价', '需求量', '订单量', '金额']
+        
+        try:
+            # 读取Excel文件
+            df_old = pd.read_excel(file_path, header=None, skiprows=3, names=columns)
+            
+            # 过滤订单量不为0的数据，并计算采购量和单价
+            df_filtered = df_old[df_old['订单量'] != 0].copy()
+            df_filtered['采购量'] = df_filtered['订单量'] * 10
+            df_filtered['采购单价'] = df_filtered['金额'] / df_filtered['采购量']
+            df_filtered = df_filtered.reset_index(drop=True)
+            
+            return df_filtered
+        except Exception as e:
+            logger.error(f"读取订单数据失败: {e}")
+            return None
+    
+    def _generate_pospal_order(self, order_data: pd.DataFrame, order_time: str) -> Optional[str]:
+        """
+        生成银豹采购单
+        
+        Args:
+            order_data: 订单数据
+            order_time: 订单时间
+            
+        Returns:
+            输出文件路径或None
+        """
+        try:
+            # 检查模板文件是否存在
+            if not os.path.exists(self.template_file):
+                logger.error(f"采购单模板文件不存在: {self.template_file}")
+                return None
+            
+            # 打开模板，准备写入
+            template_rd = xlrd.open_workbook(self.template_file, formatting_info=True)
+            template_wb = copy(template_rd)
+            template_ws = template_wb.get_sheet(0)
+            
+            # 获取模板中的表头列索引
+            header_row = template_rd.sheet_by_index(0).row_values(0)
+            barcode_col = header_row.index("条码（必填）")
+            amount_col = header_row.index("采购量（必填）")
+            gift_col = header_row.index("赠送量")
+            price_col = header_row.index("采购单价（必填）")
+            
+            # 写入数据到模板
+            for i, row in order_data.iterrows():
+                template_ws.write(i + 1, barcode_col, row['盒码'])  # 商品条码
+                template_ws.write(i + 1, amount_col, int(row['采购量']))  # 采购量
+                template_ws.write(i + 1, gift_col, "")  # 赠送量为空
+                template_ws.write(i + 1, price_col, round(row['采购单价'], 2))  # 采购单价保留两位小数
+            
+            # 确保输出目录存在
+            os.makedirs(os.path.dirname(self.output_file), exist_ok=True)
+            
+            # 保存输出文件
+            template_wb.save(self.output_file)
+            logger.info(f"采购单生成成功: {self.output_file}")
+            
+            return self.output_file
+        except Exception as e:
+            logger.error(f"生成银豹采购单失败: {e}")
+            return None
+    
+    def show_result_dialog(self, output_file, order_time, total_count, total_amount):
+        """
+        显示处理结果对话框
+        
+        Args:
+            output_file: 输出文件路径
+            order_time: 订单时间
+            total_count: 总处理条目
+            total_amount: 总金额
+        """
+        # 创建附加信息
+        additional_info = {
+            "订单来源": "烟草公司",
+            "处理时间": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        }
+        
+        # 显示自定义对话框
+        show_custom_dialog(
+            title="烟草订单处理结果",
+            message="烟草订单处理完成",
+            result_file=output_file,
+            time_info=order_time,
+            count_info=f"{total_count}个商品",
+            amount_info=f"¥{total_amount:.2f}",
+            additional_info=additional_info
+        )
+        
+        # 记录日志
+        logger.info(f"烟草公司订单处理成功，订单时间: {order_time}, 总金额: {total_amount}, 处理条目: {total_count}")