jbqt 0.1.14__tar.gz → 0.1.15__tar.gz

{jbqt-0.1.14 → jbqt-0.1.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: jbqt
-Version: 0.1.14
+Version: 0.1.15
 Summary: 
 Author: Joseph Bochinski
 Author-email: stirgejr@gmail.com
@@ -12,6 +12,7 @@ Requires-Dist: jbconsts (>=0.1.1,<0.2.0)
 Requires-Dist: jbutils (>=0.1.23,<0.2.0)
 Requires-Dist: pyqt6 (>=6.9.0,<7.0.0)
 Requires-Dist: python-levenshtein (>=0.27.1,<0.28.0)
+Requires-Dist: rich (>=14.0.0,<15.0.0)
 Description-Content-Type: text/markdown
 
 

jbqt-0.1.15/jbqt/models/__init__.py

@@ -0,0 +1,14 @@
+"""Model exports"""
+
+from jbqt.models.chips import IChipsWidget
+from jbqt.models.chip_button import IChipButton
+from jbqt.models.model_consts import RegisteredFunctions
+from jbqt.models.model_utils import get_fct, register_fct
+
+__all__ = [
+    "get_fct",
+    "IChipButton",
+    "IChipsWidget",
+    "register_fct",
+    "RegisteredFunctions",
+]

jbqt-0.1.15/jbqt/models/model_consts.py

@@ -0,0 +1,9 @@
+"""Constant values local to the models package"""
+
+from jbutils.types import Function
+
+
+RegisteredFunctions: dict[str, Function] = {}
+
+
+__all__ = ["RegisteredFunctions"]

jbqt-0.1.15/jbqt/models/model_utils.py

@@ -0,0 +1,24 @@
+"""Collection of utility functions local to the models package"""
+
+from jbutils.types import Function
+from rich import print
+
+from jbqt.models.model_consts import RegisteredFunctions
+
+
+def register_fct(name: str, fct: Function, override: bool = False) -> None:
+    if name not in RegisteredFunctions:
+        RegisteredFunctions[name] = fct
+        if override and name in RegisteredFunctions:
+            # TODO improve logging
+            print(
+                f"[yellow][Warning][/yellow]: Function '{name}' already defined and was overwritten"
+            )
+    print(f"[yellow][Warning][/yellow]: Function '{name}' already defined")
+
+
+def get_fct(name: str, fallback: Function | None = None) -> Function | None:
+    return RegisteredFunctions.get(name, fallback)
+
+
+__all__ = ["register_fct", "get_fct"]

jbqt-0.1.15/jbqt/models/toolbar_button.py

@@ -0,0 +1,58 @@
+"""Collection of models for PyQt6"""
+
+from collections.abc import Callable
+from dataclasses import dataclass
+
+from PyQt6.QtCore import QSize, QObject
+from PyQt6.QtGui import QIcon, QColor, QAction
+
+from jbconsts import COLORS
+from jbqt.consts import ICONS, QtIconSizes
+from jbqt.models import model_utils
+
+
+@dataclass
+class ToolbarButton:
+    name: str = ""
+    icon: QIcon | str = ""
+    color: QColor | str = COLORS.BLUE_GRAY
+    checkable: bool = False
+    checked: bool = False
+    status_tip: str = ""
+    size: QSize | str = QtIconSizes.ICON_MD
+    function: Callable[..., None] | str | None = None
+    text: str = ""
+
+    def __post_init__(self) -> None:
+        if isinstance(self.function, str):
+            fct = model_utils.get_fct(self.function)
+            print("fct:", fct)
+            self.function = fct if fct else None
+
+    def to_action(self, parent: QObject | None = None) -> QAction:
+        color: QColor | str = self.color
+        if isinstance(color, str):
+            color = COLORS.get(color, color)
+
+        size = self.size
+        if isinstance(size, str):
+            size = QtIconSizes.get(size)
+
+        icon = self.icon
+        if isinstance(icon, str):
+            icon_getter = ICONS.get(icon)
+            if callable(icon_getter):
+                icon = icon_getter(color, size)
+
+            icon = QIcon() if isinstance(icon, str) else icon
+
+        action = QAction(icon=icon, text=self.text, parent=parent)
+        if self.checkable:
+            action.setCheckable(True)
+            action.setChecked(self.checkable)
+        if self.status_tip:
+            action.setStatusTip(self.status_tip)
+        if self.function:
+            action.triggered.connect(self.function)
+
+        return action

{jbqt-0.1.14 → jbqt-0.1.15}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "jbqt"
-version = "0.1.14"
+version = "0.1.15"
 description = ""
 authors = [ "Joseph Bochinski <stirgejr@gmail.com>",]
 readme = "README.md"
@@ -16,3 +16,4 @@ fuzzywuzzy = "^0.18.0"
 python-levenshtein = "^0.27.1"
 jbconsts = "^0.1.1"
 jbutils = "^0.1.23"
+rich = "^14.0.0"

jbqt-0.1.14/jbqt/models/__init__.py

@@ -1,6 +0,0 @@
-"""Model exports"""
-
-from jbqt.models.chips import IChipsWidget
-from jbqt.models.chip_button import IChipButton
-
-__all__ = ["IChipButton", "IChipsWidget"]