在更新一版,更方便了

app/services/ocr_service.py

@@ -127,4 +127,67 @@ class OCRService:
         Returns:
             图片是否有效
         """
-        return self.ocr_processor.validate_image(image_path) 
+        return self.ocr_processor.validate_image(image_path)
+    
+    def _is_valid_image(self, image_path: str) -> bool:
+        """
+        检查文件是否为有效的图片格式
+        
+        Args:
+            image_path: 图片文件路径
+            
+        Returns:
+            是否为有效图片格式
+        """
+        return self.validate_image(image_path)
+    
+    def _get_excel_path(self, image_path: str) -> str:
+        """
+        根据图片路径生成对应的Excel文件路径
+        
+        Args:
+            image_path: 图片文件路径
+            
+        Returns:
+            Excel文件路径
+        """
+        # 获取文件名（不含扩展名）
+        base_name = os.path.splitext(os.path.basename(image_path))[0]
+        # 生成Excel文件路径
+        output_dir = self.config.get('Paths', 'output_folder', fallback='data/output')
+        excel_path = os.path.join(output_dir, f"{base_name}.xlsx")
+        return excel_path
+    
+    def _generate_excel(self, ocr_result: dict, image_path: str) -> Optional[str]:
+        """
+        根据OCR结果生成Excel文件
+        
+        Args:
+            ocr_result: OCR识别结果
+            image_path: 原始图片路径
+            
+        Returns:
+            生成的Excel文件路径，失败返回None
+        """
+        try:
+            excel_path = self._get_excel_path(image_path)
+            
+            # 确保输出目录存在
+            os.makedirs(os.path.dirname(excel_path), exist_ok=True)
+            
+            # 调用OCR处理器的Excel生成功能
+            if hasattr(self.ocr_processor, 'generate_excel'):
+                success = self.ocr_processor.generate_excel(ocr_result, excel_path)
+                if success:
+                    return excel_path
+            else:
+                # 如果OCR处理器没有generate_excel方法，直接返回路径
+                # 假设OCR处理器已经生成了Excel文件
+                if os.path.exists(excel_path):
+                    return excel_path
+            
+            return None
+            
+        except Exception as e:
+            logger.error(f"生成Excel文件时发生错误: {e}", exc_info=True)
+            return None

app/services/tobacco_service.py

@@ -37,7 +37,10 @@ class TobaccoService:
         # 修复配置获取方式，使用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')
+        # 将烟草订单保存到result目录
+        result_dir = "data/result"
+        os.makedirs(result_dir, exist_ok=True)
+        self.output_file = os.path.join(result_dir, '银豹采购单_烟草公司.xls')
     
     def get_latest_tobacco_order(self) -> Optional[str]:
         """
@@ -261,4 +264,4 @@ class TobaccoService:
         )
         
         # 记录日志
-        logger.info(f"烟草公司订单处理成功，订单时间: {order_time}, 总金额: {total_amount}, 处理条目: {total_count}") 
+        logger.info(f"烟草公司订单处理成功，订单时间: {order_time}, 总金额: {total_amount}, 处理条目: {total_count}")