pax
043f36ef99
- Small triangle icon inside the search input (trailing position) - Clicking empty search bar also shows history dropdown - Menu appears below the search input
203 lines
7.1 KiB
Python
203 lines
7.1 KiB
Python
"""Search bar with tag autocomplete, history, and saved searches."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from PySide6.QtCore import Qt, Signal, QTimer, QStringListModel
|
|
from PySide6.QtGui import QIcon
|
|
from PySide6.QtWidgets import (
|
|
QWidget,
|
|
QHBoxLayout,
|
|
QLineEdit,
|
|
QPushButton,
|
|
QCompleter,
|
|
QMenu,
|
|
QInputDialog,
|
|
QWidgetAction,
|
|
)
|
|
|
|
from ..core.db import Database
|
|
|
|
|
|
class SearchBar(QWidget):
|
|
"""Tag search bar with autocomplete, history dropdown, and saved searches."""
|
|
|
|
search_requested = Signal(str)
|
|
autocomplete_requested = Signal(str)
|
|
|
|
def __init__(self, db: Database | None = None, parent: QWidget | None = None) -> None:
|
|
super().__init__(parent)
|
|
self._db = db
|
|
layout = QHBoxLayout(self)
|
|
layout.setContentsMargins(0, 0, 0, 0)
|
|
layout.setSpacing(6)
|
|
|
|
self._input = QLineEdit()
|
|
self._input.setPlaceholderText("Search tags... (supports -negatives)")
|
|
self._input.returnPressed.connect(self._do_search)
|
|
|
|
# Dropdown arrow inside search bar
|
|
from PySide6.QtGui import QPixmap, QPainter, QFont
|
|
pixmap = QPixmap(16, 16)
|
|
pixmap.fill(Qt.GlobalColor.transparent)
|
|
painter = QPainter(pixmap)
|
|
painter.setPen(self._input.palette().color(self._input.palette().ColorRole.Text))
|
|
painter.setFont(QFont(self._input.font().family(), 8))
|
|
painter.drawText(pixmap.rect(), Qt.AlignmentFlag.AlignCenter, "\u25BC")
|
|
painter.end()
|
|
self._history_action = self._input.addAction(
|
|
QIcon(pixmap),
|
|
QLineEdit.ActionPosition.TrailingPosition,
|
|
)
|
|
self._history_action.setToolTip("Search history & saved searches")
|
|
self._history_action.triggered.connect(self._show_history_menu)
|
|
|
|
# Show history when focusing empty input
|
|
self._input.mousePressEvent = self._on_input_click
|
|
self._original_mouse_press = QLineEdit.mousePressEvent
|
|
|
|
layout.addWidget(self._input, stretch=1)
|
|
|
|
# Save search button
|
|
self._save_btn = QPushButton("Save")
|
|
self._save_btn.setFixedWidth(50)
|
|
self._save_btn.setToolTip("Save current search")
|
|
self._save_btn.clicked.connect(self._save_current_search)
|
|
layout.addWidget(self._save_btn)
|
|
|
|
self._btn = QPushButton("Search")
|
|
self._btn.clicked.connect(self._do_search)
|
|
layout.addWidget(self._btn)
|
|
|
|
# Autocomplete
|
|
self._completer_model = QStringListModel()
|
|
self._completer = QCompleter(self._completer_model)
|
|
self._completer.setCaseSensitivity(Qt.CaseSensitivity.CaseInsensitive)
|
|
self._completer.setCompletionMode(QCompleter.CompletionMode.PopupCompletion)
|
|
self._input.setCompleter(self._completer)
|
|
|
|
# Debounce
|
|
self._ac_timer = QTimer()
|
|
self._ac_timer.setSingleShot(True)
|
|
self._ac_timer.setInterval(300)
|
|
self._ac_timer.timeout.connect(self._request_autocomplete)
|
|
self._input.textChanged.connect(self._on_text_changed)
|
|
|
|
def _on_input_click(self, event) -> None:
|
|
self._original_mouse_press(self._input, event)
|
|
if not self._input.text().strip():
|
|
self._show_history_menu()
|
|
|
|
def _on_text_changed(self, text: str) -> None:
|
|
self._ac_timer.start()
|
|
|
|
def _request_autocomplete(self) -> None:
|
|
text = self._input.text().strip()
|
|
if not text:
|
|
return
|
|
last_tag = text.split()[-1] if text.split() else ""
|
|
query = last_tag.lstrip("-")
|
|
if len(query) >= 2:
|
|
self.autocomplete_requested.emit(query)
|
|
|
|
def set_suggestions(self, suggestions: list[str]) -> None:
|
|
self._completer_model.setStringList(suggestions)
|
|
|
|
def _do_search(self) -> None:
|
|
query = self._input.text().strip()
|
|
if self._db and query:
|
|
self._db.add_search_history(query)
|
|
self.search_requested.emit(query)
|
|
|
|
def _show_history_menu(self) -> None:
|
|
if not self._db:
|
|
return
|
|
|
|
menu = QMenu(self)
|
|
|
|
# Saved searches
|
|
saved = self._db.get_saved_searches()
|
|
if saved:
|
|
saved_header = menu.addAction("-- Saved Searches --")
|
|
saved_header.setEnabled(False)
|
|
saved_actions = {}
|
|
for sid, name, query in saved:
|
|
a = menu.addAction(f" {name} ({query})")
|
|
saved_actions[id(a)] = (sid, query)
|
|
menu.addSeparator()
|
|
|
|
# History
|
|
history = self._db.get_search_history()
|
|
if history:
|
|
hist_header = menu.addAction("-- Recent --")
|
|
hist_header.setEnabled(False)
|
|
hist_actions = {}
|
|
hist_delete_actions = {}
|
|
for query in history:
|
|
row = QWidget()
|
|
row_layout = QHBoxLayout(row)
|
|
row_layout.setContentsMargins(8, 2, 4, 2)
|
|
label = QPushButton(query)
|
|
label.setFlat(True)
|
|
label.setStyleSheet("text-align: left; border: none; padding: 2px 4px;")
|
|
delete_btn = QPushButton("x")
|
|
delete_btn.setFixedWidth(20)
|
|
delete_btn.setFlat(True)
|
|
delete_btn.setToolTip("Remove from history")
|
|
row_layout.addWidget(label, stretch=1)
|
|
row_layout.addWidget(delete_btn)
|
|
|
|
from PySide6.QtWidgets import QWidgetAction
|
|
wa = QWidgetAction(menu)
|
|
wa.setDefaultWidget(row)
|
|
menu.addAction(wa)
|
|
hist_actions[id(label)] = query
|
|
hist_delete_actions[id(delete_btn)] = query
|
|
|
|
label.clicked.connect(lambda checked, q=query, m=menu: (
|
|
self._input.setText(q), self._do_search(), m.close()
|
|
))
|
|
delete_btn.clicked.connect(lambda checked, q=query, m=menu: (
|
|
self._db.remove_search_history(q), m.close(), self._show_history_menu()
|
|
))
|
|
|
|
menu.addSeparator()
|
|
clear_action = menu.addAction("Clear History")
|
|
else:
|
|
hist_actions = {}
|
|
hist_delete_actions = {}
|
|
clear_action = None
|
|
|
|
if not saved and not history:
|
|
empty = menu.addAction("No history yet")
|
|
empty.setEnabled(False)
|
|
|
|
action = menu.exec(self._input.mapToGlobal(self._input.rect().bottomLeft()))
|
|
if not action:
|
|
return
|
|
|
|
if clear_action and action == clear_action:
|
|
self._db.clear_search_history()
|
|
elif saved and id(action) in saved_actions:
|
|
_, query = saved_actions[id(action)]
|
|
self._input.setText(query)
|
|
self._do_search()
|
|
|
|
def _save_current_search(self) -> None:
|
|
if not self._db:
|
|
return
|
|
query = self._input.text().strip()
|
|
if not query:
|
|
return
|
|
name, ok = QInputDialog.getText(self, "Save Search", "Name:", text=query)
|
|
if ok and name.strip():
|
|
self._db.add_saved_search(name.strip(), query)
|
|
|
|
def text(self) -> str:
|
|
return self._input.text().strip()
|
|
|
|
def set_text(self, text: str) -> None:
|
|
self._input.setText(text)
|
|
|
|
def focus(self) -> None:
|
|
self._input.setFocus()
|