wip

blitz/ui/components/base.py

@@ -1,7 +1,16 @@
-from typing import Any, Self, TypeVar
+from typing import Any, Generic, Self, TypeVar, Protocol
 from blitz.ui.blitz_ui import BlitzUI, get_blitz_ui
 
-T = TypeVar("T", bound="BaseComponent")
+
+class NiceGUIComponent(Protocol):
+    def __enter__(self) -> Any:
+        ...
+
+    def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
+        ...
+
+
+V = TypeVar("V", bound=NiceGUIComponent)
 
 
 # Get the blitz_ui through a metaclass
@@ -12,8 +21,9 @@ def __new__(cls, name: str, bases: tuple[type, ...], namespace: dict[str, Any])
         return super().__new__(cls, name, bases, namespace)
 
 
-class BaseComponent(metaclass=BaseComponentMeta):
+class BaseComponent(Generic[V], metaclass=BaseComponentMeta):
     def __init__(self, *args: Any, props: str = "", classes: str = "", **kwargs: Any) -> None:
+        self._ng: V
         self.props: str
         self.classes: str
         self.blitz_ui: BlitzUI
@@ -35,6 +45,14 @@ def __init__(self, *args: Any, props: str = "", classes: str = "", **kwargs: Any
     def render(self) -> None:
         raise NotImplementedError
 
+    @property
+    def ng(self) -> V:
+        return self._ng
+
+    @ng.setter
+    def ng(self, value: V) -> None:
+        self._ng = value
+
     @classmethod
     def variant(cls, name: str, *, props: str = "", classes: str = "", **kwargs: Any) -> type[Self]:
         """
@@ -58,3 +76,10 @@ def variant(cls, name: str, *, props: str = "", classes: str = "", **kwargs: Any
                 "classes": classes,
             },
         )
+
+    def __enter__(self) -> V:
+        return self.ng
+
+    def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
+        if hasattr(self.ng, "__exit__"):
+            self.ng.__exit__(exc_type, exc_value, traceback)

blitz/ui/components/buttons/base.py

@@ -3,7 +3,7 @@
 from blitz.ui.components.base import BaseComponent
 
 
-class BaseButton(BaseComponent):
+class BaseButton(BaseComponent[ui.button]):
     def __init__(
         self,
         text: str = "",
@@ -21,9 +21,13 @@ def __init__(
         super().__init__(props=props, classes=classes)
 
     def render(self) -> None:
-        ui.button(
-            self.text,
-            on_click=self.on_click,
-            color=self.color,
-            icon=self.icon,
-        ).props(self.props).classes(self.classes)
+        self.ng = (
+            ui.button(
+                self.text,
+                on_click=self.on_click,
+                color=self.color,
+                icon=self.icon,
+            )
+            .props(self.props)
+            .classes(self.classes)
+        )

blitz/ui/components/buttons/flat.py

@@ -1,4 +1,4 @@
 from .base import BaseButton
 
 
-FlatButton = BaseButton.variant("Flat", props="flat")
+FlatButton = BaseButton.variant("Flat", props="flat")