maj

DragonBank/backend/database.py

@@ -0,0 +1,142 @@
+"""
+DragonBank - Utilitaires Base de Donnees
+=========================================
+Gere la connexion a PostgreSQL et la serialisation
+des resultats de requetes en types JSON-compatibles.
+
+Version : 3.0
+"""
+
+import uuid
+import decimal
+import logging
+from datetime import datetime
+
+import psycopg2
+import psycopg2.extras
+
+from config import URL_BASE_DE_DONNEES
+
+journaliseur = logging.getLogger('dragonbank.database')
+
+
+# ============================================================
+# CONNEXION
+# ============================================================
+
+def creer_connexion():
+    """
+    Cree et retourne une connexion active a la base de donnees PostgreSQL.
+
+    Utilise RealDictCursor pour que les resultats soient des
+    dictionnaires (acces par nom de colonne) plutot que des tuples.
+
+    L'autocommit est desactive : chaque route doit appeler
+    commit() ou rollback() explicitement.
+
+    Returns:
+        psycopg2.connection: Connexion avec autocommit desactive.
+
+    Raises:
+        psycopg2.OperationalError: Si la connexion echoue.
+    """
+    connexion = psycopg2.connect(
+        URL_BASE_DE_DONNEES,
+        cursor_factory=psycopg2.extras.RealDictCursor
+    )
+    connexion.autocommit = False
+    return connexion
+
+
+# ============================================================
+# SERIALISATION JSON
+# ============================================================
+
+def serialiser_valeur(valeur):
+    """
+    Convertit une valeur PostgreSQL en type serialisable JSON.
+
+    Certains types retournes par psycopg2 (Decimal, datetime, UUID)
+    ne sont pas nativement serialisables par json.dumps().
+
+    Args:
+        valeur: La valeur a convertir.
+
+    Returns:
+        float  : si la valeur est un Decimal.
+        str    : si la valeur est un datetime ou un UUID.
+        valeur : inchangee pour tous les autres types.
+    """
+    if isinstance(valeur, decimal.Decimal):
+        return float(valeur)
+    if isinstance(valeur, datetime):
+        return valeur.isoformat()
+    if isinstance(valeur, uuid.UUID):
+        return str(valeur)
+    return valeur
+
+
+def serialiser_ligne(ligne):
+    """
+    Serialise une ligne de resultat PostgreSQL en dictionnaire Python.
+
+    Args:
+        ligne: Une RealDictRow psycopg2, ou None.
+
+    Returns:
+        dict : Toutes les colonnes avec leurs valeurs converties.
+        None : Si la ligne est None.
+    """
+    if ligne is None:
+        return None
+
+    resultat = {}
+    for cle, valeur in dict(ligne).items():
+        resultat[cle] = serialiser_valeur(valeur)
+    return resultat
+
+
+def serialiser_lignes(lignes):
+    """
+    Serialise une liste de lignes de resultat PostgreSQL.
+
+    Args:
+        lignes: Liste de RealDictRow psycopg2.
+
+    Returns:
+        list[dict]: Liste de dictionnaires serialises.
+    """
+    resultat = []
+    for ligne in lignes:
+        resultat.append(serialiser_ligne(ligne))
+    return resultat
+
+
+# ============================================================
+# AUDIT
+# ============================================================
+
+def enregistrer_audit(curseur, id_utilisateur, action, details, adresse_ip):
+    """
+    Insere une entree dans la table des journaux d'audit.
+
+    Trace toutes les actions sensibles (connexion, virement, ouverture
+    de compte) pour la conformite reglementaire (RGPD, DSP2).
+
+    Args:
+        curseur        : Curseur de base de donnees actif.
+        id_utilisateur : UUID de l'utilisateur.
+        action (str)   : Code de l'action (ex: 'LOGIN', 'TRANSFER_INTERNAL').
+        details (dict) : Informations complementaires en JSONB.
+        adresse_ip (str): IP du client.
+
+    Note:
+        Ne commit pas. La validation est a la charge de l'appelant.
+    """
+    curseur.execute(
+        """
+        INSERT INTO audit_logs (user_id, action, details, ip_address)
+        VALUES (%s, %s, %s::jsonb, %s)
+        """,
+        (id_utilisateur, action, psycopg2.extras.Json(details), adresse_ip)
+    )