From 7c2d8efa27ad3edacc438dec24e756bda3c2baa9 Mon Sep 17 00:00:00 2001 From: Aaron Chong Date: Mon, 20 May 2024 15:03:20 +0800 Subject: [PATCH] Added float field, and more comments on the steps of label creation and saving Signed-off-by: Aaron Chong --- packages/api-server/api_server/models/tortoise_models/tasks.py | 2 ++ packages/api-server/api_server/repositories/tasks.py | 3 ++- 2 files changed, 4 insertions(+), 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 index 3111fc95a..6e3f3e665 100644 --- 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 index 0bb993788..c857abfea 100644 --- 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: