262 lines
9.2 KiB
Python
262 lines
9.2 KiB
Python
from PyQt5.QtWidgets import (QDialog, QVBoxLayout, QHBoxLayout, QLabel,
|
||
QComboBox, QPushButton, QTextEdit, QFrame,
|
||
QScrollArea, QWidget)
|
||
from PyQt5.QtCore import Qt
|
||
from PyQt5.QtGui import QColor, QTextCharFormat, QTextCursor
|
||
import difflib
|
||
|
||
class VersionCompareDialog(QDialog):
|
||
"""版本比较对话框"""
|
||
|
||
def __init__(self, versions, parent=None):
|
||
super().__init__(parent)
|
||
# 按联系人分组版本
|
||
self.contact_versions = self.group_versions_by_contact(versions)
|
||
self.init_ui()
|
||
self.apply_style()
|
||
|
||
def group_versions_by_contact(self, versions):
|
||
"""将版本按联系人分组
|
||
Returns:
|
||
Dict[str, List[Dict]]: 以联系人wxid为键，版本列表为值的字典
|
||
"""
|
||
grouped = {}
|
||
for version in versions:
|
||
contact = version['contact']
|
||
wxid = contact['wxid']
|
||
if wxid not in grouped:
|
||
grouped[wxid] = []
|
||
grouped[wxid].append(version)
|
||
return grouped
|
||
|
||
def init_ui(self):
|
||
"""初始化UI"""
|
||
self.setWindowTitle("版本比较")
|
||
self.setMinimumSize(800, 600)
|
||
|
||
layout = QVBoxLayout(self)
|
||
|
||
# 联系人选择
|
||
contact_layout = QHBoxLayout()
|
||
contact_layout.addWidget(QLabel("选择联系人："))
|
||
self.contact_combo = QComboBox()
|
||
self.contact_combo.currentIndexChanged.connect(self.on_contact_changed)
|
||
contact_layout.addWidget(self.contact_combo)
|
||
layout.addLayout(contact_layout)
|
||
|
||
# 版本选择区域
|
||
select_layout = QHBoxLayout()
|
||
|
||
# 版本A选择
|
||
version_a_layout = QVBoxLayout()
|
||
version_a_layout.addWidget(QLabel("版本A："))
|
||
self.version_a_combo = QComboBox()
|
||
self.version_a_combo.currentIndexChanged.connect(self.compare_versions)
|
||
version_a_layout.addWidget(self.version_a_combo)
|
||
select_layout.addLayout(version_a_layout)
|
||
|
||
# 版本B选择
|
||
version_b_layout = QVBoxLayout()
|
||
version_b_layout.addWidget(QLabel("版本B："))
|
||
self.version_b_combo = QComboBox()
|
||
self.version_b_combo.currentIndexChanged.connect(self.compare_versions)
|
||
version_b_layout.addWidget(self.version_b_combo)
|
||
select_layout.addLayout(version_b_layout)
|
||
|
||
layout.addLayout(select_layout)
|
||
|
||
# 比较结果区域
|
||
scroll = QScrollArea()
|
||
scroll.setWidgetResizable(True)
|
||
scroll.setObjectName("compare-scroll")
|
||
|
||
content_widget = QWidget()
|
||
self.content_layout = QVBoxLayout(content_widget)
|
||
|
||
# 各部分内容比较
|
||
self.add_compare_section("新年祝福语", "greeting")
|
||
self.add_compare_section("美好祝愿", "wishes")
|
||
self.add_compare_section("祝福成语", "idioms")
|
||
self.add_compare_section("新年图片提示词", "image_prompt")
|
||
|
||
scroll.setWidget(content_widget)
|
||
layout.addWidget(scroll)
|
||
|
||
# 底部按钮
|
||
button_layout = QHBoxLayout()
|
||
close_btn = QPushButton("关闭")
|
||
close_btn.setObjectName("close-btn")
|
||
close_btn.clicked.connect(self.accept)
|
||
button_layout.addStretch()
|
||
button_layout.addWidget(close_btn)
|
||
layout.addLayout(button_layout)
|
||
|
||
# 初始化联系人列表
|
||
self.init_contacts()
|
||
|
||
def init_contacts(self):
|
||
"""初始化联系人列表"""
|
||
self.contact_combo.clear()
|
||
for wxid, versions in self.contact_versions.items():
|
||
if versions: # 确保有版本数据
|
||
contact_name = versions[0]['contact']['name'] # 使用第一个版本中的联系人名称
|
||
self.contact_combo.addItem(contact_name, wxid) # 使用wxid作为用户数据
|
||
|
||
def on_contact_changed(self, index):
|
||
"""联系人选择改变时的处理"""
|
||
if index < 0:
|
||
return
|
||
|
||
# 获取选中联系人的wxid
|
||
wxid = self.contact_combo.currentData()
|
||
versions = self.contact_versions.get(wxid, [])
|
||
|
||
# 更新版本选择框
|
||
self.version_a_combo.clear()
|
||
self.version_b_combo.clear()
|
||
|
||
contact_name = versions[0]['contact']['name']
|
||
for version in versions:
|
||
version_text = f"{contact_name} - 版本{version['version_number']} ({version['create_time']})"
|
||
self.version_a_combo.addItem(version_text, version)
|
||
self.version_b_combo.addItem(version_text, version)
|
||
|
||
# 默认选择最新的两个版本
|
||
if len(versions) >= 2:
|
||
self.version_a_combo.setCurrentIndex(len(versions) - 2)
|
||
self.version_b_combo.setCurrentIndex(len(versions) - 1)
|
||
|
||
def add_compare_section(self, title, content_key):
|
||
"""添加比较部分"""
|
||
section = QWidget()
|
||
layout = QVBoxLayout(section)
|
||
|
||
# 标题
|
||
title_label = QLabel(title)
|
||
title_label.setObjectName("title")
|
||
layout.addWidget(title_label)
|
||
|
||
# 内容比较区域
|
||
content_layout = QHBoxLayout()
|
||
|
||
# 版本A内容
|
||
text_a = QTextEdit()
|
||
text_a.setReadOnly(True)
|
||
setattr(self, f"{content_key}_a", text_a)
|
||
content_layout.addWidget(text_a)
|
||
|
||
# 版本B内容
|
||
text_b = QTextEdit()
|
||
text_b.setReadOnly(True)
|
||
setattr(self, f"{content_key}_b", text_b)
|
||
content_layout.addWidget(text_b)
|
||
|
||
layout.addLayout(content_layout)
|
||
self.content_layout.addWidget(section)
|
||
|
||
def compare_versions(self):
|
||
"""比较选中的两个版本"""
|
||
if self.version_a_combo.currentIndex() < 0 or self.version_b_combo.currentIndex() < 0:
|
||
return
|
||
|
||
version_a = self.version_a_combo.currentData()
|
||
version_b = self.version_b_combo.currentData()
|
||
|
||
if version_a and version_b:
|
||
self.compare_content("greeting", version_a, version_b)
|
||
self.compare_content("wishes", version_a, version_b)
|
||
self.compare_content("idioms", version_a, version_b)
|
||
self.compare_content("image_prompt", version_a, version_b)
|
||
|
||
def compare_content(self, content_key, version_a, version_b):
|
||
"""比较特定内容"""
|
||
text_a = version_a.get(content_key, '')
|
||
text_b = version_b.get(content_key, '')
|
||
|
||
# 获取文本框引用
|
||
text_edit_a = getattr(self, f"{content_key}_a")
|
||
text_edit_b = getattr(self, f"{content_key}_b")
|
||
|
||
# 清空文本框
|
||
text_edit_a.clear()
|
||
text_edit_b.clear()
|
||
|
||
# 计算差异
|
||
differ = difflib.Differ()
|
||
diff = list(differ.compare(text_a.splitlines(), text_b.splitlines()))
|
||
|
||
# 设置差异高亮格式
|
||
add_format = QTextCharFormat()
|
||
add_format.setBackground(QColor(200, 255, 200)) # 浅绿色背景
|
||
|
||
del_format = QTextCharFormat()
|
||
del_format.setBackground(QColor(255, 200, 200)) # 浅红色背景
|
||
|
||
# 显示差异
|
||
cursor_a = text_edit_a.textCursor()
|
||
cursor_b = text_edit_b.textCursor()
|
||
|
||
for line in diff:
|
||
if line.startswith('- '):
|
||
cursor_a.insertText(line[2:] + '\n', del_format)
|
||
cursor_b.insertText('\n')
|
||
elif line.startswith('+ '):
|
||
cursor_a.insertText('\n')
|
||
cursor_b.insertText(line[2:] + '\n', add_format)
|
||
else:
|
||
cursor_a.insertText(line[2:] + '\n')
|
||
cursor_b.insertText(line[2:] + '\n')
|
||
|
||
# 滚动到顶部
|
||
text_edit_a.verticalScrollBar().setValue(0)
|
||
text_edit_b.verticalScrollBar().setValue(0)
|
||
|
||
def apply_style(self):
|
||
"""应用样式"""
|
||
self.setStyleSheet("""
|
||
QDialog {
|
||
background-color: #F9F4E6;
|
||
min-width: 800px;
|
||
min-height: 600px;
|
||
}
|
||
QLabel {
|
||
color: #333333;
|
||
}
|
||
QLabel#title {
|
||
font-size: 18px;
|
||
font-weight: bold;
|
||
color: #D4341F;
|
||
padding: 10px;
|
||
}
|
||
QLabel#version-label {
|
||
font-size: 16px;
|
||
font-weight: bold;
|
||
color: #D4341F;
|
||
padding: 5px;
|
||
}
|
||
QTextEdit {
|
||
background-color: white;
|
||
border: 1px solid #E8C06F;
|
||
border-radius: 4px;
|
||
padding: 8px;
|
||
font-size: 14px;
|
||
}
|
||
QPushButton {
|
||
background-color: #E8C06F;
|
||
color: #333333;
|
||
border: none;
|
||
padding: 8px 16px;
|
||
border-radius: 4px;
|
||
min-width: 80px;
|
||
}
|
||
QPushButton:hover {
|
||
background-color: #D4B05F;
|
||
}
|
||
QPushButton#close-btn {
|
||
background-color: #D4341F;
|
||
color: white;
|
||
}
|
||
QPushButton#close-btn:hover {
|
||
background-color: #A61F14;
|
||
}
|
||
""") |