fix: backend cleanup

apps/api/src/routers/demo.py

@@ -1,4 +1,4 @@
-from typing import Annotated, Union, Optional, Any
+from typing import Annotated, Union, Any
 
 from fastapi import APIRouter, Cookie, Form
 
@@ -20,10 +20,12 @@ async def square(value: Annotated[int, Form()]) -> int:
 
 
 @router.post("/user")
-async def get_user(name: Annotated[str, Form()]) -> list[dict[str, Any]]:
-    results = await retrieve(Collection.USERS, {"name": name}, ["name", "ucinetid"])
-    for i in range(len(results)):
-        results[i]["_id"] = str(results[i]["_id"])
+async def get_user(search_name: Annotated[str, Form()]) -> list[dict[str, object]]:
+    results = await retrieve(
+        Collection.USERS, {"name": search_name}, ["name", "ucinetid"]
+    )
+    for result in results:
+        result["_id"] = str(result["_id"])
     return results
 
 

apps/api/src/services/mongodb_handler.py

@@ -2,7 +2,6 @@
 import os
 from enum import Enum
 from logging import getLogger
-
 from typing import Any, Mapping, Optional, Union
 
 from bson import CodecOptions
@@ -40,7 +39,6 @@ class Config:
 class Collection(str, Enum):
     USERS = "users"
     TESTING = "testing"
-
     SETTINGS = "settings"
 
 
@@ -73,12 +71,10 @@ async def retrieve(
 ) -> list[dict[str, object]]:
     """Search for and retrieve the specified fields of a document (if any exist)
     that satisfy the provided query."""
-
     COLLECTION = DB[collection.value]
 
     result = COLLECTION.find(query, fields)
     output: list[dict[str, object]] = await result.to_list(length=None)
-
     return output
 
 
@@ -105,7 +101,6 @@ async def raw_update_one(
     result: UpdateResult = await COLLECTION.update_one(query, update, upsert=upsert)
     if not result.acknowledged:
         log.error("MongoDB document update was not acknowledged")
-
         raise RuntimeError("Could not update documents in MongoDB collection")
 
     return result.modified_count > 0