telegrinder 0.1.dev158__py3-none-any.whl → 0.1.dev159__py3-none-any.whl

telegrinder/bot/cute_types/update.py

@@ -1,22 +1,29 @@
+import typing
+
 from fntypes.co import Nothing, Some
 
 from telegrinder.api import ABCAPI
 from telegrinder.msgspec_utils import Option
-from telegrinder.types import Update, UpdateType
+from telegrinder.types import Model, Update
 
 from .base import BaseCute
 
+ModelT = typing.TypeVar("ModelT", bound=Model)
+
 
 class UpdateCute(BaseCute[Update], Update, kw_only=True):
     api: ABCAPI
 
     @property
-    def update_type(self) -> Option[UpdateType]:
-        for name, update in self.to_dict(
-            exclude_fields={"update_id"},
-        ).items():
-            if update is not None:
-                return Some(UpdateType(name))
+    def incoming_update(self) -> Model:
+        return getattr(
+            self,
+            self.update_type.expect("Update object has no incoming update.").value,
+        )
+    
+    def get_event(self, event_model: type[ModelT]) -> Option[ModelT]:
+        if isinstance(self.incoming_update, event_model):
+            return Some(self.incoming_update)
         return Nothing()