Add Helper Functions For Managing Roles

Signed-off-by: Hassan Abouelela <[email protected]>

1 files changed, 36 insertions, 0 deletions

diff --git a/backend/routes/roles.py b/backend/routes/roles.py
new file mode 100644
index 0000000..b18a04b
--- /dev/null
+++ b/backend/routes/roles.py
@@ -0,0 +1,36 @@
+import starlette.background
+from pymongo.database import Database
+from spectree import Response
+from starlette.authentication import requires
+from starlette.responses import JSONResponse
+from starlette.routing import Request
+
+from backend import discord, route
+from backend.validation import OkayResponse, api
+
+
+async def refresh_roles(database: Database) -> None:
+    """Connect to the discord API and refresh the roles database."""
+    roles = await discord.get_role_info()
+    roles_collection = database.get_collection("roles")
+    roles_collection.drop()
+    roles_collection.insert_many([role.dict() for role in roles])
+
+
+class RolesRoute(route.Route):
+    """Refreshes the roles database."""
+
+    name = "roles"
+    path = "/roles"
+
+    @requires(["authenticated", "admin"])
+    @api.validate(
+        resp=Response(HTTP_200=OkayResponse),
+        tags=["roles"]
+    )
+    async def patch(self, request: Request) -> JSONResponse:
+        """Refresh the roles database."""
+        return JSONResponse(
+            {"status": "ok"},
+            background=starlette.background.BackgroundTask(refresh_roles, request.state.db)
+        )