uiprotect 3.1.6__tar.gz → 3.1.7__tar.gz

{uiprotect-3.1.6 → uiprotect-3.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: uiprotect
-Version: 3.1.6
+Version: 3.1.7
 Summary: Python API for Unifi Protect (Unofficial)
 Home-page: https://github.com/uilibs/uiprotect
 License: MIT

{uiprotect-3.1.6 → uiprotect-3.1.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "uiprotect"
-version = "3.1.6"
+version = "3.1.7"
 description = "Python API for Unifi Protect (Unofficial)"
 authors = ["UI Protect Maintainers <ui@koston.org>"]
 license = "MIT"

{uiprotect-3.1.6 → uiprotect-3.1.7}/src/uiprotect/data/websocket.py

@@ -204,10 +204,9 @@ class WSPacket:
     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
 
@@ -215,10 +214,9 @@ class WSPacket:
     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