Update keys endpoint

avgupta456 · avgupta456 · commit 5b10c9914afd · 2024-11-29T18:55:05.000-05:00
diff --git a/backend/src/data/mongo/secret/functions.py b/backend/src/data/mongo/secret/functions.py
@@ -21,9 +21,9 @@ async def get_keys(no_cache: bool = False) -> Tuple[bool, List[str]]:
 secret_keys: List[str] = []
 
 
-async def update_keys() -> None:
+async def update_keys(no_cache: bool = False) -> None:
     global secret_keys
-    secret_keys = await get_keys()
+    secret_keys = await get_keys(no_cache=no_cache)
 
 
 def get_random_key() -> str:
diff --git a/backend/src/routers/users/db.py b/backend/src/routers/users/db.py
@@ -2,12 +2,25 @@
 
 from fastapi import APIRouter, Response, status
 
+from src.data.mongo.secret import update_keys
 from src.data.mongo.user import PublicUserModel, get_public_user as db_get_public_user
 from src.utils import async_fail_gracefully
 
 router = APIRouter()
 
 
+@router.get(
+    "/update_keys",
+    status_code=status.HTTP_200_OK,
+    include_in_schema=False,
+    response_model=Dict[str, Any],
+)
+@async_fail_gracefully
+async def update_keys_endpoint(response: Response) -> bool:
+    await update_keys(no_cache=True)
+    return True
+
+
 @router.get(
     "/get/metadata/{user_id}",
     status_code=status.HTTP_200_OK,
