Update StatusPTH schema to accept lists for Max thresholds

jziolkowski · Sep 28, 2024 · 1a4229a · 1a4229a
1 parent a1b4e86
commit 1a4229a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tdmgr/schemas/status.py b/tdmgr/schemas/status.py
@@ -137,15 +137,15 @@ class StatusMQTSchema(StatusBaseModel):
 class StatusPTHSchema(StatusBaseModel):
     CurrentHigh: Union[int, List[int]]
     CurrentLow: Union[int, List[int]]
-    MaxEnergy: Optional[int] = None
-    MaxEnergyStart: Optional[int] = None
-    MaxPower: Optional[int] = None
-    MaxPowerHold: Optional[int] = None
-    MaxPowerWindow: Optional[int] = None
+    MaxEnergy: Optional[Union[int, List[int]]] = None
+    MaxEnergyStart: Optional[Union[int, List[int]]] = None
+    MaxPower: Optional[Union[int, List[int]]] = None
+    MaxPowerHold: Optional[Union[int, List[int]]] = None
+    MaxPowerWindow: Optional[Union[int, List[int]]] = None
     PowerDelta: Union[int, List[int]]
     PowerHigh: Union[int, List[int]]
     PowerLow: Union[int, List[int]]
-    VoltageHigh: Optional[int] = None
+    VoltageHigh: Optional[Union[int, List[int]]] = None
     VoltageLow: Union[int, List[int]]