Update StatusPTH schema to accept lists for Max thresholds (#293)

jziolkowski · Sep 28, 2024 · a2b070c · a2b070c
1 parent a1b4e86
commit a2b070c
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]]