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:
242
gui/widgets/checkbox_treeview.py
Normal file
242
gui/widgets/checkbox_treeview.py
Normal file
@@ -0,0 +1,242 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
CheckboxTreeview 组件
|
||||
|
||||
支持 checkbox 的 Treeview 组件,使用 Unicode 字符模拟 checkbox:
|
||||
- ☐ 未选中
|
||||
- ☑ 选中
|
||||
"""
|
||||
|
||||
import tkinter as tk
|
||||
from tkinter import ttk
|
||||
|
||||
|
||||
class CheckboxTreeview(ttk.Treeview):
|
||||
"""支持 checkbox 的 Treeview 组件
|
||||
|
||||
使用 Unicode 字符模拟 checkbox:
|
||||
- ☐ 未选中
|
||||
- ☑ 选中
|
||||
|
||||
Features:
|
||||
- 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)
|
||||
Reference in New Issue
Block a user