import os from PyQt6.QtWidgets import ( QWidget, QVBoxLayout, QHBoxLayout, QLabel, QPushButton, QMessageBox, QMenu, QSizePolicy ) from PyQt6.QtCore import Qt, pyqtSignal, QSize, QPoint, QMimeData, QByteArray from PyQt6.QtGui import ( QDragEnterEvent, QDragMoveEvent, QDropEvent, QCursor, QDrag, QPainter, QColor, QPen, QShortcut, QKeySequence, ) from db import database from ui.item import ItemWidget from ui.flow_layout import FlowContainer import ui.theme as theme import qtawesome as qta import shortcut_target import ui.dialog_style as dialog_style def item_count_for_group(group_id: int) -> int: """分组内程序/文件总数（文件夹分组数磁盘文件，普通分组数数据库条目）。""" groups = database.get_groups() group_info = next((g for g in groups if g["id"] == group_id), {}) folder_path = (group_info.get("folder_path", "") or "").strip() if folder_path and os.path.isdir(folder_path): n = 0 try: import stat as _stat for fname in os.listdir(folder_path): fpath = os.path.join(folder_path, fname) try: if _stat.S_ISREG(os.stat(fpath).st_mode): n += 1 except OSError: continue except OSError: pass return n return len(database.get_items(group_id)) class DragHandle(QLabel): """分组拖拽排序手柄""" def __init__(self, group_widget, parent=None): super().__init__(parent) self.group_widget = group_widget self.setFixedSize(18, 18) self.setCursor(Qt.CursorShape.SizeVerCursor) self.setToolTip("拖拽排序") self.setPixmap(qta.icon("fa5s.grip-vertical", color="#888").pixmap(14, 14)) def mousePressEvent(self, event): if event.button() == Qt.MouseButton.LeftButton: self._drag_start = event.pos() def mouseMoveEvent(self, event): if not (event.buttons() & Qt.MouseButton.LeftButton): return if (event.pos() - self._drag_start).manhattanLength() < 6: return drag = QDrag(self) mime = QMimeData() gid = self.group_widget.group_data["id"] mime.setData("application/x-group-id", QByteArray(str(gid).encode())) drag.setMimeData(mime) px = self.group_widget.header.grab() drag.setPixmap(px) drag.setHotSpot(QPoint(px.width() // 2, px.height() // 2)) drag.exec(Qt.DropAction.MoveAction) class FlowWidget(QWidget): item_dropped = pyqtSignal(int) refreshed = pyqtSignal() def __init__(self, group_id, parent=None): super().__init__(parent) self.group_id = group_id self.setAcceptDrops(True) self._drop_indicator = -1 layout = QVBoxLayout(self) layout.setContentsMargins(2, 2, 2, 2) layout.setSpacing(0) self._container = FlowContainer(self) layout.addWidget(self._container) layout.addStretch() self.setFocusPolicy(Qt.FocusPolicy.ClickFocus) _del_sc = QShortcut(QKeySequence.StandardKey.Delete, self) _del_sc.setContext(Qt.ShortcutContext.WidgetWithChildrenShortcut) _del_sc.activated.connect(self._delete_selected_items) _esc_sc = QShortcut(QKeySequence(Qt.Key.Key_Escape), self) _esc_sc.setContext(Qt.ShortcutContext.WidgetWithChildrenShortcut) _esc_sc.activated.connect(self._container.clear_selection) self.refresh() def refresh(self, keyword=""): self._container.clear_widgets() # 判断是否是文件夹分组 groups = database.get_groups() group_info = next((g for g in groups if g["id"] == self.group_id), {}) folder_path = (group_info.get("folder_path", "") or "").strip() if folder_path and os.path.isdir(folder_path): # 文件夹分组：直接读目录，不走数据库 items = [] try: for fname in sorted(os.listdir(folder_path)): fpath = os.path.join(folder_path, fname) try: import stat as _stat if _stat.S_ISREG(os.stat(fpath).st_mode): name = os.path.splitext(fname)[0] or fname items.append({"id": None, "name": name, "path": fpath, "group_id": self.group_id}) except Exception: continue except Exception: pass else: # 普通分组：走数据库 items = database.get_items(self.group_id) for data in items: iw = ItemWidget(data, self._container) iw.show() if keyword and not iw.matches(keyword): iw.hide() self._container.add_widget(iw) self._container._schedule_relayout() self.updateGeometry() self.refreshed.emit() def apply_theme(self): for w in self._container.get_widgets(): w._apply_theme() def filter(self, keyword: str): # 不重建，只切换可见性，然后重新布局 for w in self._container.get_widgets(): if keyword: w.setVisible(w.matches(keyword)) else: w.show() self._container._relayout() self.updateGeometry() def mousePressEvent(self, event): if event.button() == Qt.MouseButton.LeftButton: cg = self._container.geometry() if not cg.contains(event.pos()): lp = self._container.mapFrom(self, event.pos()) w, h = max(1, self._container.width()), max(1, self._container.height()) lp.setX(max(0, min(lp.x(), w - 1))) lp.setY(max(0, min(lp.y(), h - 1))) self._container._begin_rubber(lp) return super().mousePressEvent(event) def _delete_selected_items(self): widgets = [w for w in self._container.get_widgets() if getattr(w, "_selected", False)] if not widgets: return ret = dialog_style.question( self, "批量删除", f"确定删除选中的 {len(widgets)} 项？", default_button=QMessageBox.StandardButton.No, ) if ret != QMessageBox.StandardButton.Yes: return for w in widgets: w._delete_item(skip_confirm=True, skip_refresh=True) self.refresh() def dragEnterEvent(self, event: QDragEnterEvent): if event.mimeData().hasFormat("application/x-item-id") or \ event.mimeData().hasUrls(): self._container.set_drag_highlight(True) event.acceptProposedAction() else: event.ignore() def dragMoveEvent(self, event: QDragMoveEvent): if event.mimeData().hasFormat("application/x-item-id"): pos = self._container.mapFrom(self, event.position().toPoint()) self._drop_indicator = self._container.index_at(pos) self.update() event.acceptProposedAction() def dragLeaveEvent(self, event): self._container.set_drag_highlight(False) self._drop_indicator = -1 self.update() def paintEvent(self, event): super().paintEvent(event) if self._drop_indicator < 0: return from PyQt6.QtGui import QPainter, QColor, QPen all_w = self._container.get_widgets() if not all_w: return pos = self._drop_indicator p = QPainter(self) p.setPen(QPen(QColor("#4a9eff"), 2)) offset = self._container.pos() def _draw_vline(x_local: int, g): x = offset.x() + x_local y1 = offset.y() + g.top() y2 = offset.y() + g.bottom() p.drawLine(x, y1, x, y2) if pos >= len(all_w): for j in range(len(all_w) - 1, -1, -1): if all_w[j].isVisible(): g = all_w[j].geometry() _draw_vline(g.right() + 2, g) break p.end() return w = all_w[pos] if w.isVisible(): g = w.geometry() _draw_vline(g.left() - 2, g) else: placed = False for j in range(pos, len(all_w)): if all_w[j].isVisible(): g = all_w[j].geometry() _draw_vline(g.left() - 2, g) placed = True break if not placed: for j in range(pos - 1, -1, -1): if all_w[j].isVisible(): g = all_w[j].geometry() _draw_vline(g.right() + 2, g) break p.end() def dropEvent(self, event: QDropEvent): self._container.set_drag_highlight(False) self._drop_indicator = -1 self.update() mime = event.mimeData() if mime.hasFormat("application/x-item-id"): item_id = int(mime.data("application/x-item-id").data().decode()) pos = self._container.mapFrom(self, event.position().toPoint()) insert_pos = self._container.index_at(pos) database.move_item(item_id, self.group_id, 999) items = database.get_items(self.group_id) ids = [it["id"] for it in items if it["id"] != item_id] ids.insert(min(insert_pos, len(ids)), item_id) database.reorder_items(self.group_id, ids) self.refresh() self.item_dropped.emit(item_id) event.acceptProposedAction() elif mime.hasUrls(): for url in mime.urls(): path = url.toLocalFile() if path: store = shortcut_target.path_for_storage(path) name = shortcut_target.item_name_from_sources(path, store) database.add_item(self.group_id, name, store) self.refresh() event.acceptProposedAction() else: event.ignore() def contextMenuEvent(self, event): from PyQt6.QtWidgets import QApplication t = theme.current() menu_style = f""" QMenu {{ background:{t['menu_bg']}; color:{t['menu_color']}; border:1px solid {t['menu_border']}; padding:4px; border-radius:6px; }} QMenu::item {{ padding:5px 16px; border-radius:3px; }} QMenu::item:selected {{ background:{t['menu_selected']}; }} QMenu::item:disabled {{ color:#666; }} """ menu = QMenu(self) menu.setStyleSheet(menu_style) refresh_act = menu.addAction("🔄 刷新分组") menu.addSeparator() # 检测剪贴板 clipboard = QApplication.clipboard() mime = clipboard.mimeData() paste_act = None clipboard_paths = [] if mime and mime.hasUrls(): # 获取本分组信息 groups = database.get_groups() group_info = next((g for g in groups if g["id"] == self.group_id), {}) is_folder_group = bool(group_info.get("folder_path", "")) # 快捷方式扩展名 shortcut_exts = {".exe", ".lnk", ".bat", ".cmd", ".url"} for url in mime.urls(): path = url.toLocalFile() if not path: continue ext = os.path.splitext(path)[1].lower() if is_folder_group: # 文件夹分组：接受所有文件 if os.path.isfile(path): clipboard_paths.append(path) else: # 普通分组：只接受快捷方式/程序 if ext in shortcut_exts: clipboard_paths.append(path) if clipboard_paths: label = f"📋 粘贴 {len(clipboard_paths)} 个文件" if len(clipboard_paths) > 1 \ else f"📋 粘贴「{os.path.basename(clipboard_paths[0])}」" paste_act = menu.addAction(label) else: act = menu.addAction("📋 粘贴（无可用文件）") act.setEnabled(False) else: act = menu.addAction("📋 粘贴（剪贴板为空）") act.setEnabled(False) action = menu.exec(event.globalPos()) if action == refresh_act: self.refresh() elif paste_act and action == paste_act: groups = database.get_groups() group_info = next((g for g in groups if g["id"] == self.group_id), {}) folder_path = group_info.get("folder_path", "") is_folder_group = bool(folder_path) for path in clipboard_paths: if is_folder_group: # 真正复制文件到文件夹目录 import shutil dest_path = os.path.join(folder_path, os.path.basename(path)) # 避免覆盖同名文件，自动重命名 if os.path.exists(dest_path) and dest_path != path: base, ext = os.path.splitext(os.path.basename(path)) i = 1 while os.path.exists(dest_path): dest_path = os.path.join(folder_path, f"{base} ({i}){ext}") i += 1 try: if path != dest_path: shutil.copy2(path, dest_path) final_path = dest_path except Exception as e: print(f"复制失败: {path} -> {e}") final_path = path # 复制失败就用原路径 name = os.path.splitext(os.path.basename(final_path))[0] or os.path.basename(final_path) else: final_path = shortcut_target.path_for_storage(path) name = shortcut_target.item_name_from_sources(path, final_path) database.add_item(self.group_id, name, final_path) self.refresh() class GroupWidget(QWidget): request_refresh_all = pyqtSignal() def __init__(self, group_data: dict, parent=None): super().__init__(parent) self.group_data = group_data self.collapsed = False self._build_ui() self._apply_theme() def _build_ui(self): self.main_layout = QVBoxLayout(self) self.main_layout.setContentsMargins(0, 0, 0, 4) self.main_layout.setSpacing(0) self.header = QWidget() self.header.setFixedHeight(32) self.header.setCursor(Qt.CursorShape.PointingHandCursor) self.header.mousePressEvent = lambda e: self._toggle() \ if e.button() == Qt.MouseButton.LeftButton else None h_layout = QHBoxLayout(self.header) h_layout.setContentsMargins(4, 0, 6, 0) h_layout.setSpacing(4) # 拖拽排序手柄 self._drag_handle = DragHandle(self) self.toggle_icon = QLabel() self.toggle_icon.setFixedSize(14, 14) self.toggle_icon.setAttribute(Qt.WidgetAttribute.WA_TransparentForMouseEvents) self.title_label = QLabel() self.title_label.setStyleSheet("font-size:12px; font-weight:bold; background:transparent;") self.title_label.setAttribute(Qt.WidgetAttribute.WA_TransparentForMouseEvents) folder_path = self.group_data.get("folder_path", "") or "" # 只显示最后两级路径，避免太长 if folder_path: parts = folder_path.replace("\\", "/").rstrip("/").split("/") display_path = "/".join(parts[-2:]) if len(parts) >= 2 else folder_path else: display_path = "" self.path_label = QLabel(display_path) self.path_label.setStyleSheet("font-size:9px; color:#888; background:transparent;") self.path_label.setAttribute(Qt.WidgetAttribute.WA_TransparentForMouseEvents) self.path_label.setVisible(bool(folder_path)) self.path_label.setMaximumWidth(110) self.path_label.setToolTip(folder_path) self.path_label.setAlignment(Qt.AlignmentFlag.AlignRight | Qt.AlignmentFlag.AlignVCenter) self._add_btn = QPushButton() self._add_btn.setFixedSize(22, 22) self._add_btn.setToolTip("添加程序") self._add_btn.setStyleSheet("border:none; background:transparent;") self._add_btn.clicked.connect(self._add_item_dialog) self._add_btn.mousePressEvent = lambda e: ( self._add_btn.click() if e.button() == Qt.MouseButton.LeftButton else None ) self._more_btn = QPushButton() self._more_btn.setFixedSize(22, 22) self._more_btn.setStyleSheet("border:none; background:transparent;") self._more_btn.clicked.connect(self._group_menu) h_layout.addWidget(self._drag_handle) h_layout.addWidget(self.toggle_icon) h_layout.addWidget(self.title_label) h_layout.addStretch() h_layout.addWidget(self.path_label) h_layout.addWidget(self._add_btn) h_layout.addWidget(self._more_btn) self.flow = FlowWidget(self.group_data["id"]) self.flow.item_dropped.connect(lambda _: self.request_refresh_all.emit()) self.flow.refreshed.connect(self._update_title_count) self._update_title_count() self.main_layout.addWidget(self.header) self.main_layout.addWidget(self.flow) def _update_title_count(self): n = item_count_for_group(self.group_data["id"]) self.title_label.setText(f"{self.group_data['name']} ({n})") def _apply_theme(self): t = theme.current() ic = "#cccccc" if theme.name() == "dark" else "#555555" self.header.setStyleSheet(f""" QWidget {{ background:{t['header_bg']}; border-radius:5px; }} QWidget:hover {{ background:{t['header_hover']}; }} """) self.title_label.setStyleSheet( f"color:{t['item_name_color']}; font-size:12px; font-weight:bold; background:transparent;" ) # 展开/收缩图标 arrow = "fa5s.chevron-down" if not self.collapsed else "fa5s.chevron-right" self.toggle_icon.setPixmap(qta.icon(arrow, color=ic).pixmap(14, 14)) self._add_btn.setIcon(qta.icon("fa5s.plus", color=ic)) self._add_btn.setIconSize(QSize(11, 11)) self._more_btn.setIcon(qta.icon("fa5s.ellipsis-h", color=ic)) self._more_btn.setIconSize(QSize(11, 11)) self._drag_handle.setPixmap(qta.icon("fa5s.grip-vertical", color=ic).pixmap(14, 14)) path_color = "#777" if theme.name() == "dark" else "#999" self.path_label.setStyleSheet( f"font-size:9px; color:{path_color}; background:transparent;" ) self.flow.apply_theme() def _toggle(self): self.collapsed = not self.collapsed self.flow.setVisible(not self.collapsed) if not self.collapsed: self.flow.updateGeometry() self.updateGeometry() # 更新箭头图标 t = theme.current() ic = "#cccccc" if theme.name() == "dark" else "#555555" arrow = "fa5s.chevron-down" if not self.collapsed else "fa5s.chevron-right" self.toggle_icon.setPixmap(qta.icon(arrow, color=ic).pixmap(14, 14)) def _add_item_dialog(self): path, _ = dialog_style.get_open_file_name( self, "选择程序或快捷方式", "", "程序/快捷方式 (*.exe *.lnk *.bat *.cmd *.url);;所有文件 (*)", ) if path: store = shortcut_target.path_for_storage(path) name = shortcut_target.item_name_from_sources(path, store) database.add_item(self.group_data["id"], name, store) self.flow.refresh() def _group_menu(self): t = theme.current() menu = QMenu(self) menu.setStyleSheet(f""" QMenu {{ background:{t['menu_bg']}; color:{t['menu_color']}; border:1px solid {t['menu_border']}; padding:4px; border-radius:6px; }} QMenu::item {{ padding:5px 16px; border-radius:3px; }} QMenu::item:selected {{ background:{t['menu_selected']}; }} """) rename_act = menu.addAction("重命名") del_act = menu.addAction("删除分组") action = menu.exec(QCursor.pos()) if action == rename_act: name, ok = dialog_style.get_text( self, "重命名", "新名称:", text=self.group_data["name"] ) if ok and name.strip(): database.rename_group(self.group_data["id"], name.strip()) self.group_data["name"] = name.strip() self._update_title_count() elif action == del_act: ret = dialog_style.question( self, "确认", f"删除分组「{self.group_data['name']}」及其所有程序？", default_button=QMessageBox.StandardButton.No, ) if ret == QMessageBox.StandardButton.Yes: database.delete_group(self.group_data["id"]) self.request_refresh_all.emit() def refresh(self, keyword=""): self.flow.refresh(keyword) def filter(self, keyword: str): self.flow.filter(keyword) if keyword and self.collapsed: self._toggle()