fix: remove unreachable code in the websocket decoder (#112)

uilibs · Jun 23, 2024 · 235cdef · 235cdef
1 parent aa3bf3f
commit 235cdef
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/src/uiprotect/data/websocket.py b/src/uiprotect/data/websocket.py
@@ -204,21 +204,19 @@ def decode(self) -> None:
     def action_frame(self) -> BaseWSPacketFrame:
         if self._action_frame is None:
             self.decode()
-
-        if self._action_frame is None:
-            raise WSDecodeError("Packet unexpectedly not decoded")
-
+        if TYPE_CHECKING:
+            assert self._action_frame is not None
+            assert self._data_frame is not None
         self.__dict__["data_frame"] = self._data_frame
         return self._action_frame
 
     @cached_property
     def data_frame(self) -> BaseWSPacketFrame:
         if self._data_frame is None:
             self.decode()
-
-        if self._data_frame is None:
-            raise WSDecodeError("Packet unexpectedly not decoded")
-
+        if TYPE_CHECKING:
+            assert self._action_frame is not None
+            assert self._data_frame is not None
         self.__dict__["action_frame"] = self._action_frame
         return self._data_frame