BoursePython/app/__init__.py

import os
import time
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_wtf import CSRFProtect
from sqlalchemy.exc import OperationalError
from dotenv import load_dotenv

load_dotenv()

db = SQLAlchemy()
csrf = CSRFProtect()


def create_app():
    app = Flask(__name__)

    # --- Configuration depuis l'environnement ---
    app.config["SECRET_KEY"] = os.environ.get(
        "SECRET_KEY", "change-me-in-production"
    )
    app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False

    db_user = os.environ.get("MYSQL_USER", "jfgiraud")
    db_pass = os.environ.get("MYSQL_PASSWORD", "")
    db_host = os.environ.get("MYSQL_HOST", "db")
    db_port = os.environ.get("MYSQL_PORT", "3306")
    db_name = os.environ.get("MYSQL_DATABASE", "bolsa")
    # DATABASE_URL prend la priorité si défini
    app.config["SQLALCHEMY_DATABASE_URI"] = os.environ.get(
        "DATABASE_URL",
        f"mysql+pymysql://{db_user}:{db_pass}@{db_host}:{db_port}/{db_name}",
    )

    db.init_app(app)
    csrf.init_app(app)

    # Enregistrement des modèles ORM (nécessaire avant db.create_all)
    from app import models  # noqa: F401

    # Tentative de connexion au démarrage du conteneur (MySQL peut mettre du temps à être prêt)
    with app.app_context():
        max_retries = 15
        delay = 2
        for i in range(max_retries):
            try:
                db.create_all()
                print("Base de données connectée et tables initialisées avec succès.")
                break
            except OperationalError:
                print(f"Tentative {i+1}/{max_retries} : MySQL n'est pas encore prêt. "
                      f"Nouvelle tentative dans {delay}s...")
                time.sleep(delay)
        else:
            raise Exception("Impossible d'initialiser la base de données après plusieurs tentatives.")

    # Enregistrement des routes (Blueprints)
    from app.routes import (
        actions_bp,
        analyse_bp,
        auth_bp,
        comptes_bp,
        historique_bp,
        menu_bp,
        ordres_bp,
        societes_bp,
        utilisateurs_bp,
    )

    app.register_blueprint(auth_bp)
    app.register_blueprint(menu_bp)
    app.register_blueprint(utilisateurs_bp)
    app.register_blueprint(actions_bp)
    app.register_blueprint(societes_bp)
    app.register_blueprint(historique_bp)
    app.register_blueprint(analyse_bp)
    app.register_blueprint(ordres_bp)
    app.register_blueprint(comptes_bp)

    return app


# Expose 'app' pour que run.py puisse l'importer directement
app = create_app()