chore: add sunset date to cluster list

Merge pull request #91 from Getdeck/feat/cluster-list-sunset * chore: add sunset date to cluster list
Getdeck · Jul 4, 2023 · 699d98a · 699d98a
2 parents 07dbb1d + e662018
commit 699d98a
Show file tree

Hide file tree

Showing 3 changed files with 80 additions and 16 deletions.
diff --git a/app/beiboot_api/cluster/types.py b/app/beiboot_api/cluster/types.py
@@ -466,6 +466,7 @@ class ClusterStateResponse(BaseModel):
     id: str
     name: str | None
     state: BeibootState | None
+    sunset: datetime | None
 
 
 class ClusterInfoResponse(BaseModel):

diff --git a/app/beiboot_api/routers/clusters.py b/app/beiboot_api/routers/clusters.py
@@ -74,7 +74,13 @@ async def cluster_list(
 
     response = []
     for beiboot in beiboots:
-        response.append(ClusterStateResponse(id=beiboot.name, name=beiboot.labels.get("name"), state=beiboot.state))
+        if beiboot.sunset:
+            sunset = datetime.strptime(beiboot.sunset, "%Y-%m-%dT%H:%M:%S.%fZ")
+        else:
+            sunset = None
+        response.append(
+            ClusterStateResponse(id=beiboot.name, name=beiboot.labels.get("name"), state=beiboot.state, sunset=sunset)
+        )
 
     return paginate(response, params)