71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
"""
|
|
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()
|