tst

main.py

@@ -1,39 +1,50 @@
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+from fastapi import FastAPI, HTTPException, Depends
 from pymongo import MongoClient
-from bson import ObjectId
+from pydantic import BaseModel
+from bson.objectid import ObjectId
 import hashlib
+import os
 
-# Configuration MongoDB
-client = MongoClient("mongodb+srv://lebaykserver:mVef8e5F1GRtbA14@test.6bzxp.mongodb.net/")
-db = client["IA_SIGNATURE"]
-users_collection = db["USER"]
-
-# Créer une instance de FastAPI
 app = FastAPI()
 
-# Modèle pour valider les données reçues
-class User(BaseModel):
-    username: str
-    password: str
+# Connexion à MongoDB
+MONGO_URL = "mongodb://localhost:27017"
+client = MongoClient(MONGO_URL)
+db = client["mydatabase"]  # Remplace "mydatabase" par le nom de ta base
+users_collection = db["users"]
+
+# Modèle de connexion
+class LoginModel(BaseModel):
+    name: str
     etablissement: str
+    hashed_password: str
 
-# Fonction pour hasher le mot de passe
 def hash_password(password: str) -> str:
     return hashlib.sha256(password.encode()).hexdigest()
 
-@app.post("/login/")
-async def login_user(user: User):
-    # Recherche de l'utilisateur dans la base de données
-    existing_user = users_collection.find_one({"username": user.username})
-    
-    if not existing_user:
-        raise HTTPException(status_code=400, detail="Utilisateur non trouvé")
+@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
+    })
     
-    # Vérifier si le mot de passe est correct
-    hashed_password = hash_password(user.password)
+    if not found_user:
+        raise HTTPException(status_code=401, detail="Identifiants invalides")
     
-    if existing_user["password"] != hashed_password:
-        raise HTTPException(status_code=400, detail="Mot de passe incorrect")
+    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")
     
-    return {"message": "Connexion réussie"}
+    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)}