Fix port_forward.enable key not always exist (#476)

Kane610 · Sep 3, 2023 · 95b5ada · 95b5ada
1 parent 75c3cc7
commit 95b5ada
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/aiounifi/models/port_forward.py b/aiounifi/models/port_forward.py
@@ -1,6 +1,6 @@
 """Port forwarding in a UniFi network."""
 from dataclasses import dataclass
-from typing import Self, TypedDict
+from typing import NotRequired, Self, TypedDict
 
 from .api import ApiItem, ApiRequest
 
@@ -10,7 +10,7 @@ class TypedPortForward(TypedDict):
 
     _id: str
     dst_port: str
-    enabled: bool
+    enabled: NotRequired[bool]
     fwd_port: str
     fwd: str
     name: str
@@ -64,7 +64,7 @@ def destination_port(self) -> str:
     @property
     def enabled(self) -> bool:
         """Is port forward enabled."""
-        return self.raw["enabled"]
+        return self.raw.get("enabled", False)
 
     @property
     def forward_port(self) -> str: