main_basic_auth_etape2.py

from fastapi import FastAPI, Depends, HTTPException, status
from fastapi.security import HTTPBasic, HTTPBasicCredentials
from sqlalchemy.orm import Session
from contextlib import asynccontextmanager
from passlib.context import CryptContext
import logging
from db import get_db
from init_db2 import init_db
from models import User

"""
Mettre en place la vérification des données de l'utilisateur avec
un mot de passe hashé.
"""
@asynccontextmanager
async def lifespan(app: FastAPI):
    init_db()
    print("DB initialisée")
    yield
    print("App arrêtée")

app = FastAPI(lifespan=lifespan)
security = HTTPBasic()
pwd_context = CryptContext(schemes=["argon2"], deprecated="auto")

def basic_auth(db: Session = Depends(get_db), credentials: HTTPBasicCredentials = Depends(security)):
    """
    Mettre en place ici l'accès à la base de données
    :param credentials: contient le username et le password
    :return: le user provenant de la base de données
    """
    return {"username":credentials.username, "password":credentials.password}

@app.get("/users")
def users(db: Session = Depends(get_db)):
    return db.query(User).all()

@app.get("/protected/users")
def protected(db: Session = Depends(get_db), _user=Depends(basic_auth)):
    """
    Fournit la liste des utilisateurs mais le user/mot de passe de l'utilisateur ne sont pas vérifiées
    :param db: accès base de données
    :param _user: le user est fourni par la fonction basic_auth_noop
    :return:
    """
    return db.query(User).all()

@app.get('/protected/users/{user_id}')
def get_user(user_id, db: Session = Depends(get_db), _user=Depends(basic_auth)):

    # Validation : ID positif
    if user_id <= 0:
        raise HTTPException(status_code=400, detail="Invalid user ID")

    # Validation : utilisateur existe
    user_db = db.get(User, user_id)
    if not user_db:
        raise HTTPException(status_code=404, detail="User not found")

    # Validation : permissions
    if _user.id != user_db.id:
        raise HTTPException(status_code=403, detail="Access Denied")

    return user_db