add button for removing recent lib

Author: yedpodtrzitko

tagstudio/src/core/ts_core.py

@@ -4,7 +4,6 @@
 
 """The core classes and methods of TagStudio."""
 
-import enum
 import json
 import os
 
@@ -19,7 +18,6 @@
 COLLAGE_FOLDER_NAME: str = "collages"
 LIBRARY_FILENAME: str = "ts_library.json"
 
-
 # TODO: Turn this whitelist into a user-configurable blacklist.
 IMAGE_TYPES: list[str] = [
     "png",

tagstudio/src/qt/ts_qt.py

@@ -505,7 +505,11 @@ def start(self):
             )
             self.open_library(lib)
 
-        app.exec_()
+        if self.args.ci:
+            # gracefully terminate the app in CI environment
+            self.thumb_job_queue.put((self.SIGTERM.emit, []))
+
+        app.exec()
 
         self.shutdown()
 
@@ -1330,6 +1334,12 @@ def filter_items(self, query=""):
 
             # self.update_thumbs()
 
+    def remove_recent_library(self, item_key: str):
+        self.settings.beginGroup(SettingItems.LIBS_LIST)
+        self.settings.remove(item_key)
+        self.settings.endGroup()
+        self.settings.sync()
+
     def update_libs_list(self, path: str | Path):
         """add library to list in SettingItems.LIBS_LIST"""
         ITEMS_LIMIT = 5

tagstudio/src/qt/widgets/preview_panel.py

@@ -173,7 +173,7 @@ def __init__(self, library: Library, driver: "QtDriver"):
         info_layout.addWidget(self.dimensions_label)
         info_layout.addWidget(scroll_area)
 
-        # keep list of rendered libraries to avoid needles re-rendering
+        # keep list of rendered libraries to avoid needless re-rendering
         self.render_libs = set()
         self.libs_layout = QVBoxLayout()
         self.fill_libs_widget(self.libs_layout)
@@ -247,44 +247,80 @@ def fill_libs_widget(self, layout: QVBoxLayout):
         libs_sorted = sorted(lib_items.items(), key=lambda item: item[0], reverse=True)
 
         self.render_libs = new_keys
-        return self._fill_libs_widget(libs_sorted, layout)
+        self._fill_libs_widget(libs_sorted, layout)
 
     def _fill_libs_widget(
         self, libraries: list[tuple[str, tuple[str, str]]], layout: QVBoxLayout
     ):
+        def clear_layout(layout_item: QVBoxLayout):
+            for i in reversed(range(layout_item.count())):
+                child = layout_item.itemAt(i)
+                if child.widget() is not None:
+                    child.widget().deleteLater()
+                elif child.layout() is not None:
+                    clear_layout(child.layout())
+
         # remove any potential previous items
         for idx in reversed(range(layout.count())):
-            widget = layout.itemAt(idx).widget()
-            layout.removeWidget(widget)
-            # remove from GUI
-            widget.setParent(None)
+            row_layout = layout.itemAt(idx)
+            clear_layout(row_layout)
 
         label = QLabel("Recent Libraries")
         label.setAlignment(Qt.AlignCenter)
-        layout.addWidget(label)
 
-        for tstamp, (full_val, cut_val) in libraries:
-            button = QPushButton(text=cut_val)
-            button.setObjectName(f"path{tstamp}")
+        row_layout = QHBoxLayout()
+        row_layout.addWidget(label)
+        layout.addLayout(row_layout)
 
-            def open_library_button_clicked(path):
-                return lambda: self.driver.open_library(path)
+        def get_button_style(extras: list[str] | None = None):
+            base_style = [
+                f"background-color:{Theme.COLOR_BG.value};",
+                "border-radius:6px;",
+                "text-align: left;",
+                "padding-top: 3px;",
+                "padding-left: 6px;",
+                "padding-bottom: 4px;",
+            ]
 
-            button.clicked.connect(open_library_button_clicked(full_val))
-            button.setStyleSheet(
+            full_style = base_style + (extras or [])
+
+            return (
                 "QPushButton{"
-                f"background-color:{Theme.COLOR_BG.value};"
-                "border-radius:6px;"
-                "text-align: left;"
-                "padding-top: 3px;"
-                "padding-left: 6px;"
-                "padding-bottom: 4px;"
+                f"{''.join(full_style)}"
                 "}"
                 f"QPushButton::hover{{background-color:{Theme.COLOR_HOVER.value};}}"
                 f"QPushButton::pressed{{background-color:{Theme.COLOR_PRESSED.value};}}"
             )
+
+        for item_key, (full_val, cut_val) in libraries:
+            button = QPushButton(text=cut_val)
+            button.setObjectName(f"path{item_key}")
+
+            def open_library_button_clicked(path):
+                return lambda: self.driver.open_library(path)
+
+            button.clicked.connect(open_library_button_clicked(full_val))
+            button.setStyleSheet(get_button_style())
             button.setCursor(Qt.CursorShape.PointingHandCursor)
-            layout.addWidget(button)
+
+            button_remove = QPushButton("➖")
+            button_remove.setStyleSheet(get_button_style())
+            button_remove.setCursor(Qt.CursorShape.PointingHandCursor)
+            button_remove.setFixedWidth(30)
+
+            def remove_recent_library_clicked(key: str):
+                return lambda: (
+                    self.driver.remove_recent_library(key),
+                    self.fill_libs_widget(self.libs_layout),
+                )
+
+            button_remove.clicked.connect(remove_recent_library_clicked(item_key))
+
+            row_layout = QHBoxLayout()
+            row_layout.addWidget(button)
+            row_layout.addWidget(button_remove)
+
+            layout.addLayout(row_layout)
 
     def resizeEvent(self, event: QResizeEvent) -> None:
         self.update_image_size(