Added float field, and more comments on the steps of label creation a…

…nd saving Signed-off-by: Aaron Chong <[email protected]>
open-rmf · May 20, 2024 · 7c2d8ef · 7c2d8ef
1 parent bb95202
commit 7c2d8ef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/packages/api-server/api_server/models/tortoise_models/tasks.py b/packages/api-server/api_server/models/tortoise_models/tasks.py
@@ -2,6 +2,7 @@
     BigIntField,
     CharField,
     DatetimeField,
+    FloatField,
     ForeignKeyField,
     ForeignKeyRelation,
     JSONField,
@@ -38,6 +39,7 @@ class TaskLabel(Model):
     label_name = CharField(255, null=False, index=True)
     label_value_str = CharField(255, null=True, index=True)
     label_value_num = BigIntField(null=True, index=True)
+    label_value_float = FloatField(null=True, index=True)
 
 
 class TaskEventLog(Model):

diff --git a/packages/api-server/api_server/repositories/tasks.py b/packages/api-server/api_server/repositories/tasks.py
@@ -99,10 +99,11 @@ async def save_task_state(self, task_state: TaskState) -> None:
             )
             return
 
+        # Since this is updating an existing task state, we are done
         if not created:
             return
 
-        # Save the labels that we want
+        # Labels are created and saved when a new task state is first received
         labels = task_state.booking.labels
         booking_label = None
         if labels is not None: