446 lines
16 KiB
Python
446 lines
16 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
LLMClipboard 通知样式测试脚本
|
||
用于测试各种通知样式、颜色和交互
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
import logging
|
||
import traceback
|
||
|
||
# 添加项目根目录到路径
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
# 配置日志
|
||
logging.basicConfig(
|
||
level=logging.DEBUG,
|
||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||
handlers=[
|
||
logging.StreamHandler(),
|
||
logging.FileHandler('notification_style_test.log', encoding='utf-8')
|
||
]
|
||
)
|
||
|
||
logger = logging.getLogger('notification_style_test')
|
||
|
||
try:
|
||
from PyQt6.QtWidgets import (
|
||
QApplication, QMainWindow, QPushButton, QVBoxLayout, QWidget, QLabel,
|
||
QHBoxLayout, QComboBox, QLineEdit, QTextEdit, QGroupBox, QSlider, QCheckBox,
|
||
QColorDialog, QDialog
|
||
)
|
||
from PyQt6.QtCore import Qt, QTimer
|
||
from PyQt6.QtGui import QColor
|
||
|
||
# 导入LLMClipboard模块
|
||
from llmclipboard.gui import NotificationDialog
|
||
|
||
# 尝试导入apply_stylesheet函数
|
||
try:
|
||
from qt_material import apply_stylesheet
|
||
except ImportError:
|
||
# 如果qt_material未安装，创建一个空的apply_stylesheet函数
|
||
def apply_stylesheet(widget, theme=None):
|
||
logger.warning(f"qt_material未安装，无法应用主题: {theme}")
|
||
except Exception as e:
|
||
logger.error(f"导入模块时出错: {str(e)}")
|
||
traceback.print_exc()
|
||
sys.exit(1)
|
||
|
||
# 创建一个简化版的通知对话框，直接使用NotificationDialog类
|
||
class SimpleNotificationTester(QMainWindow):
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.setWindowTitle("LLMClipboard 通知样式测试")
|
||
self.resize(800, 600)
|
||
|
||
# 初始化颜色
|
||
self.bg_color = QColor("#f0f0f0")
|
||
self.title_color = QColor("#333333")
|
||
self.content_color = QColor("#555555")
|
||
|
||
# 创建通知队列
|
||
self.notification_queue = []
|
||
self.is_showing_notification = False
|
||
|
||
# 创建UI
|
||
self.setup_ui()
|
||
|
||
def setup_ui(self):
|
||
"""设置UI"""
|
||
central_widget = QWidget()
|
||
self.setCentralWidget(central_widget)
|
||
|
||
main_layout = QVBoxLayout(central_widget)
|
||
|
||
# 通知类型选择
|
||
type_group = QGroupBox("通知类型")
|
||
type_layout = QHBoxLayout()
|
||
type_group.setLayout(type_layout)
|
||
|
||
type_label = QLabel("类型:")
|
||
self.type_combo = QComboBox()
|
||
self.type_combo.addItems(["info", "success", "warning", "error"])
|
||
|
||
type_layout.addWidget(type_label)
|
||
type_layout.addWidget(self.type_combo)
|
||
|
||
# 通知内容
|
||
content_group = QGroupBox("通知内容")
|
||
content_layout = QVBoxLayout()
|
||
content_group.setLayout(content_layout)
|
||
|
||
title_layout = QHBoxLayout()
|
||
title_label = QLabel("标题:")
|
||
self.title_edit = QLineEdit("测试通知")
|
||
title_layout.addWidget(title_label)
|
||
title_layout.addWidget(self.title_edit)
|
||
|
||
message_layout = QVBoxLayout()
|
||
message_label = QLabel("消息内容:")
|
||
self.message_edit = QTextEdit("这是一条测试通知消息，用于测试LLMClipboard的通知系统。")
|
||
self.message_edit.setMaximumHeight(100)
|
||
message_layout.addWidget(message_label)
|
||
message_layout.addWidget(self.message_edit)
|
||
|
||
buttons_layout = QHBoxLayout()
|
||
buttons_label = QLabel("按钮(逗号分隔):")
|
||
self.buttons_edit = QLineEdit("确定,取消")
|
||
buttons_layout.addWidget(buttons_label)
|
||
buttons_layout.addWidget(self.buttons_edit)
|
||
|
||
content_layout.addLayout(title_layout)
|
||
content_layout.addLayout(message_layout)
|
||
content_layout.addLayout(buttons_layout)
|
||
|
||
# 通知设置
|
||
settings_group = QGroupBox("通知设置")
|
||
settings_layout = QVBoxLayout()
|
||
settings_group.setLayout(settings_layout)
|
||
|
||
duration_layout = QHBoxLayout()
|
||
duration_label = QLabel("持续时间(秒):")
|
||
self.duration_slider = QSlider(Qt.Orientation.Horizontal)
|
||
self.duration_slider.setMinimum(1)
|
||
self.duration_slider.setMaximum(10)
|
||
self.duration_slider.setValue(5)
|
||
self.duration_value_label = QLabel("5")
|
||
self.duration_slider.valueChanged.connect(lambda v: self.duration_value_label.setText(str(v)))
|
||
duration_layout.addWidget(duration_label)
|
||
duration_layout.addWidget(self.duration_slider)
|
||
duration_layout.addWidget(self.duration_value_label)
|
||
|
||
auto_close_layout = QHBoxLayout()
|
||
self.auto_close_check = QCheckBox("自动关闭")
|
||
self.auto_close_check.setChecked(True)
|
||
auto_close_layout.addWidget(self.auto_close_check)
|
||
|
||
settings_layout.addLayout(duration_layout)
|
||
settings_layout.addLayout(auto_close_layout)
|
||
|
||
# 样式设置
|
||
style_group = QGroupBox("样式设置")
|
||
style_layout = QVBoxLayout()
|
||
style_group.setLayout(style_layout)
|
||
|
||
bg_color_layout = QHBoxLayout()
|
||
bg_color_label = QLabel("背景颜色:")
|
||
self.bg_color_button = QPushButton()
|
||
self.bg_color_button.setStyleSheet(f"background-color: {self.bg_color.name()};")
|
||
self.bg_color_button.setFixedSize(30, 20)
|
||
self.bg_color_button.clicked.connect(lambda: self.choose_color("bg"))
|
||
bg_color_layout.addWidget(bg_color_label)
|
||
bg_color_layout.addWidget(self.bg_color_button)
|
||
bg_color_layout.addStretch()
|
||
|
||
title_color_layout = QHBoxLayout()
|
||
title_color_label = QLabel("标题颜色:")
|
||
self.title_color_button = QPushButton()
|
||
self.title_color_button.setStyleSheet(f"background-color: {self.title_color.name()};")
|
||
self.title_color_button.setFixedSize(30, 20)
|
||
self.title_color_button.clicked.connect(lambda: self.choose_color("title"))
|
||
title_color_layout.addWidget(title_color_label)
|
||
title_color_layout.addWidget(self.title_color_button)
|
||
title_color_layout.addStretch()
|
||
|
||
content_color_layout = QHBoxLayout()
|
||
content_color_label = QLabel("内容颜色:")
|
||
self.content_color_button = QPushButton()
|
||
self.content_color_button.setStyleSheet(f"background-color: {self.content_color.name()};")
|
||
self.content_color_button.setFixedSize(30, 20)
|
||
self.content_color_button.clicked.connect(lambda: self.choose_color("content"))
|
||
content_color_layout.addWidget(content_color_label)
|
||
content_color_layout.addWidget(self.content_color_button)
|
||
content_color_layout.addStretch()
|
||
|
||
style_layout.addLayout(bg_color_layout)
|
||
style_layout.addLayout(title_color_layout)
|
||
style_layout.addLayout(content_color_layout)
|
||
|
||
# 测试按钮
|
||
buttons_layout = QHBoxLayout()
|
||
|
||
test_button = QPushButton("测试通知")
|
||
test_button.clicked.connect(self.test_notification)
|
||
|
||
queue_button = QPushButton("测试通知队列")
|
||
queue_button.clicked.connect(self.test_notification_queue)
|
||
|
||
dialog_button = QPushButton("测试直接对话框")
|
||
dialog_button.clicked.connect(self.test_direct_dialog)
|
||
|
||
buttons_layout.addWidget(test_button)
|
||
buttons_layout.addWidget(queue_button)
|
||
buttons_layout.addWidget(dialog_button)
|
||
|
||
# 状态标签
|
||
self.status_label = QLabel("就绪")
|
||
self.status_label.setStyleSheet("color: blue;")
|
||
|
||
# 添加所有组件到主布局
|
||
main_layout.addWidget(type_group)
|
||
main_layout.addWidget(content_group)
|
||
main_layout.addWidget(settings_group)
|
||
main_layout.addWidget(style_group)
|
||
main_layout.addLayout(buttons_layout)
|
||
main_layout.addWidget(self.status_label)
|
||
|
||
def choose_color(self, target):
|
||
"""选择颜色"""
|
||
try:
|
||
if target == "bg":
|
||
color = QColorDialog.getColor(self.bg_color, self)
|
||
if color.isValid():
|
||
self.bg_color = color
|
||
self.bg_color_button.setStyleSheet(f"background-color: {color.name()};")
|
||
elif target == "title":
|
||
color = QColorDialog.getColor(self.title_color, self)
|
||
if color.isValid():
|
||
self.title_color = color
|
||
self.title_color_button.setStyleSheet(f"background-color: {color.name()};")
|
||
elif target == "content":
|
||
color = QColorDialog.getColor(self.content_color, self)
|
||
if color.isValid():
|
||
self.content_color = color
|
||
self.content_color_button.setStyleSheet(f"background-color: {color.name()};")
|
||
except Exception as e:
|
||
logger.error(f"选择颜色时出错: {str(e)}")
|
||
|
||
def update_status(self, message):
|
||
"""更新状态标签"""
|
||
try:
|
||
self.status_label.setText(message)
|
||
logger.info(message)
|
||
# 5秒后重置状态
|
||
QTimer.singleShot(5000, lambda: self.status_label.setText("就绪"))
|
||
except Exception as e:
|
||
logger.error(f"更新状态标签时出错: {str(e)}")
|
||
|
||
def get_notification_params(self):
|
||
"""获取通知参数"""
|
||
try:
|
||
notification_type = self.type_combo.currentText()
|
||
title = self.title_edit.text()
|
||
message = self.message_edit.toPlainText()
|
||
buttons = [btn.strip() for btn in self.buttons_edit.text().split(",") if btn.strip()]
|
||
duration = self.duration_slider.value() * 1000 # 转换为毫秒
|
||
auto_close = self.auto_close_check.isChecked()
|
||
|
||
return {
|
||
"type": notification_type,
|
||
"title": title,
|
||
"message": message,
|
||
"buttons": buttons,
|
||
"duration": duration,
|
||
"auto_close": auto_close
|
||
}
|
||
except Exception as e:
|
||
logger.error(f"获取通知参数时出错: {str(e)}")
|
||
return None
|
||
|
||
def show_notification(self, title, message, notification_type="info", buttons=None, file_path=None, duration=0):
|
||
"""显示通知"""
|
||
try:
|
||
# 将通知添加到队列
|
||
self.notification_queue.append({
|
||
"title": title,
|
||
"message": message,
|
||
"type": notification_type,
|
||
"buttons": buttons,
|
||
"file_path": file_path,
|
||
"duration": duration
|
||
})
|
||
|
||
# 如果当前没有显示通知，则开始处理队列
|
||
if not self.is_showing_notification:
|
||
self._process_notification_queue()
|
||
except Exception as e:
|
||
logger.error(f"添加通知到队列时出错: {str(e)}")
|
||
|
||
def _process_notification_queue(self):
|
||
"""处理通知队列"""
|
||
try:
|
||
if not self.notification_queue:
|
||
self.is_showing_notification = False
|
||
return
|
||
|
||
self.is_showing_notification = True
|
||
notification = self.notification_queue.pop(0)
|
||
|
||
# 创建通知对话框
|
||
dialog = NotificationDialog(
|
||
self,
|
||
notification["title"],
|
||
notification["message"],
|
||
notification["type"],
|
||
notification["buttons"],
|
||
notification["file_path"]
|
||
)
|
||
|
||
# 应用自定义样式
|
||
style = f"""
|
||
QDialog {{
|
||
background-color: {self.bg_color.name()};
|
||
border: 1px solid #cccccc;
|
||
border-radius: 5px;
|
||
}}
|
||
QLabel#title_label {{
|
||
color: {self.title_color.name()};
|
||
font-size: 14px;
|
||
font-weight: bold;
|
||
}}
|
||
QLabel#message_label {{
|
||
color: {self.content_color.name()};
|
||
}}
|
||
"""
|
||
dialog.setStyleSheet(style)
|
||
|
||
# 显示对话框
|
||
dialog.show()
|
||
|
||
# 如果需要自动关闭
|
||
if notification["duration"] > 0:
|
||
QTimer.singleShot(notification["duration"], dialog.close)
|
||
|
||
# 当对话框关闭时，处理下一个通知
|
||
dialog.finished.connect(self._process_notification_queue)
|
||
except Exception as e:
|
||
logger.error(f"处理通知队列时出错: {str(e)}")
|
||
self.is_showing_notification = False
|
||
|
||
def test_notification(self):
|
||
"""测试通知"""
|
||
try:
|
||
params = self.get_notification_params()
|
||
if not params:
|
||
self.update_status("获取通知参数失败")
|
||
return
|
||
|
||
self.update_status(f"正在测试通知: {params['title']}")
|
||
|
||
self.show_notification(
|
||
params["title"],
|
||
params["message"],
|
||
params["type"],
|
||
params["buttons"] if params["buttons"] else None,
|
||
None, # 文件路径
|
||
params["duration"] if params["auto_close"] else 0
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"测试通知时出错: {str(e)}")
|
||
self.update_status(f"显示通知失败: {e}")
|
||
|
||
def test_notification_queue(self):
|
||
"""测试通知队列"""
|
||
try:
|
||
params = self.get_notification_params()
|
||
if not params:
|
||
self.update_status("获取通知参数失败")
|
||
return
|
||
|
||
self.update_status("正在测试通知队列...")
|
||
|
||
# 发送多个通知
|
||
for i in range(3):
|
||
try:
|
||
self.show_notification(
|
||
f"{params['title']} {i+1}",
|
||
f"{params['message']}\n这是第 {i+1} 个队列通知",
|
||
params["type"],
|
||
params["buttons"] if params["buttons"] else None,
|
||
None, # 文件路径
|
||
params["duration"] if params["auto_close"] else 0
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"显示通知队列失败: {e}")
|
||
self.update_status(f"显示通知队列失败: {e}")
|
||
except Exception as e:
|
||
logger.error(f"测试通知队列时出错: {str(e)}")
|
||
|
||
def test_direct_dialog(self):
|
||
"""测试直接创建通知对话框"""
|
||
try:
|
||
params = self.get_notification_params()
|
||
if not params:
|
||
self.update_status("获取通知参数失败")
|
||
return
|
||
|
||
self.update_status("正在测试直接创建通知对话框...")
|
||
|
||
# 创建自定义样式
|
||
style = f"""
|
||
QDialog {{
|
||
background-color: {self.bg_color.name()};
|
||
border: 1px solid #cccccc;
|
||
border-radius: 5px;
|
||
}}
|
||
QLabel#title_label {{
|
||
color: {self.title_color.name()};
|
||
font-size: 14px;
|
||
font-weight: bold;
|
||
}}
|
||
QLabel#message_label {{
|
||
color: {self.content_color.name()};
|
||
}}
|
||
"""
|
||
|
||
# 创建通知对话框
|
||
dialog = NotificationDialog(
|
||
self,
|
||
params["title"],
|
||
params["message"],
|
||
params["type"],
|
||
params["buttons"] if params["buttons"] else None,
|
||
None # 文件路径
|
||
)
|
||
|
||
# 应用自定义样式
|
||
dialog.setStyleSheet(style)
|
||
|
||
# 显示对话框
|
||
dialog.show()
|
||
|
||
# 如果需要自动关闭
|
||
if params["auto_close"]:
|
||
QTimer.singleShot(params["duration"], dialog.close)
|
||
except Exception as e:
|
||
logger.error(f"测试直接创建通知对话框时出错: {str(e)}")
|
||
self.update_status(f"创建通知对话框失败: {e}")
|
||
|
||
def main():
|
||
try:
|
||
app = QApplication(sys.argv)
|
||
window = SimpleNotificationTester()
|
||
window.show()
|
||
sys.exit(app.exec())
|
||
except Exception as e:
|
||
logger.error(f"主函数执行时出错: {str(e)}")
|
||
traceback.print_exc()
|
||
sys.exit(1)
|
||
|
||
if __name__ == "__main__":
|
||
main()
|