feat: 益选 OCR 订单处理系统初始提交
- 智能供应商识别（蓉城易购/烟草/杨碧月/通用） - 百度 OCR 表格识别集成 - 规则引擎（列映射/数据清洗/单位转换/规格推断） - 条码映射管理与云端同步（Gitea REST API） - 云端同步支持：条码映射、供应商配置、商品资料、采购模板 - 拖拽一键处理（图片→OCR→Excel→合并） - 191 个单元测试 - 移除无用的模板管理功能 - 清理 IDE 产物目录 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,207 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
"""文件与目录操作模块"""
|
||||
|
||||
import os
|
||||
import json
|
||||
import tkinter as tk
|
||||
from tkinter import messagebox, filedialog, scrolledtext
|
||||
|
||||
from .logging_ui import add_to_log
|
||||
from .ui_widgets import center_window
|
||||
from app.config.settings import ConfigManager
|
||||
|
||||
|
||||
def select_file(log_widget, file_types=None, title="选择文件"):
|
||||
"""通用文件选择对话框"""
|
||||
if file_types is None:
|
||||
file_types = [("所有文件", "*.*")]
|
||||
file_path = filedialog.askopenfilename(title=title, filetypes=file_types)
|
||||
if file_path:
|
||||
add_to_log(log_widget, f"已选择文件: {file_path}\n", "info")
|
||||
return file_path
|
||||
|
||||
|
||||
def select_excel_file(log_widget):
|
||||
"""选择Excel文件"""
|
||||
return select_file(
|
||||
log_widget,
|
||||
[("Excel文件", "*.xlsx *.xls"), ("所有文件", "*.*")],
|
||||
"选择Excel文件"
|
||||
)
|
||||
|
||||
|
||||
def ensure_directories():
|
||||
"""确保必要的目录结构存在"""
|
||||
config = ConfigManager()
|
||||
directories = [
|
||||
config.get('Paths', 'input_folder', fallback='data/input'),
|
||||
config.get('Paths', 'output_folder', fallback='data/output'),
|
||||
'data/result',
|
||||
config.get('Paths', 'temp_folder', fallback='data/temp'),
|
||||
'logs'
|
||||
]
|
||||
for directory in directories:
|
||||
if not os.path.exists(directory):
|
||||
os.makedirs(directory, exist_ok=True)
|
||||
print(f"创建目录: {directory}")
|
||||
|
||||
|
||||
def clean_cache(log_widget):
|
||||
"""清除处理缓存"""
|
||||
from .command_runner import set_running_task
|
||||
try:
|
||||
config = ConfigManager()
|
||||
processed_record = config.get('Paths', 'processed_record', fallback='data/processed_files.json')
|
||||
output_folder = config.get('Paths', 'output_folder', fallback='data/output')
|
||||
cache_files = [
|
||||
processed_record,
|
||||
os.path.join(output_folder, "processed_files.json"),
|
||||
os.path.join(output_folder, "merged_files.json")
|
||||
]
|
||||
|
||||
for cache_file in cache_files:
|
||||
if os.path.exists(cache_file):
|
||||
os.remove(cache_file)
|
||||
add_to_log(log_widget, f"已清除缓存文件: {cache_file}\n", "success")
|
||||
|
||||
temp_dir = os.path.join("data/temp")
|
||||
if os.path.exists(temp_dir):
|
||||
for file in os.listdir(temp_dir):
|
||||
file_path = os.path.join(temp_dir, file)
|
||||
try:
|
||||
if os.path.isfile(file_path):
|
||||
os.remove(file_path)
|
||||
add_to_log(log_widget, f"已清除临时文件: {file_path}\n", "info")
|
||||
except Exception as e:
|
||||
add_to_log(log_widget, f"清除文件时出错: {file_path}, 错误: {str(e)}\n", "error")
|
||||
|
||||
log_dir = "logs"
|
||||
if os.path.exists(log_dir):
|
||||
for file in os.listdir(log_dir):
|
||||
if file.endswith(".active"):
|
||||
file_path = os.path.join(log_dir, file)
|
||||
try:
|
||||
os.remove(file_path)
|
||||
add_to_log(log_widget, f"已清除活动日志标记: {file_path}\n", "info")
|
||||
except Exception as e:
|
||||
add_to_log(log_widget, f"清除文件时出错: {file_path}, 错误: {str(e)}\n", "error")
|
||||
|
||||
set_running_task(None)
|
||||
|
||||
add_to_log(log_widget, "缓存清除完成，系统将重新处理所有文件\n", "success")
|
||||
messagebox.showinfo("缓存清除", "缓存已清除，系统将重新处理所有文件。")
|
||||
except Exception as e:
|
||||
add_to_log(log_widget, f"清除缓存时出错: {str(e)}\n", "error")
|
||||
messagebox.showerror("错误", f"清除缓存时出错: {str(e)}")
|
||||
|
||||
|
||||
def open_result_directory():
|
||||
try:
|
||||
result_dir = os.path.abspath("data/result")
|
||||
if not os.path.exists(result_dir):
|
||||
os.makedirs(result_dir, exist_ok=True)
|
||||
os.startfile(result_dir)
|
||||
except Exception as e:
|
||||
messagebox.showerror("错误", f"无法打开结果目录: {str(e)}")
|
||||
|
||||
|
||||
def _open_directory_from_settings(settings_key, default_path, label):
|
||||
"""通用的从用户设置读取路径并打开目录"""
|
||||
from .user_settings import load_user_settings
|
||||
try:
|
||||
s = load_user_settings()
|
||||
path = os.path.abspath(s.get(settings_key, default_path))
|
||||
if not os.path.exists(path):
|
||||
os.makedirs(path, exist_ok=True)
|
||||
os.startfile(path)
|
||||
except Exception as e:
|
||||
messagebox.showerror("错误", f"无法打开{label}: {str(e)}")
|
||||
|
||||
|
||||
def open_input_directory_from_settings():
|
||||
_open_directory_from_settings('input_folder', 'data/input', '输入目录')
|
||||
|
||||
|
||||
def open_output_directory_from_settings():
|
||||
_open_directory_from_settings('output_folder', 'data/output', '输出目录')
|
||||
|
||||
|
||||
def open_result_directory_from_settings():
|
||||
_open_directory_from_settings('result_folder', 'data/result', '结果目录')
|
||||
|
||||
|
||||
def clean_data_files(log_widget):
|
||||
"""清理数据文件（仅清理input和output目录）"""
|
||||
try:
|
||||
if not messagebox.askyesno("确认清理", "确定要清理input和output目录的文件吗？这将删除所有输入和输出数据。"):
|
||||
add_to_log(log_widget, "操作已取消\n", "info")
|
||||
return
|
||||
|
||||
files_cleaned = 0
|
||||
|
||||
input_dir = "data/input"
|
||||
if os.path.exists(input_dir):
|
||||
for file in os.listdir(input_dir):
|
||||
file_path = os.path.join(input_dir, file)
|
||||
if os.path.isfile(file_path):
|
||||
os.remove(file_path)
|
||||
files_cleaned += 1
|
||||
add_to_log(log_widget, "已清理input目录\n", "info")
|
||||
|
||||
output_dir = "data/output"
|
||||
if os.path.exists(output_dir):
|
||||
for file in os.listdir(output_dir):
|
||||
file_path = os.path.join(output_dir, file)
|
||||
if os.path.isfile(file_path):
|
||||
os.remove(file_path)
|
||||
files_cleaned += 1
|
||||
add_to_log(log_widget, "已清理output目录\n", "info")
|
||||
|
||||
add_to_log(log_widget, f"清理完成，共清理 {files_cleaned} 个文件\n", "success")
|
||||
messagebox.showinfo("清理完成", f"已成功清理 {files_cleaned} 个文件")
|
||||
except Exception as e:
|
||||
add_to_log(log_widget, f"清理数据文件时出错: {str(e)}\n", "error")
|
||||
messagebox.showerror("错误", f"清理数据文件时出错: {str(e)}")
|
||||
|
||||
|
||||
def clean_result_files(log_widget):
|
||||
try:
|
||||
if not messagebox.askyesno("确认清理", "确定要清理result目录的文件吗？这将删除所有已生成的采购单文件。"):
|
||||
add_to_log(log_widget, "操作已取消\n", "info")
|
||||
return
|
||||
count = 0
|
||||
result_dir = "data/result"
|
||||
if os.path.exists(result_dir):
|
||||
for file in os.listdir(result_dir):
|
||||
file_path = os.path.join(result_dir, file)
|
||||
if os.path.isfile(file_path):
|
||||
os.remove(file_path)
|
||||
count += 1
|
||||
add_to_log(log_widget, f"已清理result目录，共 {count} 个文件\n", "success")
|
||||
messagebox.showinfo("清理完成", f"已清理result目录 {count} 个文件")
|
||||
except Exception as e:
|
||||
add_to_log(log_widget, f"清理result目录时出错: {str(e)}\n", "error")
|
||||
messagebox.showerror("错误", f"清理result目录时出错: {str(e)}")
|
||||
|
||||
|
||||
def validate_unit_price_against_item_data(result_path: str, log_widget=None):
|
||||
try:
|
||||
from app.services.order_service import OrderService
|
||||
service = OrderService()
|
||||
bad_results = service.validate_unit_price(result_path)
|
||||
|
||||
if bad_results:
|
||||
display_count = min(len(bad_results), 10)
|
||||
msg = f"存在{len(bad_results)}条单价与商品资料进货价差异超过1元:\n" + "\n".join(bad_results[:display_count])
|
||||
if len(bad_results) > 10:
|
||||
msg += f"\n...(其余 {len(bad_results) - 10} 条已省略)"
|
||||
messagebox.showwarning("单价校验提示", msg)
|
||||
if log_widget is not None:
|
||||
add_to_log(log_widget, f"单价校验发现{len(bad_results)}条差异>1元\n", "warning")
|
||||
else:
|
||||
if log_widget is not None:
|
||||
add_to_log(log_widget, "单价校验通过（差异<=1元）\n", "success")
|
||||
except Exception as e:
|
||||
if log_widget is not None:
|
||||
add_to_log(log_widget, f"单价校验出错: {str(e)}\n", "error")
|
||||
Reference in New Issue
Block a user