backupmanager_ynh/sources/init_db.py

#!/usr/bin/env python3
"""Initialise (ou migre) la base de données SQLite. Appelé par les scripts install et upgrade."""
import os
import sys

if len(sys.argv) > 1:
    os.environ["BACKUPMANAGER_CONFIG"] = sys.argv[1]

from app import app, db
from sqlalchemy import text

with app.app_context():
    db.create_all()

    # Migrations manuelles pour les colonnes ajoutées après la création initiale
    with db.engine.connect() as conn:
        existing = [row[1] for row in conn.execute(text("PRAGMA table_info(jobs)"))]
        if "destination_id" not in existing:
            conn.execute(text(
                "ALTER TABLE jobs ADD COLUMN destination_id INTEGER REFERENCES destinations(id)"
            ))
            conn.commit()
            print("Migration : colonne destination_id ajoutée à jobs.")
        if "remote_instance_id" not in existing:
            conn.execute(text(
                "ALTER TABLE jobs ADD COLUMN remote_instance_id INTEGER REFERENCES remote_instances(id)"
            ))
            conn.commit()
            print("Migration : colonne remote_instance_id ajoutée à jobs.")

    print("Base de données initialisée.")