Admin pages iterations

app/main.py

@@ -1,8 +1,9 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, Form, File, UploadFile, Depends
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
-from fastapi.responses import FileResponse
-
+from fastapi.responses import FileResponse, HTMLResponse, RedirectResponse
+from fastapi.templating import Jinja2Templates
+from .admin import admin_functions as admin
 
 app = FastAPI()
 
@@ -24,4 +25,27 @@ templates = Jinja2Templates(directory="app/admin/templates")
 
 @app.get("/")
 async def get_admin_login(request: Request):
-    return templates.TemplateResponse("admin_login.html", {"request": request})
+    return templates.TemplateResponse("admin_login.html", {"request": request})
+
+# Admin Login Handler
+@app.post("/admin/login", response_class=HTMLResponse)
+async def handle_admin_login(request: Request, password: str = Form(...)):
+    # Assume get_admin_password_hash fetches the admin's password hash securely
+    stored_password_hash = "securely_stored_password_hash"
+    if admin.verify_admin_password(password, stored_password_hash):
+        # Redirect to user registration page upon successful login
+        return RedirectResponse(url="/admin/register_user", status_code=303)
+    else:
+        # Reload login page with error message
+        return templates.TemplateResponse("admin_login.html", {"request": request, "error": "Invalid password"})
+
+# User Registration Handler
+@app.post("/admin/register_user", response_class=HTMLResponse)
+async def handle_user_registration(request: Request, email: str = Form(...), name: str = Form(...), role: str = Form(...), file: UploadFile = File(...)):
+    user_id = await admin.register_user(email, name, role, file)
+    if user_id:
+        # Redirect or display a success message
+        return templates.TemplateResponse("registration_success.html", {"request": request})
+    else:
+        # Reload registration page with error message
+        return templates.TemplateResponse("user_registration.html", {"request": request, "error": "Registration failed"})