Update baserow_storage.py

baserow_storage.py

@@ -1,4 +1,4 @@
-# baserow_storage.py
+# baserow_storage.py (Version Modifiée)
 
 import os
 import requests
@@ -19,6 +19,8 @@ API_TOKEN = os.environ.get("BASEROW_API_TOKEN")
 # Les IDs de table seront récupérés depuis les variables d'environnement
 PRIMARY_USERS_TABLE_ID = os.environ.get("PRIMARY_USERS_TABLE_ID") 
 END_USERS_TABLE_ID = os.environ.get("END_USERS_TABLE_ID") 
+# NOUVEL ID DE TABLE POUR LES DÉPLOIEMENTS (Projets)
+DEPLOYMENTS_TABLE_ID = os.environ.get("DEPLOYMENTS_TABLE_ID")
 
 # Headers pour l'authentification
 HEADERS = {
@@ -27,8 +29,12 @@ HEADERS = {
 }
 
 # ----------------------------------------------------------------------
-# --- Noms de Colonnes pour la Table des Utilisateurs Principaux (Primary Users) ---
+# --- Noms de Colonnes pour les Déploiements (à adapter à votre structure Baserow) ---
 # ----------------------------------------------------------------------
+FIELD_DEPLOY_ID = 'Deployment ID' # Nom de la colonne pour l'ID de déploiement (le dossier HF)
+FIELD_USER_REF = 'User Link'     # Nom de la colonne pour la référence à l'utilisateur principal
+FIELD_HF_REPO_PATH = 'HF Repo Path' # Chemin complet vers le dossier dans le repo
+
 FIELD_ID = 'ID' # Correspond à 'user_id' dans le code
 FIELD_EMAIL = 'Email' # Correspond à 'email'
 FIELD_USERNAME = 'Nom d\'utilisateur' # Correspond à 'username'
@@ -521,4 +527,75 @@ def _make_baserow_request(method: str, url: str, params: Optional[Dict] = None,
     except requests.exceptions.RequestException as e:
         logger.error(f"BASEROW CONNECTION ERROR: {e}")
         # Retourne un objet Response factice pour l'utiliser dans la fonction appelante
-        return requests.Response()
+        return requests.Response()
+
+# --- NOUVELLES FONCTIONS POUR LES DÉPLOIEMENTS (Projets) ---
+
+
+
+def update_user_deployment_data(
+    user_baserow_row_id: int, 
+    deploy_id: str, 
+    user_link: str, # Utilisera l'ID interne de l'utilisateur
+    hf_repo_path: str
+) -> tuple[bool, str]:
+    """
+    Met à jour la ligne de l'utilisateur principal (PRIMARY_USERS_TABLE_ID) 
+    avec les données de déploiement du projet (ID, Repo, etc.).
+    """
+    if not PRIMARY_USERS_TABLE_ID:
+        return False, "Erreur de configuration: ID de la table utilisateur principal manquant."
+
+    url = f"{DATA_BASE_URL}table/{PRIMARY_USERS_TABLE_ID}/{user_baserow_row_id}/"
+    
+    # Les données à mettre à jour
+    data = {
+        FIELD_DEPLOY_ID: deploy_id,
+        FIELD_USER_REF: user_link,
+        FIELD_HF_REPO_PATH: hf_repo_path
+    }
+
+    # Utiliser PATCH pour mettre à jour seulement les champs spécifiés
+    response = _baserow_request(url, method="PATCH", data=data)
+    
+    if response.status_code == 200:
+        logger.info(f"Déploiement ID {deploy_id} enregistré sur l'utilisateur Row ID {user_baserow_row_id}.")
+        return True, "Données de déploiement enregistrées avec succès sur le compte utilisateur."
+    else:
+        logger.error(f"Échec de la mise à jour du déploiement Baserow pour l'utilisateur {user_baserow_row_id}. Status: {response.status_code}, Body: {response.text}")
+        return False, f"Échec de la mise à jour Baserow: {response.status_code}"
+
+
+# ----------------------------------------------------------------------
+# --- MODIFICATION DE FONCTION : Récupérer le déploiement par ID ---
+# La recherche doit maintenant se faire dans la table utilisateur.
+# ----------------------------------------------------------------------
+def get_deployment_by_id(deploy_id: str) -> Optional[Dict]:
+    """
+    Récupère les détails du déploiement (la ligne de l'utilisateur) par son ID unique (le nom du dossier HF).
+    """
+    if not PRIMARY_USERS_TABLE_ID:
+        return None
+        
+    # Utiliser un filtre Baserow pour chercher par l'ID de déploiement dans la table utilisateur
+    url = f"{DATA_BASE_URL}table/{PRIMARY_USERS_TABLE_ID}/"
+    
+    # Construction du filtre Baserow: filter__Deployment ID__equal={deploy_id}
+    filter_param = f"filter__{FIELD_DEPLOY_ID}__equal"
+    params = {
+        filter_param: deploy_id
+    }
+    
+    response = _baserow_request(url, method="GET", params=params)
+    
+    if response.status_code == 200:
+        results = response.json().get('results', [])
+        if results:
+            # Retourne les données de l'utilisateur contenant les données de déploiement
+            # Notez que cela renvoie la ligne utilisateur complète.
+            return results[0] 
+        else:
+            return None
+    else:
+        logger.error(f"Échec de la récupération du déploiement Baserow. Status: {response.status_code}, Body: {response.text}")
+        return None