镜像自地址
https://github.com/a13zhy/a13rollcall.git
已同步 2026-10-22 03:55:41 +08:00
863 行
32 KiB
Python
863 行
32 KiB
Python
"""
|
||
A13课堂点名系统 - 插件UI注入工具（增强版）
|
||
提供在主程序任何位置添加UI元素、修改布局、添加标签页的完整能力
|
||
"""
|
||
|
||
from PyQt5.QtWidgets import (
|
||
QPushButton, QWidget, QHBoxLayout, QVBoxLayout, QLabel, QFrame,
|
||
QSizePolicy, QMenu, QAction, QLineEdit, QComboBox, QCheckBox,
|
||
QSlider, QProgressBar, QTextEdit, QListWidget, QListWidgetItem,
|
||
QTabWidget, QScrollArea, QGroupBox, QFormLayout, QGridLayout,
|
||
QSplitter, QToolBar, QStatusBar, QDockWidget, QMessageBox,
|
||
QFileDialog, QColorDialog, QFontDialog, QInputDialog, QSpinBox,
|
||
QDoubleSpinBox, QDateEdit, QTimeEdit, QDateTimeEdit, QCalendarWidget,
|
||
QDial, QLCDNumber, QTableWidget, QTableWidgetItem, QTreeWidget,
|
||
QTreeWidgetItem, QHeaderView, QStackedWidget, QToolBox, QButtonGroup,
|
||
QRadioButton, QPlainTextEdit, QGraphicsView, QGraphicsScene,
|
||
QSystemTrayIcon, QMenuBar, QShortcut
|
||
)
|
||
from PyQt5.QtCore import Qt, QSize, QTimer, pyqtSignal, QObject, QRect
|
||
from PyQt5.QtGui import (
|
||
QFont, QIcon, QColor, QPalette, QPixmap, QPainter, QPen, QBrush,
|
||
QLinearGradient, QRadialGradient, QFontDatabase, QCursor, QKeySequence
|
||
)
|
||
import os
|
||
import json
|
||
|
||
|
||
class UIInjector:
|
||
"""UI注入工具类 - 增强版，支持复杂控件、布局修改、标签页添加"""
|
||
|
||
def __init__(self, main_window):
|
||
self.main_window = main_window
|
||
self.injected_elements = []
|
||
self._custom_tabs = []
|
||
self._custom_panels = []
|
||
|
||
# ==================== 查找控件 ====================
|
||
|
||
def find_widget_by_name(self, name):
|
||
return self.main_window.findChild(QWidget, name)
|
||
|
||
def find_all_buttons(self):
|
||
return self.main_window.findChildren(QPushButton)
|
||
|
||
def find_button_by_text(self, text):
|
||
for btn in self.find_all_buttons():
|
||
if btn.text() == text:
|
||
return btn
|
||
return None
|
||
|
||
def find_all_labels(self):
|
||
return self.main_window.findChildren(QLabel)
|
||
|
||
def find_label_by_text(self, text):
|
||
for label in self.find_all_labels():
|
||
if text in label.text():
|
||
return label
|
||
return None
|
||
|
||
def find_layout_by_widget(self, widget):
|
||
if widget and widget.parent():
|
||
return widget.parent().layout()
|
||
return None
|
||
|
||
def get_all_ui_elements(self, include_layouts=True):
|
||
elements = []
|
||
for widget in self.main_window.findChildren(QWidget):
|
||
elements.append({
|
||
"type": type(widget).__name__,
|
||
"objectName": widget.objectName(),
|
||
"text": getattr(widget, 'text', lambda: '')() if callable(getattr(widget, 'text', None)) else '',
|
||
"visible": widget.isVisible(),
|
||
"enabled": widget.isEnabled(),
|
||
"geometry": widget.geometry().getRect(),
|
||
})
|
||
return elements
|
||
|
||
def get_main_window(self):
|
||
return self.main_window
|
||
|
||
def get_central_widget(self):
|
||
return self.main_window.centralWidget()
|
||
|
||
def get_status_bar(self):
|
||
return self.main_window.statusBar()
|
||
|
||
def get_menu_bar(self):
|
||
return self.main_window.menuBar()
|
||
|
||
# ==================== 添加按钮（基础） ====================
|
||
|
||
def add_button_to_layout(self, layout, text, callback, position=None,
|
||
style="primary", icon=None, tooltip=None,
|
||
width=None, height=None):
|
||
btn = QPushButton(text)
|
||
btn.setStyleSheet(self._get_button_style(style))
|
||
if icon:
|
||
btn.setIcon(icon)
|
||
if tooltip:
|
||
btn.setToolTip(tooltip)
|
||
if width:
|
||
btn.setFixedWidth(width)
|
||
if height:
|
||
btn.setFixedHeight(height)
|
||
btn.clicked.connect(callback)
|
||
if position is not None:
|
||
layout.insertWidget(position, btn)
|
||
else:
|
||
layout.addWidget(btn)
|
||
self.injected_elements.append(btn)
|
||
return btn
|
||
|
||
def add_button_to_top_bar(self, text, callback, position=None, style="primary",
|
||
icon=None, tooltip=None, width=100, height=36):
|
||
top_bar = self._get_top_bar()
|
||
if not top_bar:
|
||
return None
|
||
layout = top_bar.layout()
|
||
if not layout:
|
||
layout = QHBoxLayout(top_bar)
|
||
return self.add_button_to_layout(layout, text, callback, position, style, icon, tooltip, width, height)
|
||
|
||
def add_button_to_bottom_bar(self, text, callback, position=None, style="ghost",
|
||
icon=None, tooltip=None, width=90, height=32):
|
||
bottom_bar = self._get_bottom_bar()
|
||
if not bottom_bar:
|
||
return None
|
||
if isinstance(bottom_bar, QStatusBar):
|
||
btn = QPushButton(text)
|
||
btn.setStyleSheet(self._get_button_style(style))
|
||
if tooltip:
|
||
btn.setToolTip(tooltip)
|
||
btn.clicked.connect(callback)
|
||
bottom_bar.addPermanentWidget(btn)
|
||
self.injected_elements.append(btn)
|
||
return btn
|
||
layout = bottom_bar.layout()
|
||
if not layout:
|
||
layout = QHBoxLayout(bottom_bar)
|
||
return self.add_button_to_layout(layout, text, callback, position, style, icon, tooltip, width, height)
|
||
|
||
def add_button_to_sidebar(self, text, callback, position=None, style="ghost",
|
||
icon=None, tooltip=None, width=120, height=40):
|
||
sidebar = self._get_sidebar()
|
||
if not sidebar:
|
||
return None
|
||
layout = sidebar.layout()
|
||
if not layout:
|
||
layout = QVBoxLayout(sidebar)
|
||
return self.add_button_to_layout(layout, text, callback, position, style, icon, tooltip, width, height)
|
||
|
||
def add_button_to_draw_area(self, text, callback, position=None, style="primary",
|
||
icon=None, tooltip=None, width=120, height=48):
|
||
draw_area = self._get_draw_area()
|
||
if not draw_area:
|
||
return None
|
||
layout = draw_area.layout()
|
||
if not layout:
|
||
layout = QVBoxLayout(draw_area)
|
||
return self.add_button_to_layout(layout, text, callback, position, style, icon, tooltip, width, height)
|
||
|
||
def add_button_next_to(self, target_button, text, callback, side="right",
|
||
style="success", icon=None, tooltip=None, width=None, height=None):
|
||
if isinstance(target_button, str):
|
||
target_button = self.find_button_by_text(target_button)
|
||
if not target_button:
|
||
return None
|
||
parent_layout = self.find_layout_by_widget(target_button)
|
||
if not parent_layout:
|
||
return None
|
||
index = parent_layout.indexOf(target_button)
|
||
position = index + 1 if side == "right" else index
|
||
return self.add_button_to_layout(parent_layout, text, callback, position, style, icon, tooltip, width, height)
|
||
|
||
def create_floating_button(self, text, callback, x=20, y=100,
|
||
size=(100, 40), style="warning", tooltip=None):
|
||
btn = DraggableButton(text, self.main_window)
|
||
btn.setStyleSheet(self._get_button_style(style))
|
||
btn.setGeometry(x, y, size[0], size[1])
|
||
if tooltip:
|
||
btn.setToolTip(tooltip)
|
||
btn.clicked.connect(callback)
|
||
btn.show()
|
||
self.injected_elements.append(btn)
|
||
return btn
|
||
|
||
# ==================== 添加复杂控件（新增） ====================
|
||
|
||
def add_input_to_layout(self, layout, placeholder="", callback=None,
|
||
position=None, width=200, height=32, echo_normal=True):
|
||
input_box = QLineEdit()
|
||
input_box.setPlaceholderText(placeholder)
|
||
input_box.setFixedHeight(height)
|
||
if width:
|
||
input_box.setFixedWidth(width)
|
||
if not echo_normal:
|
||
input_box.setEchoMode(QLineEdit.Password)
|
||
if callback:
|
||
input_box.textChanged.connect(callback)
|
||
if position is not None:
|
||
layout.insertWidget(position, input_box)
|
||
else:
|
||
layout.addWidget(input_box)
|
||
self.injected_elements.append(input_box)
|
||
return input_box
|
||
|
||
def add_combo_to_layout(self, layout, items, callback=None, position=None,
|
||
width=150, height=32, current_index=0):
|
||
combo = QComboBox()
|
||
combo.addItems(items)
|
||
combo.setFixedHeight(height)
|
||
if width:
|
||
combo.setFixedWidth(width)
|
||
combo.setCurrentIndex(current_index)
|
||
if callback:
|
||
combo.currentIndexChanged.connect(callback)
|
||
if position is not None:
|
||
layout.insertWidget(position, combo)
|
||
else:
|
||
layout.addWidget(combo)
|
||
self.injected_elements.append(combo)
|
||
return combo
|
||
|
||
def add_checkbox_to_layout(self, layout, text, callback=None, position=None,
|
||
checked=False, width=None):
|
||
checkbox = QCheckBox(text)
|
||
checkbox.setChecked(checked)
|
||
if width:
|
||
checkbox.setFixedWidth(width)
|
||
if callback:
|
||
checkbox.stateChanged.connect(callback)
|
||
if position is not None:
|
||
layout.insertWidget(position, checkbox)
|
||
else:
|
||
layout.addWidget(checkbox)
|
||
self.injected_elements.append(checkbox)
|
||
return checkbox
|
||
|
||
def add_radio_to_layout(self, layout, text, callback=None, position=None,
|
||
checked=False, group=None):
|
||
radio = QRadioButton(text)
|
||
radio.setChecked(checked)
|
||
if group:
|
||
group.addButton(radio)
|
||
if callback:
|
||
radio.toggled.connect(callback)
|
||
if position is not None:
|
||
layout.insertWidget(position, radio)
|
||
else:
|
||
layout.addWidget(radio)
|
||
self.injected_elements.append(radio)
|
||
return radio
|
||
|
||
def add_slider_to_layout(self, layout, min_val=0, max_val=100, value=50,
|
||
callback=None, position=None, orientation=Qt.Horizontal, width=200):
|
||
slider = QSlider(orientation)
|
||
slider.setRange(min_val, max_val)
|
||
slider.setValue(value)
|
||
if orientation == Qt.Horizontal and width:
|
||
slider.setFixedWidth(width)
|
||
if callback:
|
||
slider.valueChanged.connect(callback)
|
||
if position is not None:
|
||
layout.insertWidget(position, slider)
|
||
else:
|
||
layout.addWidget(slider)
|
||
self.injected_elements.append(slider)
|
||
return slider
|
||
|
||
def add_spinbox_to_layout(self, layout, min_val=0, max_val=100, value=0,
|
||
callback=None, position=None, width=100, step=1, prefix="", suffix=""):
|
||
spinbox = QSpinBox()
|
||
spinbox.setRange(min_val, max_val)
|
||
spinbox.setValue(value)
|
||
spinbox.setSingleStep(step)
|
||
spinbox.setFixedWidth(width)
|
||
if prefix:
|
||
spinbox.setPrefix(prefix)
|
||
if suffix:
|
||
spinbox.setSuffix(suffix)
|
||
if callback:
|
||
spinbox.valueChanged.connect(callback)
|
||
if position is not None:
|
||
layout.insertWidget(position, spinbox)
|
||
else:
|
||
layout.addWidget(spinbox)
|
||
self.injected_elements.append(spinbox)
|
||
return spinbox
|
||
|
||
def add_progress_to_layout(self, layout, value=0, max_val=100, position=None,
|
||
width=200, height=20, text_visible=True):
|
||
progress = QProgressBar()
|
||
progress.setRange(0, max_val)
|
||
progress.setValue(value)
|
||
progress.setTextVisible(text_visible)
|
||
progress.setFixedHeight(height)
|
||
if width:
|
||
progress.setFixedWidth(width)
|
||
if position is not None:
|
||
layout.insertWidget(position, progress)
|
||
else:
|
||
layout.addWidget(progress)
|
||
self.injected_elements.append(progress)
|
||
return progress
|
||
|
||
def add_list_to_layout(self, layout, items=None, callback=None, position=None,
|
||
width=200, height=150, selection_mode=QListWidget.SingleSelection):
|
||
list_widget = QListWidget()
|
||
list_widget.setFixedHeight(height)
|
||
if width:
|
||
list_widget.setFixedWidth(width)
|
||
list_widget.setSelectionMode(selection_mode)
|
||
if items:
|
||
list_widget.addItems(items)
|
||
if callback:
|
||
list_widget.itemClicked.connect(callback)
|
||
if position is not None:
|
||
layout.insertWidget(position, list_widget)
|
||
else:
|
||
layout.addWidget(list_widget)
|
||
self.injected_elements.append(list_widget)
|
||
return list_widget
|
||
|
||
def add_table_to_layout(self, layout, rows=0, cols=0, headers=None,
|
||
position=None, width=400, height=200):
|
||
table = QTableWidget(rows, cols)
|
||
if headers:
|
||
table.setHorizontalHeaderLabels(headers)
|
||
table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
|
||
table.setFixedHeight(height)
|
||
if width:
|
||
table.setFixedWidth(width)
|
||
if position is not None:
|
||
layout.insertWidget(position, table)
|
||
else:
|
||
layout.addWidget(table)
|
||
self.injected_elements.append(table)
|
||
return table
|
||
|
||
def add_textarea_to_layout(self, layout, placeholder="", callback=None,
|
||
position=None, width=300, height=100, read_only=False):
|
||
textarea = QTextEdit()
|
||
textarea.setPlaceholderText(placeholder)
|
||
textarea.setReadOnly(read_only)
|
||
textarea.setFixedHeight(height)
|
||
if width:
|
||
textarea.setFixedWidth(width)
|
||
if callback:
|
||
textarea.textChanged.connect(callback)
|
||
if position is not None:
|
||
layout.insertWidget(position, textarea)
|
||
else:
|
||
layout.addWidget(textarea)
|
||
self.injected_elements.append(textarea)
|
||
return textarea
|
||
|
||
def add_label_to_layout(self, layout, text, style="normal", position=None,
|
||
width=None, height=None, alignment=None):
|
||
label = QLabel(text)
|
||
label.setStyleSheet(self._get_label_style(style))
|
||
if width:
|
||
label.setFixedWidth(width)
|
||
if height:
|
||
label.setFixedHeight(height)
|
||
if alignment:
|
||
label.setAlignment(alignment)
|
||
if position is not None:
|
||
layout.insertWidget(position, label)
|
||
else:
|
||
layout.addWidget(label)
|
||
self.injected_elements.append(label)
|
||
return label
|
||
|
||
def add_separator(self, layout, position=None, orientation=Qt.Horizontal):
|
||
line = QFrame()
|
||
if orientation == Qt.Horizontal:
|
||
line.setFrameShape(QFrame.HLine)
|
||
else:
|
||
line.setFrameShape(QFrame.VLine)
|
||
line.setFrameShadow(QFrame.Sunken)
|
||
if position is not None:
|
||
layout.insertWidget(position, line)
|
||
else:
|
||
layout.addWidget(line)
|
||
self.injected_elements.append(line)
|
||
return line
|
||
|
||
def add_groupbox(self, layout, title, position=None):
|
||
group = QGroupBox(title)
|
||
group_layout = QVBoxLayout(group)
|
||
if position is not None:
|
||
layout.insertWidget(position, group)
|
||
else:
|
||
layout.addWidget(group)
|
||
self.injected_elements.append(group)
|
||
return group, group_layout
|
||
|
||
def add_custom_widget(self, layout, widget, position=None):
|
||
if position is not None:
|
||
layout.insertWidget(position, widget)
|
||
else:
|
||
layout.addWidget(widget)
|
||
self.injected_elements.append(widget)
|
||
return widget
|
||
|
||
# ==================== 添加标签页/面板（新增） ====================
|
||
|
||
def add_tab_to_main(self, tab_title, icon=None):
|
||
central = self.get_central_widget()
|
||
if not central:
|
||
return None
|
||
tab_widget = central.findChild(QTabWidget)
|
||
if not tab_widget:
|
||
tab_widget = QTabWidget(central)
|
||
layout = central.layout()
|
||
if layout:
|
||
layout.addWidget(tab_widget)
|
||
else:
|
||
new_layout = QVBoxLayout(central)
|
||
new_layout.addWidget(tab_widget)
|
||
new_tab = QWidget()
|
||
tab_layout = QVBoxLayout(new_tab)
|
||
if icon:
|
||
index = tab_widget.addTab(new_tab, icon, tab_title)
|
||
else:
|
||
index = tab_widget.addTab(new_tab, tab_title)
|
||
self._custom_tabs.append((tab_widget, new_tab, index))
|
||
self.injected_elements.append(new_tab)
|
||
return new_tab, tab_layout, index
|
||
|
||
def add_panel_to_main(self, title, area=Qt.LeftDockWidgetArea, floating=False):
|
||
dock = QDockWidget(title, self.main_window)
|
||
dock.setAllowedAreas(Qt.AllDockWidgetAreas)
|
||
content = QWidget()
|
||
content_layout = QVBoxLayout(content)
|
||
dock.setWidget(content)
|
||
self.main_window.addDockWidget(area, dock)
|
||
if floating:
|
||
dock.setFloating(True)
|
||
self._custom_panels.append(dock)
|
||
self.injected_elements.append(dock)
|
||
return dock, content, content_layout
|
||
|
||
def add_toolbar(self, title, area=Qt.TopToolBarArea):
|
||
toolbar = QToolBar(title, self.main_window)
|
||
toolbar.setObjectName(f"plugin_toolbar_{title}")
|
||
self.main_window.addToolBar(area, toolbar)
|
||
self.injected_elements.append(toolbar)
|
||
return toolbar
|
||
|
||
def add_status_widget(self, widget, permanent=True):
|
||
status_bar = self.get_status_bar()
|
||
if not status_bar:
|
||
return None
|
||
if permanent:
|
||
status_bar.addPermanentWidget(widget)
|
||
else:
|
||
status_bar.addWidget(widget)
|
||
self.injected_elements.append(widget)
|
||
return widget
|
||
|
||
# ==================== 菜单和快捷键（新增） ====================
|
||
|
||
def add_menu_action(self, menu, text, callback, icon=None, shortcut=None, tooltip=None):
|
||
action = QAction(text, self.main_window)
|
||
if icon:
|
||
action.setIcon(icon)
|
||
if shortcut:
|
||
action.setShortcut(shortcut)
|
||
if tooltip:
|
||
action.setToolTip(tooltip)
|
||
action.triggered.connect(callback)
|
||
menu.addAction(action)
|
||
self.injected_elements.append(action)
|
||
return action
|
||
|
||
def add_menu(self, menu_bar, title, icon=None):
|
||
menu = menu_bar.addMenu(title)
|
||
if icon:
|
||
menu.setIcon(icon)
|
||
self.injected_elements.append(menu)
|
||
return menu
|
||
|
||
def add_shortcut(self, key_sequence, callback, context=Qt.WindowShortcut):
|
||
shortcut = QShortcut(QKeySequence(key_sequence), self.main_window)
|
||
shortcut.setContext(context)
|
||
shortcut.activated.connect(callback)
|
||
self.injected_elements.append(shortcut)
|
||
return shortcut
|
||
|
||
# ==================== 修改现有控件 ====================
|
||
|
||
def modify_button(self, button, text=None, callback=None, style=None,
|
||
icon=None, tooltip=None, enabled=None, visible=None):
|
||
if isinstance(button, str):
|
||
button = self.find_button_by_text(button)
|
||
if not button:
|
||
return False
|
||
if text is not None:
|
||
button.setText(text)
|
||
if callback is not None:
|
||
try:
|
||
button.clicked.disconnect()
|
||
except Exception:
|
||
pass
|
||
button.clicked.connect(callback)
|
||
if style is not None:
|
||
button.setStyleSheet(self._get_button_style(style))
|
||
if icon is not None:
|
||
button.setIcon(icon)
|
||
if tooltip is not None:
|
||
button.setToolTip(tooltip)
|
||
if enabled is not None:
|
||
button.setEnabled(enabled)
|
||
if visible is not None:
|
||
button.setVisible(visible)
|
||
return True
|
||
|
||
def modify_label(self, label, text=None, style=None, color=None, font_size=None):
|
||
if isinstance(label, str):
|
||
label = self.find_label_by_text(label)
|
||
if not label:
|
||
return False
|
||
if text is not None:
|
||
label.setText(text)
|
||
if style is not None:
|
||
label.setStyleSheet(self._get_label_style(style))
|
||
if color is not None:
|
||
label.setStyleSheet(f"color: {color};")
|
||
if font_size is not None:
|
||
font = label.font()
|
||
font.setPointSize(font_size)
|
||
label.setFont(font)
|
||
return True
|
||
|
||
def hide_button(self, button):
|
||
return self.modify_button(button, visible=False)
|
||
|
||
def show_button(self, button):
|
||
return self.modify_button(button, visible=True)
|
||
|
||
def disable_button(self, button):
|
||
return self.modify_button(button, enabled=False)
|
||
|
||
def enable_button(self, button):
|
||
return self.modify_button(button, enabled=True)
|
||
|
||
# ==================== 布局操作（新增） ====================
|
||
|
||
def get_layout_of_widget(self, widget):
|
||
if widget and widget.parent():
|
||
return widget.parent().layout()
|
||
return None
|
||
|
||
def remove_widget_from_layout(self, layout, widget):
|
||
if layout and widget:
|
||
layout.removeWidget(widget)
|
||
widget.setParent(None)
|
||
return True
|
||
return False
|
||
|
||
def move_widget_in_layout(self, layout, widget, new_position):
|
||
if not layout or not widget:
|
||
return False
|
||
current_index = layout.indexOf(widget)
|
||
if current_index < 0:
|
||
return False
|
||
layout.removeWidget(widget)
|
||
layout.insertWidget(new_position, widget)
|
||
return True
|
||
|
||
def swap_widgets_in_layout(self, layout, widget1, widget2):
|
||
if not layout or not widget1 or not widget2:
|
||
return False
|
||
idx1 = layout.indexOf(widget1)
|
||
idx2 = layout.indexOf(widget2)
|
||
if idx1 < 0 or idx2 < 0:
|
||
return False
|
||
layout.removeWidget(widget1)
|
||
layout.removeWidget(widget2)
|
||
min_idx = min(idx1, idx2)
|
||
layout.insertWidget(min_idx, widget2)
|
||
layout.insertWidget(min_idx + 1, widget1)
|
||
return True
|
||
|
||
def insert_layout(self, parent_layout, position, orientation="vertical"):
|
||
if orientation == "vertical":
|
||
new_layout = QVBoxLayout()
|
||
else:
|
||
new_layout = QHBoxLayout()
|
||
parent_layout.insertLayout(position, new_layout)
|
||
return new_layout
|
||
|
||
# ==================== 对话框和提示（新增） ====================
|
||
|
||
def show_message_box(self, title, message, icon=QMessageBox.Information, buttons=QMessageBox.Ok):
|
||
msg_box = QMessageBox(self.main_window)
|
||
msg_box.setWindowTitle(title)
|
||
msg_box.setText(message)
|
||
msg_box.setIcon(icon)
|
||
msg_box.setStandardButtons(buttons)
|
||
return msg_box.exec_()
|
||
|
||
def show_input_dialog(self, title, label, default_text="", mode=QLineEdit.Normal):
|
||
text, ok = QInputDialog.getText(self.main_window, title, label, mode, default_text)
|
||
return text if ok else None
|
||
|
||
def show_file_dialog(self, title, filter="所有文件 (*.*)", directory=""):
|
||
file_path, _ = QFileDialog.getOpenFileName(self.main_window, title, directory, filter)
|
||
return file_path
|
||
|
||
def show_save_dialog(self, title, filter="所有文件 (*.*)", directory=""):
|
||
file_path, _ = QFileDialog.getSaveFileName(self.main_window, title, directory, filter)
|
||
return file_path
|
||
|
||
def show_color_dialog(self, initial_color=QColor(255, 255, 255)):
|
||
color = QColorDialog.getColor(initial_color, self.main_window)
|
||
return color if color.isValid() else None
|
||
|
||
def show_font_dialog(self, initial_font=None):
|
||
if initial_font is None:
|
||
initial_font = QFont()
|
||
font, ok = QFontDialog.getFont(initial_font, self.main_window)
|
||
return font if ok else None
|
||
|
||
# ==================== 样式和主题（新增） ====================
|
||
|
||
def get_app_style(self):
|
||
return self.main_window.styleSheet()
|
||
|
||
def set_app_style(self, style_sheet):
|
||
self.main_window.setStyleSheet(style_sheet)
|
||
|
||
def append_app_style(self, style_sheet):
|
||
current = self.main_window.styleSheet()
|
||
self.main_window.setStyleSheet(current + "\n" + style_sheet)
|
||
|
||
def set_widget_style(self, widget, style_sheet):
|
||
if widget:
|
||
widget.setStyleSheet(style_sheet)
|
||
|
||
def get_palette(self):
|
||
return self.main_window.palette()
|
||
|
||
def set_palette_color(self, role, color, group=QPalette.Active):
|
||
palette = self.main_window.palette()
|
||
palette.setColor(group, role, QColor(color))
|
||
self.main_window.setPalette(palette)
|
||
|
||
# ==================== 窗口操作（新增） ====================
|
||
|
||
def set_window_title(self, title):
|
||
self.main_window.setWindowTitle(title)
|
||
|
||
def get_window_title(self):
|
||
return self.main_window.windowTitle()
|
||
|
||
def set_window_size(self, width, height):
|
||
self.main_window.resize(width, height)
|
||
|
||
def get_window_size(self):
|
||
return self.main_window.size()
|
||
|
||
def set_window_position(self, x, y):
|
||
self.main_window.move(x, y)
|
||
|
||
def get_window_position(self):
|
||
return self.main_window.pos()
|
||
|
||
def minimize_window(self):
|
||
self.main_window.showMinimized()
|
||
|
||
def maximize_window(self):
|
||
self.main_window.showMaximized()
|
||
|
||
def restore_window(self):
|
||
self.main_window.showNormal()
|
||
|
||
def set_window_opacity(self, opacity):
|
||
self.main_window.setWindowOpacity(opacity)
|
||
|
||
def set_window_on_top(self, on_top=True):
|
||
flags = self.main_window.windowFlags()
|
||
if on_top:
|
||
flags |= Qt.WindowStaysOnTopHint
|
||
else:
|
||
flags &= ~Qt.WindowStaysOnTopHint
|
||
self.main_window.setWindowFlags(flags)
|
||
self.main_window.show()
|
||
|
||
# ==================== 定时器和动画（新增） ====================
|
||
|
||
def create_timer(self, interval, callback, single_shot=False):
|
||
timer = QTimer(self.main_window)
|
||
timer.setInterval(interval)
|
||
timer.setSingleShot(single_shot)
|
||
timer.timeout.connect(callback)
|
||
self.injected_elements.append(timer)
|
||
return timer
|
||
|
||
def start_timer(self, timer):
|
||
if timer:
|
||
timer.start()
|
||
|
||
def stop_timer(self, timer):
|
||
if timer:
|
||
timer.stop()
|
||
|
||
def delayed_call(self, milliseconds, callback):
|
||
QTimer.singleShot(milliseconds, callback)
|
||
|
||
# ==================== 资源管理（新增） ====================
|
||
|
||
def load_icon(self, path):
|
||
if os.path.exists(path):
|
||
return QIcon(path)
|
||
return None
|
||
|
||
def load_pixmap(self, path):
|
||
if os.path.exists(path):
|
||
return QPixmap(path)
|
||
return None
|
||
|
||
def load_font(self, path):
|
||
if os.path.exists(path):
|
||
font_id = QFontDatabase.addApplicationFont(path)
|
||
if font_id >= 0:
|
||
return QFontDatabase.applicationFontFamilies(font_id)[0]
|
||
return None
|
||
|
||
def create_color(self, r, g, b, a=255):
|
||
return QColor(r, g, b, a)
|
||
|
||
def create_gradient(self, x1, y1, x2, y2, color1, color2):
|
||
gradient = QLinearGradient(x1, y1, x2, y2)
|
||
gradient.setColorAt(0, QColor(color1))
|
||
gradient.setColorAt(1, QColor(color2))
|
||
return gradient
|
||
|
||
# ==================== 内部辅助方法 ====================
|
||
|
||
def _get_top_bar(self):
|
||
for name in ["top_bar", "title_bar", "header", "top_frame"]:
|
||
widget = self.find_widget_by_name(name)
|
||
if widget:
|
||
return widget
|
||
return None
|
||
|
||
def _get_bottom_bar(self):
|
||
from PyQt5.QtWidgets import QStatusBar
|
||
for name in ["status_bar", "bottom_bar", "statusBar", "footer"]:
|
||
widget = self.find_widget_by_name(name)
|
||
if widget:
|
||
return widget
|
||
status_bar = self.main_window.findChild(QStatusBar)
|
||
if status_bar:
|
||
return status_bar
|
||
return None
|
||
|
||
def _get_sidebar(self):
|
||
for name in ["sidebar", "side_panel", "left_panel", "nav_panel"]:
|
||
widget = self.find_widget_by_name(name)
|
||
if widget:
|
||
return widget
|
||
return None
|
||
|
||
def _get_draw_area(self):
|
||
for name in ["draw_area", "draw_widget", "central_area", "main_area"]:
|
||
widget = self.find_widget_by_name(name)
|
||
if widget:
|
||
return widget
|
||
return None
|
||
|
||
def _get_button_style(self, style):
|
||
styles = {
|
||
"primary": """
|
||
QPushButton { background: #0067c0; color: white; border: none;
|
||
border-radius: 6px; padding: 8px 16px; font-weight: 600; font-size: 13px; }
|
||
QPushButton:hover { background: #005a9e; }
|
||
QPushButton:pressed { background: #004578; }
|
||
QPushButton:disabled { background: #cccccc; color: #999999; }
|
||
""",
|
||
"success": """
|
||
QPushButton { background: #107c10; color: white; border: none;
|
||
border-radius: 6px; padding: 8px 16px; font-weight: 600; font-size: 13px; }
|
||
QPushButton:hover { background: #0e6c0e; }
|
||
QPushButton:pressed { background: #0a5c0a; }
|
||
""",
|
||
"warning": """
|
||
QPushButton { background: #ca5010; color: white; border: none;
|
||
border-radius: 6px; padding: 8px 16px; font-weight: 600; font-size: 13px; }
|
||
QPushButton:hover { background: #b04810; }
|
||
""",
|
||
"danger": """
|
||
QPushButton { background: #c42b1c; color: white; border: none;
|
||
border-radius: 6px; padding: 8px 16px; font-weight: 600; font-size: 13px; }
|
||
QPushButton:hover { background: #a82318; }
|
||
""",
|
||
"ghost": """
|
||
QPushButton { background: transparent; color: #0067c0; border: 1px solid #0067c0;
|
||
border-radius: 6px; padding: 8px 16px; font-weight: 600; font-size: 13px; }
|
||
QPushButton:hover { background: #e5f1fb; }
|
||
""",
|
||
"small": """
|
||
QPushButton { background: #f0f0f0; color: #333; border: 1px solid #d1d1d1;
|
||
border-radius: 4px; padding: 4px 10px; font-size: 11px; }
|
||
QPushButton:hover { background: #e0e0e0; }
|
||
""",
|
||
}
|
||
return styles.get(style, styles["primary"])
|
||
|
||
def _get_label_style(self, style):
|
||
styles = {
|
||
"normal": "color: #333333; font-size: 13px;",
|
||
"title": "color: #1a1a1a; font-size: 18px; font-weight: 700;",
|
||
"subtitle": "color: #555555; font-size: 14px; font-weight: 600;",
|
||
"hint": "color: #999999; font-size: 12px;",
|
||
"success": "color: #107c10; font-size: 13px; font-weight: 600;",
|
||
"warning": "color: #ca5010; font-size: 13px; font-weight: 600;",
|
||
"danger": "color: #c42b1c; font-size: 13px; font-weight: 600;",
|
||
}
|
||
return styles.get(style, styles["normal"])
|
||
|
||
# ==================== 清理 ====================
|
||
|
||
def cleanup(self):
|
||
for element in self.injected_elements:
|
||
try:
|
||
if hasattr(element, 'deleteLater'):
|
||
element.deleteLater()
|
||
elif hasattr(element, 'close'):
|
||
element.close()
|
||
except Exception:
|
||
pass
|
||
self.injected_elements = []
|
||
self._custom_tabs = []
|
||
self._custom_panels = []
|
||
|
||
|
||
class DraggableButton(QPushButton):
|
||
"""可拖动的按钮类"""
|
||
|
||
def __init__(self, text, parent=None):
|
||
super().__init__(text, parent)
|
||
self._drag_position = None
|
||
self.setCursor(QCursor(Qt.OpenHandCursor))
|
||
|
||
def mousePressEvent(self, event):
|
||
if event.button() == Qt.LeftButton:
|
||
self._drag_position = event.globalPos() - self.frameGeometry().topLeft()
|
||
self.setCursor(QCursor(Qt.ClosedHandCursor))
|
||
super().mousePressEvent(event)
|
||
|
||
def mouseMoveEvent(self, event):
|
||
if event.buttons() == Qt.LeftButton and self._drag_position:
|
||
self.move(event.globalPos() - self._drag_position)
|
||
super().mouseMoveEvent(event)
|
||
|
||
def mouseReleaseEvent(self, event):
|
||
self._drag_position = None
|
||
self.setCursor(QCursor(Qt.OpenHandCursor))
|
||
super().mouseReleaseEvent(event)
|