Add pressure payload

equinor · Mar 16, 2023 · 9b96b48 · 9b96b48
1 parent 61160ad
commit 9b96b48
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/src/robot_interface/telemetry/payloads.py b/src/robot_interface/telemetry/payloads.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 from datetime import datetime
-from typing import List, Union
+from typing import List
 
 from alitra import Pose
 from transitions import State
@@ -9,19 +9,25 @@
 
 
 @dataclass
-class TelemetryPosePayload:
-    pose: Pose
+class TelemetryPayload:
     isar_id: str
     robot_name: str
     timestamp: datetime
 
 
 @dataclass
-class TelemetryBatteryPayload:
+class TelemetryPosePayload(TelemetryPayload):
+    pose: Pose
+
+
+@dataclass
+class TelemetryBatteryPayload(TelemetryPayload):
     battery_level: float
-    isar_id: str
-    robot_name: str
-    timestamp: datetime
+
+
+@dataclass
+class TelemetryPressurePayload(TelemetryPayload):
+    pressure_level: float
 
 
 @dataclass