1198 lines
43 KiB
Python
1198 lines
43 KiB
Python
# coding:utf-8
|
||
import sys
|
||
import os
|
||
from PySide6.QtCore import (
|
||
Qt,
|
||
QSize,
|
||
QPoint,
|
||
QEvent,
|
||
Slot,
|
||
Signal,
|
||
QThread,
|
||
QMutex,
|
||
QWaitCondition,
|
||
)
|
||
|
||
from PySide6.QtGui import QIcon, QColor, QAction
|
||
|
||
from PySide6.QtWidgets import (
|
||
QApplication,
|
||
QWidget,
|
||
QVBoxLayout,
|
||
QHBoxLayout,
|
||
QStackedWidget,
|
||
QMessageBox,
|
||
QLabel,
|
||
QFrame,
|
||
QTreeWidgetItem,
|
||
QTreeWidgetItemIterator,
|
||
QTableWidget,
|
||
QTableWidgetItem,
|
||
QListWidget,
|
||
QListWidgetItem,
|
||
QInputDialog,
|
||
QMenu,
|
||
QSizePolicy,
|
||
QDialog,
|
||
)
|
||
|
||
from qfluentwidgets import (
|
||
TableWidget,
|
||
HorizontalFlipView,
|
||
TabBar,
|
||
TabCloseButtonDisplayMode,
|
||
setTheme,
|
||
isDarkTheme,
|
||
Theme,
|
||
PushButton,
|
||
FluentIcon as FIF,
|
||
Dialog,
|
||
)
|
||
|
||
from .gallery_interface import GalleryInterface
|
||
from ..common.translator import Translator
|
||
from ..common.style_sheet import StyleSheet
|
||
|
||
import sqlite3
|
||
|
||
|
||
class CoodFormsInterface(GalleryInterface):
|
||
"""Cood Forms interface"""
|
||
|
||
def __init__(self, parent=None):
|
||
t = Translator()
|
||
super().__init__(title=t.view, subtitle="CoordinateFormsWidget", parent=parent)
|
||
self.setObjectName("coodFormsInterface")
|
||
|
||
# CoordinateFormsWidget
|
||
self.formsWidget = CoordinateFormsWidget(self)
|
||
card = self.addExampleCard(
|
||
title=self.tr(""),
|
||
widget=self.formsWidget,
|
||
sourcePath=".",
|
||
stretch=1,
|
||
)
|
||
card.topLayout.setContentsMargins(6, 0, 6, 16)
|
||
|
||
# 动态设置高度
|
||
self.__connectMainWindowSignal()
|
||
|
||
def __connectMainWindowSignal(self):
|
||
main_window = self.window()
|
||
if main_window:
|
||
self.onWindowHeightChanged(main_window.height() - 160) # 设置窗口初始高度
|
||
main_window.heightChanged.connect(self.onWindowHeightChanged)
|
||
|
||
def onWindowHeightChanged(self, new_height):
|
||
self.formsWidget.setFixedHeight(new_height - 160)
|
||
|
||
|
||
class FormDatabase:
|
||
def __init__(self, db_path):
|
||
self.db_path = db_path
|
||
self.conn = None # 数据库连接对象
|
||
self.cursor = None # 游标对象
|
||
|
||
def init_database(self):
|
||
"""初始化SQLite数据库, 不存在则创建form_info表和form_data表"""
|
||
try:
|
||
# 数据库文件路径（db目录下的forms.db）
|
||
os.makedirs(
|
||
os.path.dirname(self.db_path), exist_ok=True
|
||
) # 确保数据库目录存在
|
||
cursor = self.__connect()
|
||
|
||
# 创建表单信息表 form_info
|
||
cursor.execute(
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS form_info (
|
||
form_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
form_name TEXT NOT NULL UNIQUE, -- 表单名称唯一
|
||
create_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||
update_time TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
||
)
|
||
"""
|
||
)
|
||
|
||
# 创建表单数据表form_data（存储每行数据）
|
||
cursor.execute(
|
||
"""
|
||
CREATE TABLE IF NOT EXISTS form_data (
|
||
data_id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
form_id INTEGER NOT NULL, -- 关联表单信息表的 form_id
|
||
row_index INTEGER NOT NULL, -- 行号(从0开始)
|
||
x TEXT,
|
||
y TEXT,
|
||
z TEXT,
|
||
rx TEXT,
|
||
ry TEXT,
|
||
rz TEXT,
|
||
FOREIGN KEY (form_id) REFERENCES form_info(form_id) ON DELETE CASCADE,
|
||
UNIQUE(form_id, row_index) -- 确保一行数据只存一次
|
||
)
|
||
"""
|
||
)
|
||
|
||
# 自动更新表单修改时间
|
||
# 更新表单修改时间的触发器
|
||
cursor.execute(
|
||
"""
|
||
CREATE TRIGGER IF NOT EXISTS update_form_time
|
||
AFTER UPDATE ON form_data
|
||
FOR EACH ROW
|
||
BEGIN
|
||
UPDATE form_info SET update_time = CURRENT_TIMESTAMP WHERE form_id = NEW.form_id;
|
||
END
|
||
"""
|
||
)
|
||
|
||
# 提交，初始化数据库成功
|
||
self.commit_and_close()
|
||
|
||
except Exception as exc:
|
||
self.rollback_and_close()
|
||
raise exc
|
||
|
||
def __connect(self):
|
||
"""连接数据库并获取游标"""
|
||
if not self.conn:
|
||
self.conn = sqlite3.connect(self.db_path)
|
||
self.cursor = self.conn.cursor()
|
||
return self.cursor
|
||
|
||
def commit_and_close(self):
|
||
"""提交事务并关闭连接"""
|
||
if self.conn:
|
||
self.conn.commit() # 提交事务
|
||
self.conn.close() # 关闭连接
|
||
self.conn = None
|
||
self.cursor = None
|
||
|
||
def rollback_and_close(self):
|
||
"""回滚事务并关闭连接"""
|
||
if self.conn:
|
||
self.conn.rollback() # 回滚事务
|
||
self.conn.close() # 关闭连接
|
||
self.conn = None
|
||
self.cursor = None
|
||
|
||
def check_form_exists(self, form_name):
|
||
"""检查表单是否已经存在
|
||
成功返回form_id
|
||
失败返回None
|
||
"""
|
||
try:
|
||
cursor = self.__connect()
|
||
cursor.execute(
|
||
"SELECT form_id FROM form_info WHERE form_name = ?", (form_name,)
|
||
)
|
||
result = cursor.fetchone()
|
||
return result[0] if result else None
|
||
except Exception as exc:
|
||
self.rollback_and_close()
|
||
raise exc
|
||
|
||
def delete_form_data(self, form_id):
|
||
"""删除指定表单的所有数据"""
|
||
try:
|
||
cursor = self.__connect()
|
||
cursor.execute("DELETE FROM form_data WHERE form_id = ?", (form_id,))
|
||
except Exception as exc:
|
||
self.rollback_and_close()
|
||
raise exc
|
||
|
||
def add_new_form(self, form_name):
|
||
"""新增表单 (返回新表单的form_id)"""
|
||
try:
|
||
cursor = self.__connect()
|
||
cursor.execute("INSERT INTO form_info (form_name) VALUES (?)", (form_name,))
|
||
return cursor.lastrowid # 返回新插入的ID
|
||
except Exception as exc:
|
||
self.rollback_and_close()
|
||
raise exc
|
||
|
||
def insert_form_data(self, form_id, data_rows):
|
||
"""批量插入form_data表"""
|
||
try:
|
||
cursor = self.__connect()
|
||
for row in data_rows:
|
||
# row格式: [row_idx, x, y, z, rx, ry, rz]
|
||
cursor.execute(
|
||
"""INSERT INTO form_data
|
||
(form_id, row_index, x, y, z, rx, ry, rz)
|
||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)""",
|
||
(form_id, row[0], row[1], row[2], row[3], row[4], row[5], row[6]),
|
||
)
|
||
except Exception as exc:
|
||
self.rollback_and_close()
|
||
raise exc
|
||
|
||
def get_all_form_id_name(self):
|
||
"""获取所有form_id 和 对应的 form_name"""
|
||
try:
|
||
cursor = self.__connect()
|
||
cursor.execute(
|
||
"SELECT form_id, form_name FROM form_info ORDER BY create_time"
|
||
)
|
||
# [(form_id1, form_name1), (form_id2, form_name2), ...]
|
||
form_id_name = cursor.fetchall()
|
||
return form_id_name
|
||
except Exception as exc:
|
||
self.rollback_and_close()
|
||
raise exc
|
||
|
||
def get_row_idx_coordinate_list(self, form_id):
|
||
"""根据form_id查询对应的表单数据 data_rows (包含行索引row_idx, x, y, z, rx, ry, rz)"""
|
||
try:
|
||
cursor = self.__connect()
|
||
cursor.execute(
|
||
"SELECT row_index, x, y, z, rx, ry, rz FROM form_data "
|
||
"WHERE form_id = ?",
|
||
(form_id,),
|
||
)
|
||
# [(row_idx, x, y, z, rx, ry, rz), ...]
|
||
data_rows = cursor.fetchall()
|
||
return data_rows
|
||
except Exception as exc:
|
||
self.rollback_and_close()
|
||
raise exc
|
||
|
||
def get_sorted_coordinate_list(self, form_name):
|
||
"""
|
||
form_name 等同于线名
|
||
form_name 存在, 返回 [(x1, y1, z1, rx1, ry1, rz1), ......], 按照row_idx排序
|
||
不存在, 返回 None
|
||
"""
|
||
pass
|
||
|
||
# 包含 坐标 和 对应的状态（运动类型？点位类型？）
|
||
def get_coordinate_state_list(self, form_name):
|
||
pass
|
||
|
||
|
||
# 数据库保存线程
|
||
class DatabaseSaveThread(QThread):
|
||
# 信号1：需要用户确认是否覆盖数据库中的数据
|
||
need_confirm_signal = Signal()
|
||
# 信号2：数据库操作结果
|
||
result_signal = Signal(str)
|
||
|
||
def __init__(self, db_path, form_name, data_rows):
|
||
super().__init__()
|
||
self.db_path = db_path
|
||
self.form_name = form_name
|
||
self.data_rows = data_rows # 主线程传递过来的UI数据
|
||
self.user_choice = None # 存储用户选择（True=覆盖，False=取消）
|
||
self.mutex = QMutex()
|
||
self.condition = QWaitCondition()
|
||
|
||
def run(self):
|
||
try:
|
||
db = FormDatabase(self.db_path)
|
||
|
||
# 初始化数据库，至少执行一次
|
||
db.init_database()
|
||
|
||
# 检查form_name名的表单是否已经存在
|
||
exist_form_id = db.check_form_exists(self.form_name)
|
||
|
||
if exist_form_id is not None:
|
||
# 存在表单，发送信号给主线程, 向用户请求确认
|
||
self.need_confirm_signal.emit()
|
||
|
||
# 等待UI线程传入用户选择
|
||
self.wait_for_user_choice()
|
||
|
||
if not self.user_choice:
|
||
# 用户选择不覆盖，结束操作
|
||
self.result_signal.emit(f"已取消，表单「{self.form_name}」未修改")
|
||
return
|
||
|
||
# 用户选择覆盖：删除旧数据
|
||
db.delete_form_data(exist_form_id)
|
||
# 插入新数据（复用exist_form_id）
|
||
db.insert_form_data(exist_form_id, self.data_rows)
|
||
self.result_signal.emit(f"表单「{self.form_name}」已覆盖并保存")
|
||
|
||
else:
|
||
# 表单不存在：直接新增
|
||
new_form_id = db.add_new_form(self.form_name)
|
||
db.insert_form_data(new_form_id, self.data_rows)
|
||
self.result_signal.emit(f"表单「{self.form_name}」已保存")
|
||
|
||
# 操作完成，提交并关闭连接
|
||
db.commit_and_close()
|
||
# db.rollback_and_close()
|
||
|
||
except Exception as e:
|
||
# 数据库操作异常
|
||
self.result_signal.emit(f"保存失败：{str(e)}")
|
||
|
||
def wait_for_user_choice(self):
|
||
"""阻塞线程，等待用户选择（条件变量）"""
|
||
self.mutex.lock()
|
||
# 等待，直到set_user_choice调用wakeAll()
|
||
self.condition.wait(self.mutex)
|
||
self.mutex.unlock()
|
||
|
||
def set_user_choice(self, choice):
|
||
"""UI线程调用: 设置用户选择并唤醒数据库保存操作线程"""
|
||
self.mutex.lock()
|
||
self.user_choice = choice
|
||
self.condition.wakeAll() # 唤醒等待的数据库保存操作线程
|
||
self.mutex.unlock()
|
||
|
||
|
||
# 数据库读取操作线程
|
||
class DatabaseReadThread(QThread):
|
||
# 信号1: 从form_info 读取所有的 id-name 发送
|
||
# [(form_id1, form_name1),...]
|
||
form_id_name_signal = Signal(list)
|
||
|
||
# 信号2: 从 from_data 读取所有的 选择的表单的数据发送
|
||
# {form_name1: data_rows1, ...}
|
||
form_data_signal = Signal(dict)
|
||
|
||
# 信号3: 状态信息
|
||
state_signal = Signal(str)
|
||
|
||
def __init__(self, db_path):
|
||
super().__init__()
|
||
self.db_path = db_path
|
||
self.selected_form_id_name = None # 选择的 表单id-name
|
||
|
||
# 初始化信号量
|
||
self.mutex = QMutex()
|
||
self.condition = QWaitCondition()
|
||
|
||
def run(self):
|
||
try:
|
||
db = FormDatabase(self.db_path)
|
||
|
||
# 初始化数据库
|
||
db.init_database()
|
||
|
||
# 从form_info表 查询所有的 form_id 和 form_name
|
||
all_form_id_name = db.get_all_form_id_name()
|
||
self.form_id_name_signal.emit(all_form_id_name)
|
||
|
||
# 等待用户选择 想要查看的 form
|
||
self.wait_for_selection()
|
||
|
||
if not self.selected_form_id_name:
|
||
self.state_signal.emit("未选择任何表单")
|
||
return
|
||
|
||
form_date_dict = dict()
|
||
for form_id, form_name in self.selected_form_id_name:
|
||
data_rows = db.get_row_idx_coordinate_list(form_id)
|
||
form_date_dict[form_name] = data_rows
|
||
|
||
self.form_data_signal.emit(form_date_dict)
|
||
db.commit_and_close()
|
||
|
||
except Exception as exc:
|
||
# 数据库异常
|
||
self.state_signal.emit(f"读取失败：{str(exc)}")
|
||
|
||
def wait_for_selection(self):
|
||
self.mutex.lock()
|
||
self.condition.wait(self.mutex)
|
||
self.mutex.unlock()
|
||
|
||
def set_selected_forms(self, selected_form_id_name):
|
||
"""UI线程调用, 设置选择的 表单的 id-name"""
|
||
self.mutex.lock()
|
||
self.selected_form_id_name = selected_form_id_name
|
||
self.condition.wakeAll()
|
||
self.mutex.unlock()
|
||
|
||
|
||
class CoordinateTableWidget(QWidget):
|
||
# 选中行更新信号
|
||
update_table_signal = Signal()
|
||
|
||
# 移动到选中行的 坐标的信号
|
||
move_to_coodinate_signal = Signal(list)
|
||
|
||
def __init__(
|
||
self, parent=None, form_name="default", hideVHeader=False, initRowCount=True
|
||
):
|
||
super().__init__(parent=parent)
|
||
# 保存当前表单名，根据form_name设置行名
|
||
self.form_name = form_name
|
||
self.hideVHeader = hideVHeader # 是否隐藏行名
|
||
self.initRowCount = initRowCount # 是否需要初始的行数
|
||
self.setObjectName("CoordinateTableWidget")
|
||
|
||
# 主布局
|
||
self.mainLayout = QVBoxLayout(self)
|
||
self.mainLayout.setContentsMargins(30, 30, 30, 30)
|
||
self.mainLayout.setSpacing(20)
|
||
|
||
# 创建表格
|
||
self.createTable()
|
||
|
||
# 创建按钮区域
|
||
self.createButtons()
|
||
|
||
# 状态标签
|
||
self.status_label = QLabel("未选中任何行")
|
||
self.mainLayout.addWidget(self.status_label)
|
||
|
||
# 应用主题样式
|
||
self.applyThemeStyle()
|
||
# StyleSheet.VIEW_INTERFACE.apply(self)
|
||
|
||
# 初始化数据（行名会关联表单名）
|
||
self.initTableData()
|
||
|
||
# 更新表单名（供外部调用，修改后同步更新行名）
|
||
def update_form_name(self, new_name):
|
||
self.form_name = new_name
|
||
self.updateRowHeaders() # 同步更新所有行名
|
||
|
||
def createTable(self):
|
||
self.table = TableWidget(self)
|
||
self.table.setBorderVisible(True)
|
||
self.table.setBorderRadius(8)
|
||
self.table.setWordWrap(False)
|
||
|
||
# 表单的初始列数和行数
|
||
if self.initRowCount:
|
||
self.table.setRowCount(5) # 表的行数
|
||
|
||
self.table.setColumnCount(6) # 表的列数
|
||
|
||
self.table.setHorizontalHeaderLabels(["x", "y", "z", "rx", "ry", "rz"])
|
||
self.table.horizontalHeader().setStyleSheet(
|
||
"QHeaderView::section {font-size: 19px; font-weight: bold;}"
|
||
)
|
||
self.table.verticalHeader().setStyleSheet(
|
||
"QHeaderView::section {font-size: 14px; font-weight: bold;}"
|
||
)
|
||
|
||
if self.hideVHeader:
|
||
self.table.verticalHeader().hide() # 隐藏行标题（行名）
|
||
|
||
for i in range(6):
|
||
self.table.horizontalHeader().setSectionResizeMode(
|
||
i, self.table.horizontalHeader().ResizeMode.Stretch
|
||
)
|
||
|
||
self.table.setSelectionBehavior(self.table.SelectionBehavior.SelectRows)
|
||
self.table.setSelectionMode(self.table.SelectionMode.ExtendedSelection)
|
||
self.table.setContextMenuPolicy(Qt.ContextMenuPolicy.CustomContextMenu)
|
||
self.table.customContextMenuRequested.connect(self.showContextMenu)
|
||
self.table.itemSelectionChanged.connect(self.onCellSelected)
|
||
self.mainLayout.addWidget(self.table, stretch=1)
|
||
|
||
def createButtons(self):
|
||
btnLayout = QHBoxLayout()
|
||
btnLayout.setSpacing(15)
|
||
|
||
self.addBtn = PushButton("添加行")
|
||
self.addBtn.clicked.connect(self.addNewRow)
|
||
btnLayout.addWidget(self.addBtn)
|
||
|
||
self.deleteBtn = PushButton("删除行")
|
||
self.deleteBtn.clicked.connect(self.deleteSelectedRow)
|
||
btnLayout.addWidget(self.deleteBtn)
|
||
|
||
self.setBtn = PushButton("设置点位")
|
||
self.setBtn.clicked.connect(self.setSelectedRow)
|
||
btnLayout.addWidget(self.setBtn)
|
||
|
||
self.resetBtn = PushButton("重置点位")
|
||
self.resetBtn.clicked.connect(self.resetSelectedRow)
|
||
btnLayout.addWidget(self.resetBtn)
|
||
|
||
# 移动按钮
|
||
self.readBtn = PushButton("移动")
|
||
self.readBtn.setIcon(FIF.RIGHT_ARROW.icon()) # FOLDER
|
||
self.readBtn.clicked.connect(self.moveToSelectedCoodinate)
|
||
btnLayout.addWidget(self.readBtn)
|
||
|
||
# 保存按钮
|
||
self.saveBtn = PushButton("保存数据")
|
||
self.saveBtn.setIcon(FIF.SAVE.icon()) # 使用保存图标
|
||
self.saveBtn.clicked.connect(self.saveToDatabase) # 关联保存函数
|
||
btnLayout.addWidget(self.saveBtn)
|
||
|
||
self.mainLayout.addLayout(btnLayout)
|
||
|
||
def get_ui_data(self):
|
||
"""从表格UI中获取数据 [[行索引row_idx, x, y, z, rx, ry, rz], ......]"""
|
||
row_count = self.table.rowCount()
|
||
column_count = self.table.columnCount()
|
||
data_rows = []
|
||
for row_idx in range(row_count):
|
||
row_data = [row_idx]
|
||
for col_idx in range(column_count):
|
||
item = self.table.item(row_idx, col_idx)
|
||
|
||
# 数据合法性和完整性判定: 保证获取到的是有效数据
|
||
row_data.append(
|
||
item.text().strip() if item and item.text().strip() else "-9999"
|
||
) # 该表格为空，填充 -9999
|
||
|
||
# 放入这一行的数据 行索引row_idx, x, y, z, rx, ry, rz
|
||
data_rows.append(row_data)
|
||
|
||
return data_rows
|
||
|
||
def showConfirmDialog(self):
|
||
"""显示覆盖确认对话框"""
|
||
msg_box = QMessageBox(self)
|
||
msg_box.setWindowTitle("覆盖确认")
|
||
msg_box.setText(f"表单「{self.form_name}」已存在，是否覆盖？")
|
||
msg_box.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
|
||
msg_box.setDefaultButton(QMessageBox.No)
|
||
msg_box.setStyleSheet("QMessageBox QLabel { color: black !important; }")
|
||
reply = msg_box.exec()
|
||
|
||
# 将用户选择（Yes=True，No=False）传给线程，唤醒数据库线程继续执行
|
||
if (
|
||
hasattr(self, "db_thread")
|
||
and self.db_thread is not None
|
||
and not self.db_thread.isFinished()
|
||
):
|
||
self.db_thread.set_user_choice(reply == QMessageBox.Yes)
|
||
|
||
def applyThemeStyle(self):
|
||
if isDarkTheme():
|
||
self.setStyleSheet(
|
||
"""
|
||
#CoordinateTableWidget {background-color: rgb(30, 30, 30);}
|
||
#CoordinateTableWidget QLabel {font-size: 16px; color: rgb(240, 240, 240);}
|
||
#CoordinateTableWidget QTableWidget {color: rgb(240, 240, 240); gridline-color: rgb(60, 60, 60);}
|
||
#CoordinateTableWidget QHeaderView::section {
|
||
background-color: rgb(40, 40, 40);
|
||
color: rgb(200, 200, 200);
|
||
border: 1px solid rgb(60, 60, 60);
|
||
padding: 5px;
|
||
}
|
||
"""
|
||
)
|
||
else:
|
||
self.setStyleSheet(
|
||
"""
|
||
#CoordinateTableWidget {background-color: rgb(245, 245, 245);}
|
||
#CoordinateTableWidget QLabel {font-size: 16px; color: rgb(60, 60, 60);}
|
||
#CoordinateTableWidget QTableWidget {color: rgb(60, 60, 60); gridline-color: rgb(200, 200, 200);}
|
||
#CoordinateTableWidget QHeaderView::section {
|
||
background-color: rgb(230, 230, 230);
|
||
color: rgb(60, 60, 60);
|
||
border: 1px solid rgb(200, 200, 200);
|
||
padding: 5px;
|
||
}
|
||
"""
|
||
)
|
||
|
||
# 初始化表
|
||
def initTableData(self):
|
||
for row in range(self.table.rowCount()):
|
||
# 行名格式：表单名_行号
|
||
header_item = QTableWidgetItem(f"{self.form_name}_{row+1}")
|
||
header_item.setTextAlignment(Qt.AlignCenter)
|
||
self.setThemeTextColor(header_item)
|
||
self.table.setVerticalHeaderItem(row, header_item)
|
||
|
||
for col in range(self.table.columnCount()):
|
||
item = QTableWidgetItem("")
|
||
item.setTextAlignment(Qt.AlignCenter)
|
||
self.setThemeTextColor(item)
|
||
self.table.setItem(row, col, item)
|
||
|
||
def setThemeTextColor(self, item):
|
||
item.setForeground(Qt.white if isDarkTheme() else Qt.black)
|
||
|
||
def showNoSelectWarning(self, msg):
|
||
msg_box = QMessageBox(self)
|
||
msg_box.setIcon(QMessageBox.Warning)
|
||
msg_box.setWindowTitle("提示")
|
||
msg_box.setText(msg)
|
||
msg_box.setStandardButtons(QMessageBox.Ok)
|
||
msg_box.setStyleSheet("QLabel{color: black;}")
|
||
msg_box.exec()
|
||
|
||
def update_table_data(self, positionList=None):
|
||
selectedRows = self.getSelectedRows()
|
||
for row in selectedRows:
|
||
for col in range(self.table.columnCount()):
|
||
item = QTableWidgetItem(
|
||
str(positionList[col])
|
||
if (positionList and 0 <= col < len(positionList))
|
||
else "-9999"
|
||
)
|
||
item.setTextAlignment(Qt.AlignCenter)
|
||
self.setThemeTextColor(item)
|
||
self.table.setItem(row, col, item)
|
||
|
||
if positionList:
|
||
self.status_label.setText(f"已将选中行设置为{positionList}")
|
||
else:
|
||
self.status_label.setText(
|
||
"已将选中行设置为[-9999, -9999, -9999, -9999, -9999, -9999]"
|
||
)
|
||
|
||
"""
|
||
======================================槽函数=================================
|
||
"""
|
||
|
||
# 添加新行时，行名同样关联当前表单名
|
||
def addNewRow(self):
|
||
# 检查有没有选中行
|
||
# 如果有选中行，在选中行的最下方新增一行
|
||
# 没有选中行，直接在最下方新增一行
|
||
selected_rows = self.getSelectedRows()
|
||
insert_pos = max(selected_rows) + 1 if selected_rows else self.table.rowCount()
|
||
|
||
# 新增一行
|
||
self.table.insertRow(insert_pos)
|
||
|
||
# 更新行名
|
||
self.updateRowHeaders()
|
||
|
||
# 初始化单元格
|
||
for col in range(self.table.columnCount()):
|
||
item = QTableWidgetItem("")
|
||
item.setTextAlignment(Qt.AlignCenter)
|
||
self.setThemeTextColor(item)
|
||
self.table.setItem(insert_pos, col, item)
|
||
|
||
self.status_label.setText(f"已添加第 {insert_pos+1} 行")
|
||
|
||
# 删除行
|
||
def deleteSelectedRow(self):
|
||
selectedRows = self.getSelectedRows()
|
||
if not selectedRows:
|
||
self.showNoSelectWarning("请先选中再删除！")
|
||
return
|
||
|
||
msg_box = QMessageBox(self)
|
||
msg_box.setWindowTitle("提示")
|
||
rowsStr = ", ".join(map(str, [row + 1 for row in selectedRows]))
|
||
msg_box.setText(f"确定要删除选中的第 {rowsStr} 行吗？")
|
||
msg_box.setStandardButtons(QMessageBox.Yes | QMessageBox.No)
|
||
msg_box.setStyleSheet("QLabel{color: black;}")
|
||
if msg_box.exec() != QMessageBox.Yes:
|
||
return
|
||
|
||
for row in sorted(selectedRows, reverse=True):
|
||
self.table.removeRow(row)
|
||
self.updateRowHeaders() # 删除后重新编号行名
|
||
self.status_label.setText("已删除选中行")
|
||
|
||
# 设置选中行
|
||
def setSelectedRow(self):
|
||
selectedRows = self.getSelectedRows()
|
||
if not selectedRows:
|
||
self.showNoSelectWarning("请先选中再设置点位！")
|
||
return
|
||
|
||
# 选中行默认设置
|
||
self.update_table_data()
|
||
|
||
# 发送选中行设置（更新）信号
|
||
self.update_table_signal.emit()
|
||
|
||
# 重置点位
|
||
def resetSelectedRow(self):
|
||
selectedRows = self.getSelectedRows()
|
||
if not selectedRows:
|
||
self.showNoSelectWarning("请先选中再重置！")
|
||
return
|
||
|
||
for row in selectedRows:
|
||
for col in range(self.table.columnCount()):
|
||
item = QTableWidgetItem("")
|
||
item.setTextAlignment(Qt.AlignCenter)
|
||
self.setThemeTextColor(item)
|
||
self.table.setItem(row, col, item)
|
||
|
||
self.status_label.setText("已重置选中行")
|
||
|
||
# 移动
|
||
def moveToSelectedCoodinate(self):
|
||
selectedRows = self.getSelectedRows()
|
||
if not selectedRows:
|
||
self.showNoSelectWarning("请先选中再移动！")
|
||
return
|
||
|
||
# 得到选中行的 点位(坐标) 列表，然后发送
|
||
sortedRowIdList = sorted(selectedRows) # 将选中行的行索引进行排序
|
||
coordinate_rows = list()
|
||
for row_idx in sortedRowIdList:
|
||
row_coordinate = list() # 保存这一行的坐标
|
||
for col_idx in range(6): # 0-5 ，这些列为坐标 x, y, z, rx, ry, rz
|
||
item = self.table.item(row_idx, col_idx)
|
||
|
||
# 坐标合法性监测
|
||
|
||
row_coordinate.append(item.text().strip())
|
||
|
||
# 放入这一行的坐标
|
||
# 结果：[[x1, y1, z1, rx1, ry1, rz1], ......]
|
||
# 里面的 x1等都为字符串类型
|
||
coordinate_rows.append(row_coordinate)
|
||
# 发送移动到这些坐标的信号
|
||
self.move_to_coodinate_signal.emit(coordinate_rows)
|
||
|
||
# 保存数据
|
||
def saveToDatabase(self):
|
||
# 检查是否已有正在运行的数据库线程
|
||
if hasattr(self, "db_thread") and self.db_thread and self.db_thread.isRunning():
|
||
self.status_label.setText("有未完成的数据库保存操作, 请稍后再试......")
|
||
return
|
||
|
||
# 获取UI数据
|
||
data_rows = self.get_ui_data()
|
||
if not data_rows:
|
||
self.status_label.setText("没有数据需要保存")
|
||
return
|
||
|
||
# 创建数据库操作线程并启动
|
||
self.db_thread = DatabaseSaveThread(
|
||
db_path="db/forms.db", form_name=self.form_name, data_rows=data_rows
|
||
)
|
||
|
||
# 连接信号：线程需要确认时，UI线程弹对话框
|
||
self.db_thread.need_confirm_signal.connect(self.showConfirmDialog)
|
||
# 连接信号：数据库操作线程返回最终结果，UI线程更新 状态标签
|
||
self.db_thread.result_signal.connect(self.status_label.setText)
|
||
|
||
# 启动线程
|
||
self.db_thread.start()
|
||
|
||
def showContextMenu(self, position):
|
||
"""显示右键菜单（复制行数据）"""
|
||
if not self.getSelectedRows():
|
||
return
|
||
|
||
menu = QMenu()
|
||
copyAction = QAction("复制此行数据", self)
|
||
copyAction.triggered.connect(self.copyRowData)
|
||
menu.addAction(copyAction)
|
||
menu.exec(self.table.mapToGlobal(position))
|
||
|
||
def copyRowData(self):
|
||
"""复制选中行数据到剪贴板"""
|
||
selectedRows = self.getSelectedRows()
|
||
if not selectedRows:
|
||
return
|
||
|
||
# 目前只支持复制一行数据，
|
||
# 选中多行，默认复制选中的多行中行数最小的那行
|
||
row = selectedRows[0]
|
||
rowData = []
|
||
for col in range(self.table.columnCount()):
|
||
item = self.table.item(row, col)
|
||
rowData.append(item.text() if item else "")
|
||
|
||
dataStr = ", ".join(rowData)
|
||
clipboard = QApplication.clipboard()
|
||
clipboard.setText(dataStr)
|
||
self.status_label.setText(f"已复制第 {row + 1} 行的数据: {dataStr}")
|
||
|
||
def getSelectedRows(self):
|
||
return list(set(item.row() for item in self.table.selectedItems()))
|
||
|
||
def onCellSelected(self):
|
||
selectedRows = self.getSelectedRows()
|
||
if selectedRows:
|
||
self.status_label.setText(
|
||
f"选中行：{', '.join(str(row + 1) for row in selectedRows)}"
|
||
)
|
||
else:
|
||
self.status_label.setText("未选中任何行")
|
||
|
||
# 更新行名
|
||
def updateRowHeaders(self):
|
||
for row in range(self.table.rowCount()):
|
||
# 行名随表单名动态变化：表单名_行号
|
||
header_item = QTableWidgetItem(f"{self.form_name}_{row+1}")
|
||
header_item.setTextAlignment(Qt.AlignCenter)
|
||
self.setThemeTextColor(header_item)
|
||
self.table.setVerticalHeaderItem(row, header_item)
|
||
|
||
|
||
class CoordinateFormsWidget(QWidget):
|
||
# 表单中 更新信号 (用于更新表单中的选中行数据)
|
||
form_update_signal = Signal(CoordinateTableWidget)
|
||
|
||
# 表单中 移动到坐标的信号 (用于机器臂的移动)
|
||
form_move_signal = Signal(list)
|
||
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
self.setWindowTitle("Coordinate Forms")
|
||
self.setObjectName("CoordinateFormsWidget")
|
||
self.resize(1100, 750)
|
||
self.formTotalCnt = 0 # 创建过的总表单数
|
||
|
||
# 主布局
|
||
self.mainLayout = QVBoxLayout(self)
|
||
self.mainLayout.setContentsMargins(0, 0, 0, 0)
|
||
self.mainLayout.setSpacing(0)
|
||
|
||
# 标签栏（支持双击修改名称）
|
||
self.tabBar = TabBar(self)
|
||
self.tabBar.setMovable(True)
|
||
self.tabBar.setTabMaximumWidth(200)
|
||
self.tabBar.setCloseButtonDisplayMode(TabCloseButtonDisplayMode.ON_HOVER)
|
||
|
||
# 绑定槽函数
|
||
self.tabBar.tabAddRequested.connect(self.addNewForm)
|
||
self.tabBar.tabCloseRequested.connect(self.closeForm)
|
||
self.tabBar.currentChanged.connect(self.switchForm)
|
||
|
||
# 给tabBar安装事件过滤器，用于监听双击事件
|
||
self.tabBar.installEventFilter(self)
|
||
self.mainLayout.addWidget(self.tabBar)
|
||
|
||
# 表单内容区
|
||
self.formStack = QStackedWidget(self)
|
||
self.formStack.setObjectName("CoordinateFormStack")
|
||
self.mainLayout.addWidget(self.formStack)
|
||
|
||
# 读取按钮
|
||
self.readBtn = PushButton("读取数据")
|
||
self.readBtn.setIcon(FIF.LIBRARY.icon())
|
||
self.readBtn.clicked.connect(self.readFromDatabase)
|
||
self.mainLayout.addWidget(self.readBtn)
|
||
|
||
# 初始化第一个表单
|
||
self.initFirstForm()
|
||
|
||
# 初始化样式，应用主题样式
|
||
self.applyThemeStyle()
|
||
|
||
def applyThemeStyle(self):
|
||
"""根据当前主题应用样式"""
|
||
if isDarkTheme():
|
||
# 深色主题样式
|
||
self.setStyleSheet(
|
||
"""
|
||
#CoordinateFormStack {
|
||
background-color: rgb(30, 30, 30); /* 堆叠窗口背景设为深色 */
|
||
}
|
||
#CoordinateFormsWidget {
|
||
background-color: rgb(30, 30, 30); /* 主窗口背景设为深色 */
|
||
}
|
||
"""
|
||
)
|
||
else:
|
||
# 浅色主题样式
|
||
self.setStyleSheet(
|
||
"""
|
||
#CoordinateFormStack {
|
||
background-color: rgb(245, 245, 245); /* 堆叠窗口背景设为浅色 */
|
||
}
|
||
#CoordinateFormsWidget {
|
||
background-color: rgb(245, 245, 245); /* 主窗口背景设为浅色 */
|
||
}
|
||
"""
|
||
)
|
||
|
||
# 事件过滤器：监听TabBar的鼠标双击事件
|
||
def eventFilter(self, obj, event):
|
||
# 只处理TabBar的双击事件
|
||
if obj == self.tabBar and event.type() == QEvent.MouseButtonDblClick:
|
||
if event.button() == Qt.LeftButton: # 左键双击
|
||
# 获取双击位置（相对于TabBar）
|
||
pos = event.position().toPoint()
|
||
# 判断双击位置是否在标签区域内
|
||
tab_region = self.tabBar.tabRegion()
|
||
if tab_region.contains(pos):
|
||
# 计算双击的是哪个标签
|
||
index = self.getTabIndexByPos(pos)
|
||
if index != -1:
|
||
self.renameForm(index) # 调用修改表单名的方法
|
||
return True # 事件已处理
|
||
return super().eventFilter(obj, event)
|
||
|
||
# 根据点击位置计算对应的标签索引
|
||
def getTabIndexByPos(self, pos):
|
||
"""通过坐标判断双击的是哪个标签"""
|
||
x = 0
|
||
for i in range(self.tabBar.count()):
|
||
# 获取第i个标签的宽度
|
||
tab_rect = self.tabBar.tabRect(i)
|
||
if x <= pos.x() < x + tab_rect.width():
|
||
return i # 返回标签索引
|
||
x += tab_rect.width()
|
||
return -1 # 未命中任何标签
|
||
|
||
# 初始化第一个表单
|
||
def initFirstForm(self):
|
||
self.formTotalCnt += 1 # 1
|
||
firstFormName = f"default"
|
||
formKey = f"form_{self.formTotalCnt}" # 唯一标识, form_1
|
||
|
||
# 创建表单实例时传入表单名
|
||
newForm = CoordinateTableWidget(form_name=firstFormName, parent=self)
|
||
newForm.update_table_signal.connect(self.handleFormUpdate)
|
||
newForm.move_to_coodinate_signal.connect(self.form_move_signal)
|
||
|
||
self.tabBar.addTab(routeKey=formKey, text=firstFormName, icon=FIF.TILES.icon())
|
||
self.formStack.addWidget(newForm)
|
||
|
||
# 添加新表单
|
||
def addNewForm(self):
|
||
"""添加新表单（带自定义名称输入）"""
|
||
self.formTotalCnt += 1
|
||
formCount = self.tabBar.count()
|
||
defaultName = f"default{formCount}"
|
||
|
||
# 弹出输入框让用户输入表单名
|
||
# formName, ok = QInputDialog.getText(
|
||
# self, "创建新表单名称", "请输入新表单的名称：", text=defaultName
|
||
# )
|
||
|
||
formName, ok = QInputDialog.getText(
|
||
self, "请输入新创建的表单名", "", text=defaultName
|
||
)
|
||
|
||
# 用户取消，不添加新表单
|
||
if not ok:
|
||
return
|
||
|
||
# 用户输入为空，使用默认名称
|
||
if not formName.strip():
|
||
formName = defaultName
|
||
|
||
formKey = f"form_{self.formTotalCnt}" # 唯一标识
|
||
|
||
# 创建表单实例时传入表单名
|
||
newForm = CoordinateTableWidget(form_name=formName, parent=self)
|
||
newForm.update_table_signal.connect(self.handleFormUpdate)
|
||
newForm.move_to_coodinate_signal.connect(self.form_move_signal)
|
||
|
||
self.tabBar.addTab(routeKey=formKey, text=formName, icon=FIF.TILES.icon())
|
||
self.formStack.addWidget(newForm)
|
||
|
||
# 切换到新创建的表单, 并同步内容
|
||
self.tabBar.setCurrentIndex(self.tabBar.count() - 1)
|
||
self.tabBar.currentChanged.emit(self.tabBar.currentIndex())
|
||
|
||
# 关闭表单
|
||
def closeForm(self, index):
|
||
# if self.tabBar.count() <= 1: # 必须保留一个表单
|
||
# msg_box = QMessageBox(self)
|
||
# msg_box.setIcon(QMessageBox.Warning)
|
||
# msg_box.setWindowTitle("提示")
|
||
# msg_box.setText("至少保留一个表单")
|
||
# msg_box.setStandardButtons(QMessageBox.Ok)
|
||
# msg_box.setStyleSheet("QMessageBox QLabel { color: black !important; }")
|
||
# msg_box.exec()
|
||
# return
|
||
|
||
self.tabBar.removeTab(index)
|
||
formToRemove = self.formStack.widget(index)
|
||
self.formStack.removeWidget(formToRemove)
|
||
formToRemove.deleteLater()
|
||
|
||
def switchForm(self, index):
|
||
if 0 <= index < self.formStack.count():
|
||
self.formStack.setCurrentIndex(index)
|
||
|
||
# 更新表单名
|
||
def renameForm(self, index):
|
||
"""双击标签修改表单名，并同步更新行名"""
|
||
if index < 0 or index >= self.tabBar.count():
|
||
return
|
||
|
||
# 获取当前表单名作为默认值
|
||
currentName = self.tabBar.tabText(index)
|
||
|
||
# 弹出输入框修改名称
|
||
# newName, ok = QInputDialog.getText(
|
||
# self, "修改表单名称", "请输入新的表单名称：", text=currentName
|
||
# )
|
||
newName, ok = QInputDialog.getText(
|
||
self, "请输入修改后的表单名", "", text=currentName
|
||
)
|
||
if not ok or not newName.strip():
|
||
return # 取消或空输入则不修改
|
||
|
||
# 更新标签栏显示的名称
|
||
self.tabBar.setTabText(index, newName)
|
||
|
||
# 获取对应的表单实例，更新表单并同步更新行名
|
||
formWidget = self.formStack.widget(index)
|
||
if isinstance(formWidget, CoordinateTableWidget):
|
||
formWidget.update_form_name(newName) # 同步更新行名
|
||
|
||
# 处理表单选中行更新
|
||
def handleFormUpdate(self):
|
||
sender_form = self.sender()
|
||
if isinstance(sender_form, CoordinateTableWidget):
|
||
self.form_update_signal.emit(sender_form)
|
||
|
||
def readFromDatabase(self):
|
||
"""启动读取线程，处理表单选择和数据生成"""
|
||
# 检查是否已有运行中的读取线程
|
||
if (
|
||
hasattr(self, "read_thread")
|
||
and self.read_thread
|
||
and self.read_thread.isRunning()
|
||
):
|
||
# self.status_label.setText("有未完成的数据库读取操作，请稍后再试......")
|
||
return
|
||
|
||
# 创建读取线程
|
||
self.read_thread = DatabaseReadThread(db_path="db/forms.db")
|
||
|
||
# bind
|
||
# 连接信号：接收所有表单名，显示选择页面
|
||
self.read_thread.form_id_name_signal.connect(self.showFormSelection)
|
||
# 连接信号：接收表单数据，生成新表单
|
||
self.read_thread.form_data_signal.connect(self.generateNewForms)
|
||
# 连接信号：状态提示 (数据库错误等)
|
||
self.read_thread.state_signal.connect(self.showDatabaseReadWarning)
|
||
|
||
# 启动线程
|
||
self.read_thread.start()
|
||
|
||
def showDatabaseReadWarning(self, msg):
|
||
msg_box = QMessageBox(self)
|
||
msg_box.setIcon(QMessageBox.Warning)
|
||
msg_box.setWindowTitle("提示")
|
||
msg_box.setText(msg)
|
||
msg_box.setStandardButtons(QMessageBox.Ok)
|
||
msg_box.setStyleSheet("QLabel{color: black;}")
|
||
msg_box.exec()
|
||
|
||
def showFormSelection(self, all_form_id_name):
|
||
"""显示多选表单选择页面 (UI线程处理)"""
|
||
if not all_form_id_name:
|
||
# self.status_label.setText("数据库为空, 没有任何表单")
|
||
self.read_thread.set_selected_forms(list())
|
||
return
|
||
|
||
# 提取表单名（用于列表显示）
|
||
form_names = [name for (_, name) in all_form_id_name]
|
||
|
||
# 创建多选列表对话框
|
||
dialog = QDialog(self)
|
||
|
||
# 设置对话框宽高
|
||
dialog.resize(400, 300) # 可根据需要调整数值
|
||
|
||
dialog.setWindowTitle("选择表单")
|
||
layout = QVBoxLayout(dialog)
|
||
list_widget = QListWidget()
|
||
list_widget.setSelectionMode(QListWidget.ExtendedSelection) # 支持多选
|
||
|
||
# 设置列表样式
|
||
list_widget.setStyleSheet(
|
||
"""
|
||
QListWidget {
|
||
font-size: 14px; /* 字体大小 */
|
||
padding: 5px; /* 内边距 */
|
||
}
|
||
QListWidget::item {
|
||
height: 30px; /* 行高 */
|
||
}
|
||
"""
|
||
)
|
||
|
||
# 表单名添加到列表
|
||
list_widget.addItems(form_names)
|
||
layout.addWidget(list_widget)
|
||
|
||
# 确认/取消按钮
|
||
btn_layout = QHBoxLayout()
|
||
confirm_btn = PushButton("确认")
|
||
confirm_btn.setStyleSheet("color: black;")
|
||
cancel_btn = PushButton("取消")
|
||
cancel_btn.setStyleSheet("color: black;")
|
||
btn_layout.addWidget(confirm_btn)
|
||
btn_layout.addWidget(cancel_btn)
|
||
layout.addLayout(btn_layout)
|
||
|
||
# 确认按钮逻辑：获取选中的表单ID和名称
|
||
def onDialogConfirm():
|
||
selected_items = list_widget.selectedItems()
|
||
if not selected_items: # 没有选择，则退出读取
|
||
# self.read_thread.set_selected_forms(list())
|
||
# QMessageBox.warning(dialog, "提示", "请至少选择一个表单")
|
||
dialog.reject() # 按取消处理，在onDialogClosed中处理
|
||
return
|
||
|
||
# 映射选中的名称到 (form_id, form_name)
|
||
selected = [
|
||
all_form_id_name[i]
|
||
for i in range(len(all_form_id_name))
|
||
if list_widget.item(i).isSelected()
|
||
]
|
||
self.read_thread.set_selected_forms(selected) # 通知线程用户选择
|
||
dialog.accept()
|
||
|
||
# 对话框关闭时的统一处理
|
||
def onDialogClosed(result):
|
||
# 如果不是用户确认（result != QDialog.Accepted），则通知线程取消
|
||
if result != QDialog.Accepted:
|
||
if (
|
||
hasattr(self, "read_thread")
|
||
and self.read_thread
|
||
and not self.read_thread.isFinished()
|
||
):
|
||
self.read_thread.set_selected_forms(list())
|
||
|
||
# bind
|
||
confirm_btn.clicked.connect(onDialogConfirm)
|
||
cancel_btn.clicked.connect(dialog.reject)
|
||
dialog.finished.connect(onDialogClosed)
|
||
|
||
# 显示表单选择框
|
||
dialog.exec()
|
||
|
||
# 由读取的数据生成新表单
|
||
def generateNewForms(self, form_data):
|
||
"""根据读取到的数据生成新表单(UI线程处理)"""
|
||
# form_data格式：{form_name: [(row_idx, x, y, z, rx, ry, rz), ...], ...}
|
||
for form_name, data_rows in form_data.items():
|
||
# 创建新表单
|
||
self.formTotalCnt += 1 # 创建的总的表单数加一
|
||
new_form = CoordinateTableWidget(
|
||
form_name=f"{form_name}", parent=self, initRowCount=False
|
||
)
|
||
new_form.update_table_signal.connect(self.handleFormUpdate)
|
||
new_form.move_to_coodinate_signal.connect(self.form_move_signal)
|
||
|
||
# 填充数据到新表单
|
||
for row_idx, x, y, z, rx, ry, rz in data_rows:
|
||
# 如果数单行数超过表格行数，自动添加行
|
||
while row_idx >= new_form.table.rowCount():
|
||
new_form.addNewRow()
|
||
|
||
# 填充单元格
|
||
new_form.table.setItem(row_idx, 0, QTableWidgetItem(x))
|
||
new_form.table.setItem(row_idx, 1, QTableWidgetItem(y))
|
||
new_form.table.setItem(row_idx, 2, QTableWidgetItem(z))
|
||
new_form.table.setItem(row_idx, 3, QTableWidgetItem(rx))
|
||
new_form.table.setItem(row_idx, 4, QTableWidgetItem(ry))
|
||
new_form.table.setItem(row_idx, 5, QTableWidgetItem(rz))
|
||
# 将新表单添加到UI
|
||
form_key = f"form_{self.formTotalCnt}" # 唯一标识
|
||
self.tabBar.addTab(
|
||
routeKey=form_key, text=f"{form_name}", icon=FIF.TILES.icon()
|
||
)
|
||
self.formStack.addWidget(new_form)
|
||
|
||
|
||
# 更新表单中选中行数据 x,y,z,rx,ry,rz
|
||
# def test(form_obj: CoordinateTableWidget):
|
||
# form_obj.update_table_data([666, 666, 666, 666, 666, 666])
|
||
|
||
|
||
# if __name__ == "__main__":
|
||
# app = QApplication(sys.argv)
|
||
# setTheme(Theme.DARK)
|
||
# window = CoordinateFormsWidget()
|
||
# window.form_update_signal.connect(test)
|
||
# window.show()
|
||
# sys.exit(app.exec())
|