refactor: extract common tab functionality to BaseTab and add decorators
- Add BaseTab base class for common tab initialization and logging - Add constants module for shared GUI constants - Move CheckboxTreeview to separate widget file - Add admin_only and require_session decorators to utils.py - Refactor DataExtractionTab and MaterialValidationTab to inherit BaseTab - Remove duplicate _update_log method from tab classes Co-Authored-By: Claude (glm-5) <noreply@anthropic.com>
This commit is contained in:
Misaka
@@ -18,7 +18,8 @@ from pathlib import Path
|
||||
from io import StringIO
|
||||
from contextlib import redirect_stdout
|
||||
from typing import List, Dict
|
||||
from gui.widgets import FileSelector, LogText, GuiTextHandler, DeleteProgressWindow
|
||||
from gui.base_tab import BaseTab
|
||||
from gui.widgets import FileSelector, LogText, GuiTextHandler, DeleteProgressWindow, CheckboxTreeview
|
||||
from gui.config_manager import ConfigManager
|
||||
from gui.log_config import setup_gui_logging, get_logger
|
||||
from gui.material_type_management_dialog import MaterialTypeManagementDialog
|
||||
@@ -27,232 +28,7 @@ import pandas as pd
|
||||
import tempfile
|
||||
|
||||
|
||||
class CheckboxTreeview(ttk.Treeview):
|
||||
"""支持 checkbox 的 Treeview 组件
|
||||
|
||||
使用 Unicode 字符模拟 checkbox：
|
||||
- ☐ 未选中
|
||||
- ☑ 选中
|
||||
"""
|
||||
|
||||
def __init__(self, parent, on_checkbox_change=None, **kwargs):
|
||||
"""初始化 CheckboxTreeview
|
||||
|
||||
Args:
|
||||
parent: 父容器
|
||||
on_checkbox_change: checkbox 状态改变时的回调函数
|
||||
**kwargs: 传递给 Treeview 的参数
|
||||
"""
|
||||
super().__init__(parent, **kwargs)
|
||||
self.checkboxes = {} # item_id -> bool
|
||||
self.checkbox_column = "选择"
|
||||
self.on_checkbox_change = on_checkbox_change # checkbox 状态改变回调
|
||||
|
||||
# 排序状态
|
||||
self.sort_column = None # 当前排序列的列标识符
|
||||
self.sort_direction = None # 'asc', 'desc', 或 None
|
||||
self.sortable_columns = ["选择", "材料名称"] # 可排序的列白名单
|
||||
self.original_headings = {} # 存储原始列标题文本（不含箭头）
|
||||
|
||||
# 存储原始列标题（延迟执行以确保标题已设置）
|
||||
self.after(100, self._store_original_headings)
|
||||
|
||||
# 绑定点击事件
|
||||
self.bind("<Button-1>", self._on_click)
|
||||
# 绑定表头点击事件
|
||||
self.bind("<ButtonRelease-1>", self._on_heading_click)
|
||||
|
||||
def _on_click(self, event):
|
||||
"""处理点击事件，切换 checkbox 状态"""
|
||||
# 获取点击位置对应的 item 和 column
|
||||
region = self.identify_region(event.x, event.y)
|
||||
|
||||
# 仅处理单元格点击，不处理表头点击
|
||||
if region == "cell":
|
||||
column = self.identify_column(event.x)
|
||||
item = self.identify_row(event.y)
|
||||
|
||||
# 检查是否点击了 checkbox 列（第一列）
|
||||
if column == "#1" and item:
|
||||
# 切换 checkbox 状态
|
||||
current_state = self.checkboxes.get(item, False)
|
||||
new_state = not current_state
|
||||
self.set_checked(item, new_state)
|
||||
|
||||
# 通知父组件 checkbox 状态已改变
|
||||
if self.on_checkbox_change:
|
||||
self.on_checkbox_change(item, new_state)
|
||||
|
||||
return "break" # 阻止默认行为
|
||||
|
||||
def set_checked(self, item, checked: bool):
|
||||
"""设置指定 item 的 checkbox 状态
|
||||
|
||||
Args:
|
||||
item: Treeview item ID
|
||||
checked: 是否选中
|
||||
"""
|
||||
self.checkboxes[item] = checked
|
||||
|
||||
# 更新显示
|
||||
checkbox_char = "☑" if checked else "☐"
|
||||
values = list(self.item(item, "values"))
|
||||
if values:
|
||||
values[0] = checkbox_char
|
||||
self.item(item, values=values)
|
||||
|
||||
def get_checked_items(self) -> list:
|
||||
"""获取所有选中的 item
|
||||
|
||||
Returns:
|
||||
List of item IDs
|
||||
"""
|
||||
return [item for item, checked in self.checkboxes.items() if checked]
|
||||
|
||||
def check_all(self, checked: bool = True):
|
||||
"""全选或取消全选
|
||||
|
||||
Args:
|
||||
checked: True 为全选，False 为取消全选
|
||||
"""
|
||||
for item in self.get_children():
|
||||
self.set_checked(item, checked)
|
||||
|
||||
def insert(self, parent, index, values=None, **kwargs):
|
||||
"""重写 insert 方法，初始化 checkbox 状态"""
|
||||
if values is None:
|
||||
values = []
|
||||
|
||||
# 确保第一个值是 checkbox
|
||||
if not values or values[0] not in ["☐", "☑"]:
|
||||
values = ["☐"] + list(values)
|
||||
|
||||
item = super().insert(parent, index, values=values, **kwargs)
|
||||
|
||||
# 初始化 checkbox 状态为未选中
|
||||
checkbox_char = values[0] if values else "☐"
|
||||
self.checkboxes[item] = (checkbox_char == "☑")
|
||||
|
||||
return item
|
||||
|
||||
def delete(self, *items):
|
||||
"""重写 delete 方法，清理 checkbox 状态"""
|
||||
for item in items:
|
||||
if item in self.checkboxes:
|
||||
del self.checkboxes[item]
|
||||
super().delete(*items)
|
||||
|
||||
def _store_original_headings(self):
|
||||
"""存储原始列标题文本（不含箭头）"""
|
||||
for col in self['columns']:
|
||||
self.original_headings[col] = self.heading(col, 'text')
|
||||
|
||||
def _get_column_id_from_column_index(self, column_index):
|
||||
"""将列索引 ('#1', '#2') 转换为列标识符
|
||||
|
||||
Args:
|
||||
column_index: 列索引字符串，如 '#1', '#2'
|
||||
|
||||
Returns:
|
||||
列标识符，如 '选择', '材料名称'
|
||||
"""
|
||||
index = int(column_index[1:]) - 1
|
||||
columns = self['columns']
|
||||
if 0 <= index < len(columns):
|
||||
return columns[index]
|
||||
return None
|
||||
|
||||
def _on_heading_click(self, event):
|
||||
"""处理表头点击事件，触发排序"""
|
||||
region = self.identify_region(event.x, event.y)
|
||||
|
||||
if region == "heading":
|
||||
column = self.identify_column(event.x)
|
||||
column_id = self._get_column_id_from_column_index(column)
|
||||
|
||||
# 仅对可排序列进行排序
|
||||
if column_id in self.sortable_columns:
|
||||
self._toggle_sort(column_id)
|
||||
|
||||
def _toggle_sort(self, column_id):
|
||||
"""切换指定列的排序状态
|
||||
|
||||
Args:
|
||||
column_id: 列标识符（如 '选择', '材料名称'）
|
||||
"""
|
||||
# 确定新的排序方向
|
||||
if self.sort_column == column_id:
|
||||
# 同一列：asc -> desc -> None
|
||||
if self.sort_direction == 'asc':
|
||||
new_direction = 'desc'
|
||||
elif self.sort_direction == 'desc':
|
||||
new_direction = None
|
||||
else:
|
||||
new_direction = 'asc'
|
||||
else:
|
||||
# 不同列：从升序开始
|
||||
new_direction = 'asc'
|
||||
|
||||
# 应用排序
|
||||
if new_direction:
|
||||
self._sort_by_column(column_id, new_direction)
|
||||
self.sort_column = column_id
|
||||
self.sort_direction = new_direction
|
||||
else:
|
||||
# 清除排序状态
|
||||
self.sort_column = None
|
||||
self.sort_direction = None
|
||||
|
||||
# 更新表头显示
|
||||
self._update_heading_display()
|
||||
|
||||
def _sort_by_column(self, column_id, direction):
|
||||
"""按指定列和方向排序
|
||||
|
||||
Args:
|
||||
column_id: 列标识符
|
||||
direction: 'asc' 或 'desc'
|
||||
"""
|
||||
# 收集所有项目及其数据和复选框状态
|
||||
items_data = []
|
||||
for item in self.get_children():
|
||||
values = self.item(item, "values")
|
||||
checkbox_state = self.checkboxes.get(item, False)
|
||||
items_data.append({
|
||||
'item_id': item,
|
||||
'values': values,
|
||||
'checked': checkbox_state
|
||||
})
|
||||
|
||||
# 根据列和方向排序
|
||||
if column_id == "选择":
|
||||
# 按复选框状态排序（选中在前，未选中在后）
|
||||
items_data.sort(key=lambda x: x['checked'], reverse=(direction == 'desc'))
|
||||
elif column_id == "材料名称":
|
||||
# 按材料名称排序
|
||||
items_data.sort(
|
||||
key=lambda x: str(x['values'][1]) if len(x['values']) > 1 else "",
|
||||
reverse=(direction == 'desc')
|
||||
)
|
||||
|
||||
# 重新排列项目顺序（使用 detach 和 move 保留项目ID和状态）
|
||||
for item_data in items_data:
|
||||
self.move(item_data['item_id'], '', 'end')
|
||||
|
||||
def _update_heading_display(self):
|
||||
"""更新列标题显示（添加/移除排序箭头）"""
|
||||
for col in self['columns']:
|
||||
original = self.original_headings.get(col, col)
|
||||
if col == self.sort_column:
|
||||
# 添加排序箭头
|
||||
arrow = " ↑" if self.sort_direction == 'asc' else " ↓"
|
||||
self.heading(col, text=original + arrow)
|
||||
else:
|
||||
# 移除箭头，显示原始标题
|
||||
self.heading(col, text=original)
|
||||
|
||||
|
||||
class MaterialValidationTab(ttk.Frame):
|
||||
class MaterialValidationTab(BaseTab):
|
||||
"""物料校验标签页"""
|
||||
|
||||
def __init__(self, parent, config: ConfigManager, session_manager, main_window=None):
|
||||
@@ -265,10 +41,8 @@ class MaterialValidationTab(ttk.Frame):
|
||||
session_manager: SessionManager 实例，用于权限控制
|
||||
main_window: 主窗口引用，用于获取共享的 Production ID
|
||||
"""
|
||||
super().__init__(parent)
|
||||
self.config = config
|
||||
super().__init__(parent, config, main_window)
|
||||
self.session_manager = session_manager
|
||||
self.main_window = main_window
|
||||
self.validating = False
|
||||
self.validation_results = None
|
||||
self.material_records_cache = None # 缓存完整的物料记录
|
||||
@@ -290,8 +64,8 @@ class MaterialValidationTab(ttk.Frame):
|
||||
# 稍后显示就绪消息
|
||||
try:
|
||||
self.log_text.info("物料校验标签页已就绪")
|
||||
except:
|
||||
pass # 如果窗口还未完全就绪，忽略错误
|
||||
except Exception as e:
|
||||
self.logger.debug(f"初始化日志消息失败: {e}")
|
||||
|
||||
def create_widgets(self):
|
||||
"""创建界面组件"""
|
||||
@@ -1179,8 +953,8 @@ class MaterialValidationTab(ttk.Frame):
|
||||
if temp_production_id_file and os.path.exists(temp_production_id_file):
|
||||
try:
|
||||
os.unlink(temp_production_id_file)
|
||||
except:
|
||||
pass
|
||||
except OSError as e:
|
||||
self.logger.debug(f"清理临时文件失败: {e}")
|
||||
# 更新 UI 状态
|
||||
self.after(0, self._validation_complete)
|
||||
|
||||
@@ -1441,26 +1215,6 @@ class MaterialValidationTab(ttk.Frame):
|
||||
# 刷新筛选器和结果（可能添加了新的负责人）
|
||||
self._initialize_manager_filter()
|
||||
|
||||
def _update_log(self, message: str, level: str = "INFO"):
|
||||
"""
|
||||
线程安全的日志更新（兼容接口）
|
||||
|
||||
通过统一的 logging 系统输出日志，自动同时输出到控制台和 GUI。
|
||||
|
||||
Args:
|
||||
message: 日志消息
|
||||
level: 日志级别 (INFO, SUCCESS, WARNING, ERROR, DEBUG)
|
||||
"""
|
||||
# 将自定义级别映射到 logging 级别
|
||||
level_upper = level.upper()
|
||||
if level_upper == "SUCCESS":
|
||||
# SUCCESS 映射到 INFO，但在 UI 中仍显示为 SUCCESS
|
||||
self.logger.info(message)
|
||||
else:
|
||||
# 其他级别直接映射
|
||||
log_level = getattr(logging, level_upper, logging.INFO)
|
||||
self.logger.log(log_level, message)
|
||||
|
||||
def export_results(self):
|
||||
"""导出结果到 Excel"""
|
||||
# 从配置获取输出文件路径
|
||||
@@ -1676,8 +1430,8 @@ class MaterialValidationTab(ttk.Frame):
|
||||
try:
|
||||
if 'temp_file' in locals() and os.path.exists(temp_file):
|
||||
os.unlink(temp_file)
|
||||
except:
|
||||
pass
|
||||
except OSError as e:
|
||||
self.logger.debug(f"清理临时文件失败: {e}")
|
||||
|
||||
def _delete_complete(self, report: str, stats: dict):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user