Add tkinter GUI for OrbitIn

src/gui.py

@@ -0,0 +1,415 @@
+#!/usr/bin/env python3
+"""
+码头作业日志管理工具 - GUI 界面
+"""
+import tkinter as tk
+from tkinter import ttk, messagebox, scrolledtext
+import threading
+from datetime import datetime
+import sys
+import os
+
+# 添加项目根目录到 Python 路径
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from src.confluence import ConfluenceClient
+from src.extractor import HTMLTextExtractor
+from src.parser import HandoverLogParser
+from src.database import DailyLogsDatabase
+from src.report import DailyReportGenerator
+
+
+class OrbitInGUI:
+    """码头作业日志管理工具 GUI"""
+    
+    def __init__(self, root):
+        self.root = root
+        self.root.title("码头作业日志管理工具 - OrbitIn")
+        self.root.geometry("800x600")
+        self.root.resizable(True, True)
+        
+        # 设置样式
+        style = ttk.Style()
+        style.theme_use('clam')
+        
+        # 创建主框架
+        main_frame = ttk.Frame(root, padding="10")
+        main_frame.pack(fill=tk.BOTH, expand=True)
+        
+        # 左侧控制面板
+        left_frame = ttk.Frame(main_frame)
+        left_frame.pack(side=tk.LEFT, fill=tk.Y, padx=(0, 10))
+        
+        # 右侧输出区域
+        right_frame = ttk.Frame(main_frame)
+        right_frame.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
+        
+        # === 左侧控制面板 ===
+        
+        # 获取数据按钮
+        btn_fetch = ttk.Button(
+            left_frame, 
+            text="📥 获取并处理数据",
+            command=self.fetch_data,
+            width=20
+        )
+        btn_fetch.pack(pady=5)
+        
+        btn_fetch_debug = ttk.Button(
+            left_frame,
+            text="📥 获取 (Debug模式)",
+            command=self.fetch_debug,
+            width=20
+        )
+        btn_fetch_debug.pack(pady=5)
+        
+        # 分隔线
+        ttk.Separator(left_frame, orient=tk.HORIZONTAL).pack(fill=tk.X, pady=10)
+        
+        # 生成日报
+        ttk.Label(left_frame, text="生成日报:").pack(anchor=tk.W, pady=(10, 5))
+        
+        date_frame = ttk.Frame(left_frame)
+        date_frame.pack(fill=tk.X, pady=5)
+        
+        self.date_var = tk.StringVar(value=datetime.now().strftime('%Y-%m-%d'))
+        date_entry = ttk.Entry(date_frame, textvariable=self.date_var, width=12)
+        date_entry.pack(side=tk.LEFT)
+        
+        btn_report = ttk.Button(
+            left_frame,
+            text="📊 生成日报",
+            command=self.generate_report,
+            width=20
+        )
+        btn_report.pack(pady=5)
+        
+        btn_report_today = ttk.Button(
+            left_frame,
+            text="📊 今日日报",
+            command=self.generate_today_report,
+            width=20
+        )
+        btn_report_today.pack(pady=5)
+        
+        # 分隔线
+        ttk.Separator(left_frame, orient=tk.HORIZONTAL).pack(fill=tk.X, pady=10)
+        
+        # 未统计数据
+        ttk.Label(left_frame, text="添加未统计数据:").pack(anchor=tk.W, pady=(10, 5))
+        
+        unaccounted_frame = ttk.Frame(left_frame)
+        unaccounted_frame.pack(fill=tk.X, pady=5)
+        
+        ttk.Label(unaccounted_frame, text="月份:").pack(side=tk.LEFT)
+        self.month_var = tk.StringVar(value=datetime.now().strftime('%Y-%m'))
+        month_entry = ttk.Entry(unaccounted_frame, textvariable=self.month_var, width=8)
+        month_entry.pack(side=tk.LEFT, padx=(5, 10))
+        
+        ttk.Label(unaccounted_frame, text="TEU:").pack(side=tk.LEFT)
+        self.teu_var = tk.StringVar()
+        teu_entry = ttk.Entry(unaccounted_frame, textvariable=self.teu_var, width=8)
+        teu_entry.pack(side=tk.LEFT, padx=(5, 0))
+        
+        btn_unaccounted = ttk.Button(
+            left_frame,
+            text="➕ 添加",
+            command=self.add_unaccounted,
+            width=20
+        )
+        btn_unaccounted.pack(pady=5)
+        
+        # 分隔线
+        ttk.Separator(left_frame, orient=tk.HORIZONTAL).pack(fill=tk.X, pady=10)
+        
+        # 数据库统计
+        btn_stats = ttk.Button(
+            left_frame,
+            text="📈 数据库统计",
+            command=self.show_stats,
+            width=20
+        )
+        btn_stats.pack(pady=5)
+        
+        # 清空输出按钮
+        btn_clear = ttk.Button(
+            left_frame,
+            text="🗑️ 清空输出",
+            command=self.clear_output,
+            width=20
+        )
+        btn_clear.pack(pady=5)
+        
+        # === 右侧输出区域 ===
+        
+        # 状态标签
+        self.status_var = tk.StringVar(value="就绪")
+        status_label = ttk.Label(right_frame, textvariable=self.status_var)
+        status_label.pack(anchor=tk.W)
+        
+        # 输出文本框
+        self.output_text = scrolledtext.ScrolledText(
+            right_frame, 
+            wrap=tk.WORD,
+            font=('Consolas', 10),
+            bg='#f5f5f5'
+        )
+        self.output_text.pack(fill=tk.BOTH, expand=True, pady=(5, 0))
+        
+        # 绑定快捷键
+        self.root.bind('<Control-Return>', lambda e: self.fetch_data())
+        
+        # 初始消息
+        self.log_message("码头作业日志管理工具 - OrbitIn")
+        self.log_message("=" * 50)
+        self.log_message("日期格式: YYYY-MM-DD")
+        self.log_message("月份格式: YYYY-MM")
+        self.log_message("")
+        self.log_message("按 Ctrl+Enter 快速获取数据")
+    
+    def log_message(self, message, is_error=False):
+        """输出日志消息"""
+        timestamp = datetime.now().strftime('%H:%M:%S')
+        prefix = "❌" if is_error else "📝"
+        self.output_text.insert(tk.END, f"[{timestamp}] {prefix} {message}\n")
+        self.output_text.see(tk.END)
+        self.root.update()
+    
+    def clear_output(self):
+        """清空输出"""
+        self.output_text.delete(1.0, tk.END)
+        self.log_message("输出已清空")
+    
+    def set_status(self, status):
+        """设置状态"""
+        self.status_var.set(status)
+        self.root.update()
+    
+    def run_in_thread(self, func, *args, **kwargs):
+        """在后台线程中运行函数"""
+        def worker():
+            try:
+                func(*args, **kwargs)
+            except Exception as e:
+                self.root.after(0, lambda: self.log_message(str(e), is_error=True))
+                self.root.after(0, lambda: self.set_status("错误"))
+            finally:
+                self.root.after(0, lambda: self.set_status("就绪"))
+        
+        thread = threading.Thread(target=worker, daemon=True)
+        thread.start()
+    
+    def fetch_data(self):
+        """获取并处理数据"""
+        self.set_status("正在获取数据...")
+        self.log_message("开始获取数据...")
+        
+        try:
+            # 加载配置
+            from dotenv import load_dotenv
+            load_dotenv()
+            
+            base_url = os.getenv('CONFLUENCE_BASE_URL')
+            token = os.getenv('CONFLUENCE_TOKEN')
+            content_id = os.getenv('CONFLUENCE_CONTENT_ID')
+            
+            if not base_url or not token or not content_id:
+                self.log_message("错误: 未配置 Confluence 信息，请检查 .env 文件", is_error=True)
+                return
+            
+            # 获取 HTML
+            self.log_message("正在从 Confluence 获取 HTML...")
+            client = ConfluenceClient(base_url, token)
+            html = client.get_html(content_id)
+            
+            if not html:
+                self.log_message("错误: 未获取到 HTML 内容", is_error=True)
+                return
+            
+            self.log_message(f"获取成功，共 {len(html)} 字符")
+            
+            # 提取文本
+            self.log_message("正在提取布局文本...")
+            extractor = HTMLTextExtractor()
+            layout_text = extractor.extract(html)
+            self.log_message(f"提取完成，共 {len(layout_text)} 字符")
+            
+            # 解析数据
+            self.log_message("正在解析日志数据...")
+            parser = HandoverLogParser()
+            logs = parser.parse(layout_text)
+            self.log_message(f"解析到 {len(logs)} 条记录")
+            
+            # 保存到数据库
+            if logs:
+                self.log_message("正在保存到数据库...")
+                db = DailyLogsDatabase()
+                count = db.insert_many([log.to_dict() for log in logs])
+                db.close()
+                self.log_message(f"已保存 {count} 条记录")
+                
+                # 显示统计
+                db = DailyLogsDatabase()
+                stats = db.get_stats()
+                db.close()
+                self.log_message(f"数据库总计: {stats['total']} 条记录, {len(stats['ships'])} 艘船")
+            else:
+                self.log_message("未解析到任何记录")
+            
+            self.set_status("完成")
+            
+        except Exception as e:
+            self.log_message(f"错误: {e}", is_error=True)
+            self.set_status("错误")
+    
+    def fetch_debug(self):
+        """Debug模式获取数据"""
+        self.set_status("正在获取 Debug 数据...")
+        self.log_message("使用本地 layout_output.txt 进行 Debug...")
+        
+        try:
+            # 检查本地文件
+            if os.path.exists('layout_output.txt'):
+                filepath = 'layout_output.txt'
+            elif os.path.exists('debug/layout_output.txt'):
+                filepath = 'debug/layout_output.txt'
+            else:
+                self.log_message("错误: 未找到 layout_output.txt 文件", is_error=True)
+                return
+            
+            self.log_message(f"使用文件: {filepath}")
+            
+            with open(filepath, 'r', encoding='utf-8') as f:
+                text = f.read()
+            
+            self.log_message(f"读取完成，共 {len(text)} 字符")
+            
+            # 解析数据
+            self.log_message("正在解析日志数据...")
+            parser = HandoverLogParser()
+            logs = parser.parse(text)
+            self.log_message(f"解析到 {len(logs)} 条记录")
+            
+            if logs:
+                self.log_message("正在保存到数据库...")
+                db = DailyLogsDatabase()
+                count = db.insert_many([log.to_dict() for log in logs])
+                db.close()
+                self.log_message(f"已保存 {count} 条记录")
+            
+            self.set_status("完成")
+            
+        except Exception as e:
+            self.log_message(f"错误: {e}", is_error=True)
+            self.set_status("错误")
+    
+    def generate_report(self):
+        """生成指定日期的日报"""
+        date = self.date_var.get().strip()
+        
+        if not date:
+            self.log_message("错误: 请输入日期", is_error=True)
+            return
+        
+        try:
+            datetime.strptime(date, '%Y-%m-%d')
+        except ValueError:
+            self.log_message("错误: 日期格式无效，请使用 YYYY-MM-DD", is_error=True)
+            return
+        
+        self.set_status("正在生成日报...")
+        self.log_message(f"生成 {date} 的日报...")
+        
+        try:
+            g = DailyReportGenerator()
+            report = g.generate_report(date)
+            g.close()
+            
+            self.log_message("")
+            self.log_message("=" * 40)
+            self.log_message(report)
+            self.log_message("=" * 40)
+            
+            self.set_status("完成")
+            
+        except Exception as e:
+            self.log_message(f"错误: {e}", is_error=True)
+            self.set_status("错误")
+    
+    def generate_today_report(self):
+        """生成今日日报"""
+        today = datetime.now().strftime('%Y-%m-%d')
+        self.date_var.set(today)
+        self.generate_report()
+    
+    def add_unaccounted(self):
+        """添加未统计数据"""
+        year_month = self.month_var.get().strip()
+        teu = self.teu_var.get().strip()
+        
+        if not year_month or not teu:
+            self.log_message("错误: 请输入月份和 TEU", is_error=True)
+            return
+        
+        try:
+            teu = int(teu)
+        except ValueError:
+            self.log_message("错误: TEU 必须是数字", is_error=True)
+            return
+        
+        self.set_status("正在添加...")
+        self.log_message(f"添加 {year_month} 月未统计数据: {teu}TEU")
+        
+        try:
+            db = DailyLogsDatabase()
+            result = db.insert_unaccounted(year_month, teu, '')
+            db.close()
+            
+            if result:
+                self.log_message("添加成功!")
+            else:
+                self.log_message("添加失败!", is_error=True)
+            
+            self.set_status("完成")
+            
+        except Exception as e:
+            self.log_message(f"错误: {e}", is_error=True)
+            self.set_status("错误")
+    
+    def show_stats(self):
+        """显示数据库统计"""
+        self.set_status("正在统计...")
+        self.log_message("数据库统计信息:")
+        self.log_message("-" * 30)
+        
+        try:
+            db = DailyLogsDatabase()
+            stats = db.get_stats()
+            db.close()
+            
+            self.log_message(f"总记录数: {stats['total']}")
+            self.log_message(f"船次数量: {len(stats['ships'])}")
+            self.log_message(f"日期范围: {stats['date_range']['start']} ~ {stats['date_range']['end']}")
+            
+            if stats['ships']:
+                self.log_message("")
+                self.log_message("船次列表:")
+                for ship in sorted(stats['ships']):
+                    self.log_message(f"  - {ship}")
+            
+            self.set_status("完成")
+            
+        except Exception as e:
+            self.log_message(f"错误: {e}", is_error=True)
+            self.set_status("错误")
+
+
+def main():
+    """主函数"""
+    root = tk.Tk()
+    app = OrbitInGUI(root)
+    root.mainloop()
+
+
+if __name__ == '__main__':
+    main()