fix(stats); check if frame stats is None (#71)

ooliver1 · May 3, 2023 · 978918e · 978918e
1 parent 1f78c7c
commit 978918e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/mafic/stats.py b/mafic/stats.py
@@ -119,5 +119,7 @@ def __init__(self, data: Stats) -> None:
         self.memory: MemoryStats = MemoryStats(data["memory"])
         self.cpu: CPUStats = CPUStats(data["cpu"])
         self.frame_stats: FrameStats | None = (
-            FrameStats(data["frameStats"]) if "frameStats" in data else None
+            FrameStats(data["frameStats"])
+            if "frameStats" in data and data["frameStats"] is not None
+            else None
         )
diff --git a/mafic/typings/common.py b/mafic/typings/common.py
@@ -171,4 +171,5 @@ class Stats(TypedDict):
     uptime: int
     memory: Memory
     cpu: CPU
-    frameStats: NotRequired[FrameStats]
+    # V3 is NotRequired, V4 is None
+    frameStats: NotRequired[FrameStats | None]