From 5f90fc4b56dc571dbb757fc674e8ee11ed8f2a38 Mon Sep 17 00:00:00 2001
From: Andrey <andrey@simiotics.com>
Date: Mon, 20 Nov 2023 11:34:35 +0200
Subject: [PATCH] Add publish to request body.

---
 engineapi/engineapi/data.py               | 4 ++++
 engineapi/engineapi/routes/leaderboard.py | 8 ++++----
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/engineapi/engineapi/data.py b/engineapi/engineapi/data.py
index acf1fa6e..6b7e3f82 100644
--- a/engineapi/engineapi/data.py
+++ b/engineapi/engineapi/data.py
@@ -450,3 +450,7 @@ class LeaderboardVersion(BaseModel):
     published: bool
     created_at: datetime
     updated_at: datetime
+
+
+class LeaderboardVersionUpdateRequest(BaseModel):
+    publish: bool
diff --git a/engineapi/engineapi/routes/leaderboard.py b/engineapi/engineapi/routes/leaderboard.py
index f0079a48..01c93d7e 100644
--- a/engineapi/engineapi/routes/leaderboard.py
+++ b/engineapi/engineapi/routes/leaderboard.py
@@ -1092,9 +1092,9 @@ async def update_leaderboard_version_handler(
     request: Request,
     leaderboard_id: UUID = Path(..., description="Leaderboard ID"),
     version: int = Path(..., description="Version of the leaderboard."),
-    publish: bool = Query(
-        False,
-        description="If enabled, this will publish the leaderboard version.",
+    request_body: data.LeaderboardVersionUpdateRequest = Body(
+        ...,
+        description="JSON object specifying whether to publish or unpublish version.",
     ),
     db_session: Session = Depends(db.yield_db_session),
     Authorization: str = AuthHeader,
@@ -1125,7 +1125,7 @@ async def update_leaderboard_version_handler(
             db_session=db_session,
             leaderboard_id=leaderboard_id,
             version_number=version,
-            published=publish,
+            published=request_body.publish,
         )
     except Exception as e:
         logger.error(f"Error while updating leaderboard version: {e}")