return blocked sites

ashdeck · Oct 17, 2024 · 035c35a · 035c35a
1 parent b6a4af8
commit 035c35a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,4 +3,3 @@
 __pycache__/
 .pytest_cache/
 .vscode/
-/__pycache__
diff --git a/sessions/__pycache__/routes.cpython-39.pyc b/sessions/__pycache__/routes.cpython-39.pyc
diff --git a/sessions/routes.py b/sessions/routes.py
@@ -41,8 +41,16 @@ async def get_sessions(user=Depends(get_current_user)):
 async def get_active_sessions():
     sessions = db["sessions"].find({"end_time": {"$gte": datetime.now(tz=timezone.utc)}})
     if sessions:
-        return list(sessions)
-    return []
+        sessions = list(sessions)
+        for i in sessions:
+            if i["block_lists"]:
+                block_lists = db["block_lists"].find({"_id": {"$in": i["block_lists"]}})
+                if block_lists:
+                    block_lists = list(block_lists)
+                    for block_list in block_lists:
+                        i["block_lists"] = block_list["entries"]
+            i["id"] = i.pop("_id")
+        return sessions
 
 
 @router.get("/{id}", status_code=200)