chore: Add ToastType type alias

Author: gadenbuie

shiny/playwright/controller/_toast.py

@@ -3,7 +3,7 @@
 from playwright.sync_api import Locator, Page
 from playwright.sync_api import expect as playwright_expect
 
-from ...ui._toast import _normalize_toast_position, _normalize_toast_type
+from ...ui._toast import ToastType, _normalize_toast_position, _normalize_toast_type
 from .._types import PatternOrStr, Timeout
 from ..expect._internal import (
     expect_attribute_to_have_value as _expect_attribute_to_have_value,
@@ -94,7 +94,7 @@ def expect_header(self, value: PatternOrStr, *, timeout: Timeout = None) -> None
         """
         playwright_expect(self.loc_header).to_have_text(value, timeout=timeout)
 
-    def expect_type(self, value: str, *, timeout: Timeout = None) -> None:
+    def expect_type(self, value: ToastType, *, timeout: Timeout = None) -> None:
         """
         Expects the toast to have the specified type class.
 

shiny/ui/_toast.py

@@ -28,6 +28,19 @@
 if TYPE_CHECKING:
     from ..session import Session
 
+# Valid toast types
+ToastType = Literal[
+    "primary",
+    "secondary",
+    "success",
+    "info",
+    "warning",
+    "danger",
+    "error",  # alias for "danger", handled in normalization
+    "light",
+    "dark",
+]
+
 
 class ToastPayload(RenderedDeps):
     id: str
@@ -50,7 +63,7 @@ def __init__(
         header: Optional[ToastHeader | TagNode] = None,
         icon: Optional[TagNode] = None,
         id: Optional[str] = None,
-        type: Optional[str] = None,
+        type: Optional[ToastType] = None,
         duration: Optional[float] = 5000,
         position: str = "top-right",
         closable: bool = True,
@@ -174,19 +187,7 @@ def toast(
     header: Optional[str | ToastHeader | TagNode] = None,
     icon: Optional[TagNode] = None,
     id: Optional[str] = None,
-    type: Optional[
-        Literal[
-            "primary",
-            "secondary",
-            "success",
-            "info",
-            "warning",
-            "danger",
-            "error",
-            "light",
-            "dark",
-        ]
-    ] = None,
+    type: Optional[ToastType] = None,
     duration_s: Optional[int | float] = 5,
     position: str | list[str] | tuple[str, ...] = "top-right",
     closable: bool = True,
@@ -479,14 +480,14 @@ def hide_toast(
 
 
 @overload
-def _normalize_toast_type(type: str) -> str: ...
+def _normalize_toast_type(type: ToastType) -> ToastType: ...
 
 
 @overload
 def _normalize_toast_type(type: None) -> None: ...
 
 
-def _normalize_toast_type(type: Optional[str]) -> Optional[str]:
+def _normalize_toast_type(type: Optional[ToastType]) -> Optional[ToastType]:
     """Normalize toast type, converting "error" to "danger"."""
     if type == "error":
         return "danger"