ipwatch

backend/app/migrations/__init__.py

@@ -0,0 +1,3 @@
+"""
+Module de migrations pour la base de données IPWatch
+"""

backend/app/migrations/add_architecture_node_table.py

@@ -0,0 +1,57 @@
+"""
+Script de migration pour ajouter la table architecture_node
+Exécuter avec: python -m backend.app.migrations.add_architecture_node_table
+"""
+from sqlalchemy import text, create_engine
+import sys
+import os
+
+
+def migrate():
+    try:
+        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:
+            result = conn.execute(text("SELECT name FROM sqlite_master WHERE type='table' AND name='architecture_node'"))
+            if result.fetchone():
+                print("✓ Table 'architecture_node' existe déjà")
+                return
+
+            print("→ Création de la table 'architecture_node'...")
+            conn.execute(text("""
+                CREATE TABLE architecture_node (
+                    id TEXT PRIMARY KEY,
+                    type TEXT NOT NULL,
+                    x INTEGER NOT NULL,
+                    y INTEGER NOT NULL,
+                    width INTEGER NOT NULL,
+                    height INTEGER NOT NULL,
+                    rotation INTEGER NOT NULL,
+                    payload TEXT NOT NULL,
+                    created_at DATETIME NOT NULL
+                )
+            """))
+            conn.execute(text("CREATE INDEX IF NOT EXISTS idx_arch_node_created_at ON architecture_node(created_at)"))
+            conn.commit()
+            print("✓ Table 'architecture_node' créée")
+
+    except Exception as e:
+        print(f"✗ Erreur migration architecture_node: {str(e)}", file=sys.stderr)
+        sys.exit(1)
+
+
+def rollback():
+    try:
+        print("⚠ Rollback non implémenté pour SQLite")
+    except Exception as e:
+        print(f"✗ Erreur 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()

backend/app/migrations/add_hardware_bench_field.py

@@ -0,0 +1,62 @@
+"""
+Script de migration pour ajouter le champ 'hardware_bench' à la table IP
+Exécuter avec: python -m backend.app.migrations.add_hardware_bench_field
+"""
+from sqlalchemy import text, create_engine
+import sys
+import os
+
+
+def migrate():
+    """Ajoute la colonne 'hardware_bench' et son index à la table IP"""
+    try:
+        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:
+            result = conn.execute(text("PRAGMA table_info(ip)"))
+            columns = [row[1] for row in result]
+
+            if 'hardware_bench' in columns:
+                print("✓ La colonne 'hardware_bench' existe déjà dans la table IP")
+                return
+
+            print("→ Ajout de la colonne 'hardware_bench' à la table IP...")
+            conn.execute(text("ALTER TABLE ip ADD COLUMN hardware_bench BOOLEAN DEFAULT 0"))
+
+            print("→ Création de l'index sur 'hardware_bench'...")
+            conn.execute(text("CREATE INDEX IF NOT EXISTS idx_ip_hardware_bench ON ip(hardware_bench)"))
+
+            conn.commit()
+            print("✓ Migration terminée avec succès!")
+            print("  - Colonne 'hardware_bench' ajoutée")
+            print("  - Index 'idx_ip_hardware_bench' 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 'hardware_bench' (rollback de la migration)"""
+    try:
+        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:
+            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()

backend/app/migrations/add_icon_fields.py

@@ -0,0 +1,54 @@
+"""
+Script de migration pour ajouter les champs 'icon_filename' et 'icon_url' à la table IP
+Exécuter avec: python -m backend.app.migrations.add_icon_fields
+"""
+from sqlalchemy import text, create_engine
+import sys
+import os
+
+
+def migrate():
+    try:
+        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:
+            result = conn.execute(text("PRAGMA table_info(ip)"))
+            columns = [row[1] for row in result]
+
+            if 'icon_filename' not in columns:
+                print("→ Ajout de la colonne 'icon_filename'...")
+                conn.execute(text("ALTER TABLE ip ADD COLUMN icon_filename TEXT"))
+                conn.execute(text("CREATE INDEX IF NOT EXISTS idx_ip_icon_filename ON ip(icon_filename)"))
+            else:
+                print("✓ Colonne 'icon_filename' déjà présente")
+
+            if 'icon_url' not in columns:
+                print("→ Ajout de la colonne 'icon_url'...")
+                conn.execute(text("ALTER TABLE ip ADD COLUMN icon_url TEXT"))
+                conn.execute(text("CREATE INDEX IF NOT EXISTS idx_ip_icon_url ON ip(icon_url)"))
+            else:
+                print("✓ Colonne 'icon_url' déjà présente")
+
+            conn.commit()
+            print("✓ Migration terminée avec succès")
+
+    except Exception as e:
+        print(f"✗ Erreur lors de la migration: {str(e)}", file=sys.stderr)
+        sys.exit(1)
+
+
+def rollback():
+    try:
+        print("⚠ Rollback non implémenté pour SQLite")
+    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()

backend/app/migrations/add_network_device_field.py

@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+"""
+Migration: Ajouter le champ network_device à la table ip
+"""
+import os
+import sys
+from sqlalchemy import text, create_engine
+
+def main():
+    """Ajoute le champ network_device à la table ip"""
+    # Récupérer le chemin de la base de données
+    db_path = os.getenv('DB_PATH', './data/db.sqlite')
+    db_url = f"sqlite:///{db_path}"
+
+    # Créer l'engine directement
+    engine = create_engine(db_url, echo=False)
+
+    print(f"📦 Migration: Ajout du champ network_device")
+    print(f"🗄️  Base de données: {db_path}")
+
+    try:
+        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 'network_device' in columns:
+                print("⚠️  La colonne 'network_device' existe déjà. Migration ignorée.")
+                return
+
+            # Ajouter la colonne network_device
+            print("➕ Ajout de la colonne 'network_device'...")
+            conn.execute(text("""
+                ALTER TABLE ip
+                ADD COLUMN network_device BOOLEAN DEFAULT 0
+            """))
+
+            # Créer un index sur la colonne
+            print("🔍 Création de l'index sur 'network_device'...")
+            conn.execute(text("""
+                CREATE INDEX IF NOT EXISTS idx_ip_network_device ON ip(network_device)
+            """))
+
+            conn.commit()
+            print("✅ Migration réussie!")
+
+    except Exception as e:
+        print(f"❌ Erreur lors de la migration: {e}")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()

backend/app/migrations/add_scan_log_table.py

@@ -0,0 +1,54 @@
+"""
+Script de migration pour ajouter la table scan_log
+Exécuter avec: python -m backend.app.migrations.add_scan_log_table
+"""
+from sqlalchemy import text, create_engine
+import sys
+import os
+
+
+def migrate():
+    try:
+        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:
+            result = conn.execute(text("SELECT name FROM sqlite_master WHERE type='table' AND name='scan_log'"))
+            if result.fetchone():
+                print("✓ Table 'scan_log' existe déjà")
+                return
+
+            print("→ Création de la table 'scan_log'...")
+            conn.execute(text("""
+                CREATE TABLE scan_log (
+                    id INTEGER PRIMARY KEY AUTOINCREMENT,
+                    ip TEXT,
+                    status TEXT,
+                    message TEXT NOT NULL,
+                    created_at DATETIME NOT NULL
+                )
+            """))
+            conn.execute(text("CREATE INDEX IF NOT EXISTS idx_scan_log_created_at ON scan_log(created_at)"))
+            conn.execute(text("CREATE INDEX IF NOT EXISTS idx_scan_log_ip ON scan_log(ip)"))
+            conn.commit()
+            print("✓ Table 'scan_log' créée")
+
+    except Exception as e:
+        print(f"✗ Erreur migration scan_log: {str(e)}", file=sys.stderr)
+        sys.exit(1)
+
+
+def rollback():
+    try:
+        print("⚠ Rollback non implémenté pour SQLite")
+    except Exception as e:
+        print(f"✗ Erreur 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()

backend/app/migrations/add_tracked_field.py

@@ -0,0 +1,70 @@
+"""
+Script de migration pour ajouter le champ 'tracked' à la table IP
+Exécuter avec: python -m backend.app.migrations.add_tracked_field
+"""
+from sqlalchemy import text, create_engine
+import sys
+import os
+
+
+def migrate():
+    """Ajoute la colonne 'tracked' 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 'tracked' in columns:
+                print("✓ La colonne 'tracked' existe déjà dans la table IP")
+                return
+
+            # Ajouter la colonne tracked
+            print("→ Ajout de la colonne 'tracked' à la table IP...")
+            conn.execute(text("ALTER TABLE ip ADD COLUMN tracked BOOLEAN DEFAULT 0"))
+
+            # Créer l'index
+            print("→ Création de l'index sur 'tracked'...")
+            conn.execute(text("CREATE INDEX IF NOT EXISTS idx_ip_tracked ON ip(tracked)"))
+
+            conn.commit()
+            print("✓ Migration terminée avec succès!")
+            print("  - Colonne 'tracked' ajoutée")
+            print("  - Index 'idx_ip_tracked' 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 'tracked' (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()

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()

backend/app/migrations/create_architecture_db.py

@@ -0,0 +1,60 @@
+"""
+Script pour créer la base SQLite dédiée à l'architecture
+Exécuter avec: python -m backend.app.migrations.create_architecture_db
+"""
+from sqlalchemy import text, create_engine
+import sys
+import os
+from pathlib import Path
+
+
+def migrate():
+    try:
+        db_path = os.getenv("ARCH_DB_PATH", "./architecture/database/architecture.sqlite")
+        db_file = Path(db_path)
+        db_file.parent.mkdir(parents=True, exist_ok=True)
+        db_url = f"sqlite:///{db_file}"
+        engine = create_engine(db_url, echo=False)
+
+        with engine.connect() as conn:
+            result = conn.execute(text("SELECT name FROM sqlite_master WHERE type='table' AND name='architecture_node'"))
+            if result.fetchone():
+                print("✓ Table 'architecture_node' existe déjà")
+                return
+
+            print("→ Création de la table 'architecture_node'...")
+            conn.execute(text("""
+                CREATE TABLE architecture_node (
+                    id TEXT PRIMARY KEY,
+                    type TEXT NOT NULL,
+                    x INTEGER NOT NULL,
+                    y INTEGER NOT NULL,
+                    width INTEGER NOT NULL,
+                    height INTEGER NOT NULL,
+                    rotation INTEGER NOT NULL,
+                    payload TEXT NOT NULL,
+                    created_at DATETIME NOT NULL
+                )
+            """))
+            conn.execute(text("CREATE INDEX IF NOT EXISTS idx_arch_node_created_at ON architecture_node(created_at)"))
+            conn.commit()
+            print(f"✓ Base architecture créée: {db_file}")
+
+    except Exception as e:
+        print(f"✗ Erreur création base architecture: {str(e)}", file=sys.stderr)
+        sys.exit(1)
+
+
+def rollback():
+    try:
+        print("⚠ Rollback non implémenté pour SQLite")
+    except Exception as e:
+        print(f"✗ Erreur 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()