ipwatch

backend/app/migrations/add_vm_field.py

@@ -0,0 +1,70 @@
+"""
+Script de migration pour ajouter le champ 'vm' à la table IP
+Exécuter avec: python -m backend.app.migrations.add_vm_field
+"""
+from sqlalchemy import text, create_engine
+import sys
+import os
+
+
+def migrate():
+    """Ajoute la colonne 'vm' et son index à la table IP"""
+    try:
+        # Charger le chemin de la base de données depuis config.yaml ou utiliser le défaut
+        db_path = os.getenv('DB_PATH', './data/db.sqlite')
+        db_url = f"sqlite:///{db_path}"
+
+        # Créer l'engine
+        engine = create_engine(db_url, echo=False)
+
+        with engine.connect() as conn:
+            # Vérifier si la colonne existe déjà
+            result = conn.execute(text("PRAGMA table_info(ip)"))
+            columns = [row[1] for row in result]
+
+            if 'vm' in columns:
+                print("✓ La colonne 'vm' existe déjà dans la table IP")
+                return
+
+            # Ajouter la colonne vm
+            print("→ Ajout de la colonne 'vm' à la table IP...")
+            conn.execute(text("ALTER TABLE ip ADD COLUMN vm BOOLEAN DEFAULT 0"))
+
+            # Créer l'index
+            print("→ Création de l'index sur 'vm'...")
+            conn.execute(text("CREATE INDEX IF NOT EXISTS idx_ip_vm ON ip(vm)"))
+
+            conn.commit()
+            print("✓ Migration terminée avec succès!")
+            print("  - Colonne 'vm' ajoutée")
+            print("  - Index 'idx_ip_vm' créé")
+
+    except Exception as e:
+        print(f"✗ Erreur lors de la migration: {str(e)}", file=sys.stderr)
+        sys.exit(1)
+
+
+def rollback():
+    """Supprime la colonne 'vm' (rollback de la migration)"""
+    try:
+        # Charger le chemin de la base de données depuis config.yaml ou utiliser le défaut
+        db_path = os.getenv('DB_PATH', './data/db.sqlite')
+        db_url = f"sqlite:///{db_path}"
+        engine = create_engine(db_url, echo=False)
+
+        with engine.connect() as conn:
+            # SQLite ne supporte pas DROP COLUMN directement
+            # Il faut recréer la table sans la colonne
+            print("⚠ Rollback non implémenté pour SQLite")
+            print("  Pour annuler, restaurez une sauvegarde de la base de données")
+
+    except Exception as e:
+        print(f"✗ Erreur lors du rollback: {str(e)}", file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1 and sys.argv[1] == "rollback":
+        rollback()
+    else:
+        migrate()