nouveau Back

main.py

@@ -1,50 +1,58 @@
-from fastapi import FastAPI, HTTPException, Depends
+# backend.py
+from fastapi import FastAPI, Request, Depends, HTTPException
+from fastapi.responses import HTMLResponse, RedirectResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 from pymongo import MongoClient
-from pydantic import BaseModel
-from bson.objectid import ObjectId
-import hashlib
-import os
+from starlette.middleware.sessions import SessionMiddleware
 
 app = FastAPI()
+app.mount("/static", StaticFiles(directory="static"), name="static")
+app.add_middleware(SessionMiddleware, secret_key="votre_secret_key_secure")
 
-# Connexion à MongoDB
-MONGO_URL = "mongodb+srv://lebaykserver:mVef8e5F1GRtbA14@test.6bzxp.mongodb.net/"
-client = MongoClient(MONGO_URL)
-db = client["IA_SIGNATURE"]  # Remplace "mydatabase" par le nom de ta base
-users_collection = db["USER"]
-
-# Modèle de connexion
-class LoginModel(BaseModel):
-    name: str
-    etablissement: str
-    hashed_password: str
-
-def hash_password(password: str) -> str:
-    return hashlib.sha256(password.encode()).hexdigest()
-
-@app.post("/login")
-def login(user: LoginModel):
-    hashed_password = hash_password(user.hashed_password)
-    found_user = users_collection.find_one({
-        "name": user.name,
-        "etablissement": user.etablissement,
-        "hashed_password": hashed_password
+# Configuration MongoDB
+client = MongoClient("mongodb+srv://lebaykserver:mVef8e5F1GRtbA14@test.6bzxp.mongodb.net/")
+db = client["auth_demo"]
+users = db["users"]
+establishments = db["establishments"]
+
+# Créer quelques données de test
+if establishments.count_documents({}) == 0:
+    establishments.insert_many([{"name": "École 1"}, {"name": "Université A"}, {"name": "Lycée B"}])
+
+@app.get("/", response_class=HTMLResponse)
+async def root(request: Request):
+    if not request.session.get("user"):
+        return RedirectResponse(url="/login")
+    return templates.TemplateResponse("dashboard.html", {"request": request})
+
+@app.get("/login", response_class=HTMLResponse)
+async def login_page(request: Request):
+    if request.session.get("user"):
+        return RedirectResponse(url="/")
+    return templates.TemplateResponse("login.html", {"request": request})
+
+@app.get("/api/establishments")
+async def get_establishments():
+    return [e["name"] for e in establishments.find({}, {"_id": 0, "name": 1})]
+
+@app.get("/api/login")
+async def login(request: Request, username: str, password: str, establishment: str):
+    user = users.find_one({
+        "username": username,
+        "password": password,
+        "establishment": establishment
     })
-    
-    if not found_user:
-        raise HTTPException(status_code=401, detail="Identifiants invalides")
-    
-    return {"message": "Connexion réussie", "user_id": str(found_user["_id"]) }
-
-@app.post("/register")
-def register(user: LoginModel):
-    if users_collection.find_one({"name": user.name, "etablissement": user.etablissement}):
-        raise HTTPException(status_code=400, detail="Utilisateur déjà existant")
-    
-    user_data = {
-        "name": user.name,
-        "etablissement": user.etablissement,
-        "hashed_password": hash_password(user.hashed_password)
-    }
-    inserted_id = users_collection.insert_one(user_data).inserted_id
-    return {"message": "Utilisateur enregistré", "user_id": str(inserted_id)}
+    if not user:
+        raise HTTPException(status_code=401, detail="Identifiants incorrects")
+    request.session["user"] = username
+    return {"status": "success"}
+
+@app.get("/api/logout")
+async def logout(request: Request):
+    request.session.pop("user", None)
+    return RedirectResponse(url="/login")
+
+@app.get("/api/check_session")
+async def check_session(request: Request):
+    return {"authenticated": bool(request.session.get("user"))}

requirements.txt

@@ -1,4 +1,3 @@
 fastapi
 pymongo
-motor
-uvicorn
+uvicorn