add body battery samples

tryterra · Aug 23, 2024 · e0d2371 · e0d2371
1 parent d176612
commit e0d2371
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/terra/models/v2/daily.py b/terra/models/v2/daily.py
@@ -81,6 +81,7 @@ class StressData(base_model.TerraDataModel):
     avg_stress_level: typing.Optional[float] = dataclasses.field(default=None)
     max_stress_level: typing.Optional[float] = dataclasses.field(default=None)
     samples: typing.List[samples_.StressSample] = dataclasses.field(default_factory=list)
+    body_battery_samples: typing.List[samples_.BodyBatterySample] = dataclasses.field(default_factory=list)
 
 
 @dataclasses.dataclass

diff --git a/terra/models/v2/samples.py b/terra/models/v2/samples.py
@@ -49,6 +49,7 @@
     "PulseVelocitySample",
     "AFibClassificationSample",
     "DrinkSample",
+    "BodyBatterySample",
 ]
 
 
@@ -322,3 +323,9 @@ class DrinkSample(base_model.TerraDataModel):
 class MenstruationFlowSample(base_model.TerraDataModel):
     timestamp: typing.Optional[str] = dataclasses.field(default=None)
     flow: typing.Optional[int] = dataclasses.field(default=None)
+
+@dataclasses.dataclass
+class BodyBatterySample(base_model.TerraDataModel):
+    timestamp: typing.Optional[str] = dataclasses.field(default=None)
+    level: typing.Optional[float] = dataclasses.field(default=None)
+