Create new endpoint for checking service availability

via `/status`
beeldengeluid · Oct 3, 2024 · f3c15a1 · f3c15a1
1 parent 5f13238
commit f3c15a1
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/whisper_api.py b/whisper_api.py
@@ -97,12 +97,23 @@ def get_all_tasks():
     return {"data": all_tasks}
 
 
+@api.get("/status")
+def get_status(response: Response):
+    global current_task
+
+    if current_task and current_task.status == Status.PROCESSING:
+        response.status_code = status.HTTP_503_SERVICE_UNAVAILABLE
+        return {"msg": "The worker is currently processing a task. Try again later!"}
+
+    response.status_code = status.HTTP_200_OK
+    return {"msg": "The worker is available!"}
+
+
 @api.post("/tasks", status_code=status.HTTP_201_CREATED)
 async def create_task(
     task: Task, background_tasks: BackgroundTasks, response: Response
 ):
     global current_task
-    print(current_task)
     if current_task and current_task.status == Status.PROCESSING:
         response.status_code = status.HTTP_503_SERVICE_UNAVAILABLE
         return {"msg": "The worker is currently processing a task. Try again later!"}