413 lines
15 KiB
Python
413 lines
15 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
OCR订单处理系统启动器
|
||
-----------------
|
||
提供简单的图形界面，方便用户选择功能
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import time
|
||
import subprocess
|
||
import shutil
|
||
import tkinter as tk
|
||
from tkinter import messagebox, filedialog, scrolledtext
|
||
from threading import Thread
|
||
import datetime
|
||
|
||
def ensure_directories():
|
||
"""确保必要的目录结构存在"""
|
||
directories = ["data/input", "data/output", "data/temp", "logs"]
|
||
for directory in directories:
|
||
if not os.path.exists(directory):
|
||
os.makedirs(directory, exist_ok=True)
|
||
print(f"创建目录: {directory}")
|
||
|
||
class LogRedirector:
|
||
"""日志重定向器，用于捕获命令输出并显示到界面"""
|
||
def __init__(self, text_widget):
|
||
self.text_widget = text_widget
|
||
self.buffer = ""
|
||
|
||
def write(self, string):
|
||
self.buffer += string
|
||
# 在UI线程中更新文本控件
|
||
self.text_widget.after(0, self.update_text_widget)
|
||
|
||
def update_text_widget(self):
|
||
self.text_widget.configure(state=tk.NORMAL)
|
||
self.text_widget.insert(tk.END, self.buffer)
|
||
# 自动滚动到底部
|
||
self.text_widget.see(tk.END)
|
||
self.text_widget.configure(state=tk.DISABLED)
|
||
self.buffer = ""
|
||
|
||
def flush(self):
|
||
pass
|
||
|
||
def run_command_with_logging(command, log_widget):
|
||
"""运行命令并将输出重定向到日志窗口"""
|
||
def run_in_thread():
|
||
# 记录命令开始执行的时间
|
||
start_time = datetime.datetime.now()
|
||
log_widget.configure(state=tk.NORMAL)
|
||
log_widget.delete(1.0, tk.END) # 清空之前的日志
|
||
log_widget.insert(tk.END, f"执行命令: {' '.join(command)}\n")
|
||
log_widget.insert(tk.END, f"开始时间: {start_time.strftime('%Y-%m-%d %H:%M:%S')}\n")
|
||
log_widget.insert(tk.END, "=" * 50 + "\n\n")
|
||
log_widget.configure(state=tk.DISABLED)
|
||
|
||
# 设置环境变量，强制OCR模块输出到data目录
|
||
env = os.environ.copy()
|
||
env["OCR_OUTPUT_DIR"] = os.path.abspath("data/output")
|
||
env["OCR_INPUT_DIR"] = os.path.abspath("data/input")
|
||
env["OCR_LOG_LEVEL"] = "DEBUG" # 设置更详细的日志级别
|
||
|
||
try:
|
||
# 运行命令并捕获输出
|
||
process = subprocess.Popen(
|
||
command,
|
||
stdout=subprocess.PIPE,
|
||
stderr=subprocess.STDOUT,
|
||
text=True,
|
||
bufsize=1,
|
||
universal_newlines=True,
|
||
env=env
|
||
)
|
||
|
||
# 读取并显示输出
|
||
for line in process.stdout:
|
||
log_widget.after(0, lambda l=line: add_to_log(log_widget, l))
|
||
|
||
# 等待进程结束
|
||
process.wait()
|
||
|
||
# 记录命令结束时间
|
||
end_time = datetime.datetime.now()
|
||
duration = end_time - start_time
|
||
|
||
log_widget.after(0, lambda: add_to_log(
|
||
log_widget,
|
||
f"\n{'=' * 50}\n执行完毕！返回码: {process.returncode}\n"
|
||
f"结束时间: {end_time.strftime('%Y-%m-%d %H:%M:%S')}\n"
|
||
f"耗时: {duration.total_seconds():.2f} 秒\n"
|
||
))
|
||
|
||
# 如果处理成功，显示成功信息
|
||
if process.returncode == 0:
|
||
log_widget.after(0, lambda: messagebox.showinfo("操作成功", "处理完成！\n请在data/output目录查看结果。"))
|
||
else:
|
||
log_widget.after(0, lambda: messagebox.showerror("操作失败", f"处理失败，返回码：{process.returncode}"))
|
||
|
||
except Exception as e:
|
||
log_widget.after(0, lambda: add_to_log(log_widget, f"\n执行出错: {str(e)}\n"))
|
||
log_widget.after(0, lambda: messagebox.showerror("执行错误", f"执行命令时出错: {str(e)}"))
|
||
|
||
# 在新线程中运行，避免UI阻塞
|
||
Thread(target=run_in_thread).start()
|
||
|
||
def add_to_log(log_widget, text):
|
||
"""向日志窗口添加文本"""
|
||
log_widget.configure(state=tk.NORMAL)
|
||
log_widget.insert(tk.END, text)
|
||
log_widget.see(tk.END) # 自动滚动到底部
|
||
log_widget.configure(state=tk.DISABLED)
|
||
|
||
def select_file(log_widget):
|
||
"""选择图片文件并复制到data/input目录"""
|
||
# 确保目录存在
|
||
ensure_directories()
|
||
|
||
# 获取输入目录的绝对路径
|
||
input_dir = os.path.abspath("data/input")
|
||
|
||
file_path = filedialog.askopenfilename(
|
||
title="选择要处理的图片文件",
|
||
initialdir=input_dir, # 默认打开data/input目录
|
||
filetypes=[("图片文件", "*.jpg *.jpeg *.png *.bmp")]
|
||
)
|
||
|
||
if not file_path:
|
||
return None
|
||
|
||
# 记录选择文件的信息
|
||
add_to_log(log_widget, f"已选择文件: {file_path}\n")
|
||
|
||
# 计算目标路径，始终放在data/input中
|
||
output_path = os.path.join("data/input", os.path.basename(file_path))
|
||
abs_output_path = os.path.abspath(output_path)
|
||
|
||
# 检查是否是同一个文件
|
||
if os.path.normpath(os.path.abspath(file_path)) != os.path.normpath(abs_output_path):
|
||
# 如果是不同的文件，则复制
|
||
try:
|
||
shutil.copy2(file_path, output_path)
|
||
add_to_log(log_widget, f"已复制文件到处理目录: {output_path}\n")
|
||
except Exception as e:
|
||
add_to_log(log_widget, f"复制文件失败: {e}\n")
|
||
messagebox.showerror("错误", f"复制文件失败: {e}")
|
||
return None
|
||
|
||
# 返回绝对路径，确保命令行处理正确
|
||
return abs_output_path
|
||
|
||
def select_excel_file(log_widget):
|
||
"""选择Excel文件并复制到data/output目录"""
|
||
# 确保目录存在
|
||
ensure_directories()
|
||
|
||
# 获取输出目录的绝对路径
|
||
output_dir = os.path.abspath("data/output")
|
||
|
||
file_path = filedialog.askopenfilename(
|
||
title="选择要处理的Excel文件",
|
||
initialdir=output_dir, # 默认打开data/output目录
|
||
filetypes=[("Excel文件", "*.xlsx *.xls")]
|
||
)
|
||
|
||
if not file_path:
|
||
return None
|
||
|
||
# 记录选择文件的信息
|
||
add_to_log(log_widget, f"已选择文件: {file_path}\n")
|
||
|
||
# 计算目标路径，始终放在data/output中
|
||
output_path = os.path.join("data/output", os.path.basename(file_path))
|
||
abs_output_path = os.path.abspath(output_path)
|
||
|
||
# 检查是否是同一个文件
|
||
if os.path.normpath(os.path.abspath(file_path)) != os.path.normpath(abs_output_path):
|
||
# 如果是不同的文件，则复制
|
||
try:
|
||
shutil.copy2(file_path, output_path)
|
||
add_to_log(log_widget, f"已复制文件到处理目录: {output_path}\n")
|
||
except Exception as e:
|
||
add_to_log(log_widget, f"复制文件失败: {e}\n")
|
||
messagebox.showerror("错误", f"复制文件失败: {e}")
|
||
return None
|
||
|
||
# 返回绝对路径，确保命令行处理正确
|
||
return abs_output_path
|
||
|
||
def process_single_image(log_widget):
|
||
"""处理单个图片"""
|
||
file_path = select_file(log_widget)
|
||
if file_path:
|
||
# 确保文件存在
|
||
if os.path.exists(file_path):
|
||
add_to_log(log_widget, f"正在处理图片: {os.path.basename(file_path)}\n")
|
||
# 使用绝对路径，并指定直接输出到data/output
|
||
run_command_with_logging(["python", "run.py", "ocr", "--input", file_path], log_widget)
|
||
else:
|
||
add_to_log(log_widget, f"文件不存在: {file_path}\n")
|
||
messagebox.showerror("错误", f"文件不存在: {file_path}")
|
||
else:
|
||
add_to_log(log_widget, "未选择文件，操作已取消\n")
|
||
|
||
def process_excel_file(log_widget):
|
||
"""处理Excel文件"""
|
||
file_path = select_excel_file(log_widget)
|
||
if file_path:
|
||
# 确保文件存在
|
||
if os.path.exists(file_path):
|
||
add_to_log(log_widget, f"正在处理Excel文件: {os.path.basename(file_path)}\n")
|
||
# 使用绝对路径
|
||
run_command_with_logging(["python", "run.py", "excel", "--input", file_path], log_widget)
|
||
else:
|
||
add_to_log(log_widget, f"文件不存在: {file_path}\n")
|
||
messagebox.showerror("错误", f"文件不存在: {file_path}")
|
||
else:
|
||
# 如果未选择文件，尝试处理最新的Excel
|
||
add_to_log(log_widget, "未选择文件，尝试处理最新的Excel文件\n")
|
||
run_command_with_logging(["python", "run.py", "excel"], log_widget)
|
||
|
||
def organize_project_files(log_widget):
|
||
"""整理项目中的文件到正确位置"""
|
||
# 确保目录存在
|
||
ensure_directories()
|
||
|
||
add_to_log(log_widget, "开始整理项目文件...\n")
|
||
|
||
# 转移根目录文件
|
||
files_moved = 0
|
||
|
||
# 处理日志文件
|
||
log_files = [f for f in os.listdir('.') if f.endswith('.log')]
|
||
for log_file in log_files:
|
||
try:
|
||
src_path = os.path.join('.', log_file)
|
||
dst_path = os.path.join('logs', log_file)
|
||
if not os.path.exists(dst_path) or os.path.getmtime(src_path) > os.path.getmtime(dst_path):
|
||
shutil.copy2(src_path, dst_path)
|
||
add_to_log(log_widget, f"已移动日志文件: {src_path} -> {dst_path}\n")
|
||
files_moved += 1
|
||
except Exception as e:
|
||
add_to_log(log_widget, f"移动日志文件出错: {e}\n")
|
||
|
||
# 处理JSON文件
|
||
json_files = [f for f in os.listdir('.') if f.endswith('.json')]
|
||
for json_file in json_files:
|
||
try:
|
||
src_path = os.path.join('.', json_file)
|
||
dst_path = os.path.join('data', json_file)
|
||
if not os.path.exists(dst_path) or os.path.getmtime(src_path) > os.path.getmtime(dst_path):
|
||
shutil.copy2(src_path, dst_path)
|
||
add_to_log(log_widget, f"已移动记录文件: {src_path} -> {dst_path}\n")
|
||
files_moved += 1
|
||
except Exception as e:
|
||
add_to_log(log_widget, f"移动记录文件出错: {e}\n")
|
||
|
||
# 处理input和output目录
|
||
for old_dir, new_dir in {"input": "data/input", "output": "data/output"}.items():
|
||
if os.path.exists(old_dir) and os.path.isdir(old_dir):
|
||
for file in os.listdir(old_dir):
|
||
src_path = os.path.join(old_dir, file)
|
||
dst_path = os.path.join(new_dir, file)
|
||
try:
|
||
if os.path.isfile(src_path):
|
||
if not os.path.exists(dst_path) or os.path.getmtime(src_path) > os.path.getmtime(dst_path):
|
||
shutil.copy2(src_path, dst_path)
|
||
add_to_log(log_widget, f"已转移文件: {src_path} -> {dst_path}\n")
|
||
files_moved += 1
|
||
except Exception as e:
|
||
add_to_log(log_widget, f"移动文件出错: {e}\n")
|
||
|
||
# 显示结果
|
||
if files_moved > 0:
|
||
add_to_log(log_widget, f"整理完成，共整理 {files_moved} 个文件\n")
|
||
messagebox.showinfo("整理完成", f"已整理 {files_moved} 个文件到正确位置。\n"
|
||
"原始文件保留在原位置，以确保数据安全。")
|
||
else:
|
||
add_to_log(log_widget, "没有需要整理的文件\n")
|
||
messagebox.showinfo("整理完成", "没有需要整理的文件。")
|
||
|
||
def main():
|
||
"""主函数"""
|
||
# 确保必要的目录结构存在并转移旧目录内容
|
||
ensure_directories()
|
||
|
||
# 创建窗口
|
||
root = tk.Tk()
|
||
root.title("OCR订单处理系统 v2.0")
|
||
root.geometry("800x600") # 增加窗口宽度以容纳日志
|
||
|
||
# 创建主区域分割
|
||
main_pane = tk.PanedWindow(root, orient=tk.HORIZONTAL)
|
||
main_pane.pack(fill=tk.BOTH, expand=1, padx=5, pady=5)
|
||
|
||
# 左侧操作区域
|
||
left_frame = tk.Frame(main_pane, width=300)
|
||
main_pane.add(left_frame)
|
||
|
||
# 标题
|
||
tk.Label(left_frame, text="OCR订单处理系统", font=("Arial", 16)).pack(pady=10)
|
||
|
||
# 功能按钮区域
|
||
buttons_frame = tk.Frame(left_frame)
|
||
buttons_frame.pack(pady=10, fill=tk.Y)
|
||
|
||
# 创建日志显示区域
|
||
log_frame = tk.Frame(main_pane)
|
||
main_pane.add(log_frame)
|
||
|
||
# 日志标题
|
||
tk.Label(log_frame, text="处理日志", font=("Arial", 12)).pack(pady=5)
|
||
|
||
# 日志文本区域
|
||
log_text = scrolledtext.ScrolledText(log_frame, wrap=tk.WORD, height=30, width=60)
|
||
log_text.pack(fill=tk.BOTH, expand=True, padx=5, pady=5)
|
||
log_text.configure(state=tk.DISABLED) # 设置为只读
|
||
|
||
# 日志初始内容
|
||
add_to_log(log_text, "OCR订单处理系统启动器 v2.0\n")
|
||
add_to_log(log_text, f"当前工作目录: {os.getcwd()}\n")
|
||
add_to_log(log_text, "系统已准备就绪，请选择要执行的操作。\n")
|
||
|
||
# OCR识别按钮
|
||
tk.Button(
|
||
buttons_frame,
|
||
text="OCR图像识别 (批量)",
|
||
width=20,
|
||
height=2,
|
||
command=lambda: run_command_with_logging(["python", "run.py", "ocr", "--batch"], log_text)
|
||
).pack(pady=5)
|
||
|
||
# 单个图片处理
|
||
tk.Button(
|
||
buttons_frame,
|
||
text="处理单个图片",
|
||
width=20,
|
||
height=2,
|
||
command=lambda: process_single_image(log_text)
|
||
).pack(pady=5)
|
||
|
||
# Excel处理按钮
|
||
tk.Button(
|
||
buttons_frame,
|
||
text="处理Excel文件",
|
||
width=20,
|
||
height=2,
|
||
command=lambda: process_excel_file(log_text)
|
||
).pack(pady=5)
|
||
|
||
# 订单合并按钮
|
||
tk.Button(
|
||
buttons_frame,
|
||
text="合并采购单",
|
||
width=20,
|
||
height=2,
|
||
command=lambda: run_command_with_logging(["python", "run.py", "merge"], log_text)
|
||
).pack(pady=5)
|
||
|
||
# 完整流程按钮
|
||
tk.Button(
|
||
buttons_frame,
|
||
text="完整处理流程",
|
||
width=20,
|
||
height=2,
|
||
command=lambda: run_command_with_logging(["python", "run.py", "pipeline"], log_text)
|
||
).pack(pady=5)
|
||
|
||
# 整理文件按钮
|
||
tk.Button(
|
||
buttons_frame,
|
||
text="整理项目文件",
|
||
width=20,
|
||
height=2,
|
||
command=lambda: organize_project_files(log_text)
|
||
).pack(pady=5)
|
||
|
||
# 打开输入目录
|
||
tk.Button(
|
||
buttons_frame,
|
||
text="打开输入目录",
|
||
width=20,
|
||
command=lambda: os.startfile(os.path.abspath("data/input"))
|
||
).pack(pady=5)
|
||
|
||
# 打开输出目录
|
||
tk.Button(
|
||
buttons_frame,
|
||
text="打开输出目录",
|
||
width=20,
|
||
command=lambda: os.startfile(os.path.abspath("data/output"))
|
||
).pack(pady=5)
|
||
|
||
# 清空日志按钮
|
||
tk.Button(
|
||
buttons_frame,
|
||
text="清空日志",
|
||
width=20,
|
||
command=lambda: log_text.delete(1.0, tk.END)
|
||
).pack(pady=5)
|
||
|
||
# 底部说明
|
||
tk.Label(left_frame, text="© 2025 OCR订单处理系统", font=("Arial", 10)).pack(side=tk.BOTTOM, pady=10)
|
||
|
||
# 启动主循环
|
||
root.mainloop()
|
||
|
||
if __name__ == "__main__":
|
||
main() |