Files
serv_benchmark/backend/apply_migration_011.py
Gilles Soulier c67befc549 addon
2026-01-05 16:08:01 +01:00

47 lines
1.2 KiB
Python
Executable File

#!/usr/bin/env python3
"""
Apply migration 011: Add fabricant and produit fields
"""
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent))
from app.db.session import get_peripherals_db
def apply_migration():
"""Apply migration 011"""
db = next(get_peripherals_db())
try:
print("\ud83d\udd27 Applying migration 011: Add fabricant and produit")
print("=" * 60)
migration_file = Path(__file__).parent / "migrations" / "011_add_fabricant_produit.sql"
with open(migration_file, "r") as f:
sql_commands = f.read()
for command in sql_commands.split(';'):
command = command.strip()
if command and not command.startswith('--'):
db.execute(command)
db.commit()
print("\n\u2705 Migration 011 applied successfully!")
print("=" * 60)
print("Added columns:")
print(" - fabricant (TEXT)")
print(" - produit (TEXT)")
except Exception as e:
print(f"\u274c Error applying migration: {e}")
db.rollback()
raise
finally:
db.close()
if __name__ == "__main__":
apply_migration()