fuel_track/app/__init__.py

from flask import Flask
import click
from sqlalchemy import text
from .extensions import db, login_manager, socketio
from .models import User, CompanySettings, AppSetting
from .config import Config

THEMES = {name: {} for name in ("bootstrap", "flatly", "darkly", "pulse")}

def _themes(app):
    return {
        "bootstrap": {"label":"Bootstrap", "url":app.config["THEME_BOOTSTRAP_URL"], "mode":"light"},
        "flatly": {"label":"Flatly", "url":app.config["THEME_FLATLY_URL"], "mode":"light"},
        "darkly": {"label":"Darkly", "url":app.config["THEME_DARKLY_URL"], "mode":"dark"},
        "pulse": {"label":"Pulse", "url":app.config["THEME_PULSE_URL"], "mode":"light"},
    }


def create_app(test_config=None):
    app = Flask(__name__)
    app.config.from_object(Config)
    if test_config: app.config.update(test_config)
    db.init_app(app); login_manager.init_app(app); socketio.init_app(app)
    from .auth import auth; from .main import main
    from .api import api
    from .openapi import docs
    app.register_blueprint(auth); app.register_blueprint(main); app.register_blueprint(api); app.register_blueprint(docs)

    @login_manager.user_loader
    def load_user(user_id): return db.session.get(User, int(user_id))

    @app.context_processor
    def inject_theme():
        themes = _themes(app)
        record = db.session.get(AppSetting, "theme")
        theme_name = record.value if record and record.value in themes else "bootstrap"
        return {"app_theme": themes[theme_name], "app_theme_name": theme_name, "themes": themes,
                "asset_urls": {"bootstrap_js": app.config["BOOTSTRAP_JS_URL"], "choices_css": app.config["CHOICES_CSS_URL"],
                "choices_js": app.config["CHOICES_JS_URL"], "chart_js": app.config["CHART_JS_URL"], "socket_io_js": app.config["SOCKET_IO_JS_URL"]}}

    @app.get("/health")
    def health(): return {"status":"ok"}

    @app.cli.command("reset-admin-password")
    @click.option("--email", required=True, help="Adres e-mail konta administratora.")
    @click.option("--password", prompt=True, hide_input=True, confirmation_prompt=True,
                  help="Nowe hasło. Bez tej opcji polecenie poprosi o hasło bez wyświetlania go.")
    def reset_admin_password(email, password):
        """Resetuje hasło istniejącego administratora."""
        normalized_email = email.strip().lower()
        user = User.query.filter_by(email=normalized_email).first()
        if not user:
            raise click.ClickException("Nie znaleziono użytkownika o podanym adresie e-mail.")
        if user.role != "admin":
            raise click.ClickException("Wskazany użytkownik nie ma roli administratora.")
        if len(password) < 8:
            raise click.ClickException("Hasło musi mieć co najmniej 8 znaków.")
        user.set_password(password)
        user.active = True
        db.session.commit()
        click.echo(f"Zresetowano hasło administratora: {normalized_email}")

    with app.app_context():
        db.create_all()
        if db.engine.dialect.name == "sqlite":
            def add_column(table, column, ddl, index_sql=None):
                columns = {row[1] for row in db.session.execute(text(f"PRAGMA table_info({table})")).fetchall()}
                if column not in columns:
                    db.session.execute(text(f"ALTER TABLE {table} ADD COLUMN {ddl}"))
                    if index_sql:
                        db.session.execute(text(index_sql))
            add_column("fuel_entry", "station_company_id", "station_company_id INTEGER REFERENCES fuel_station_company(id)", "CREATE INDEX IF NOT EXISTS ix_fuel_entry_station_company_id ON fuel_entry (station_company_id)")
            add_column("fuel_entry", "fuel_card_id", "fuel_card_id INTEGER REFERENCES fuel_card(id)", "CREATE INDEX IF NOT EXISTS ix_fuel_entry_fuel_card_id ON fuel_entry (fuel_card_id)")
            add_column("vehicle", "fuel_card_id", "fuel_card_id INTEGER REFERENCES fuel_card(id)", "CREATE INDEX IF NOT EXISTS ix_vehicle_fuel_card_id ON vehicle (fuel_card_id)")
            add_column("user", "fuel_card_id", "fuel_card_id INTEGER REFERENCES fuel_card(id)", "CREATE INDEX IF NOT EXISTS ix_user_fuel_card_id ON user (fuel_card_id)")
            add_column("user", "company_id", "company_id INTEGER REFERENCES company_settings(id)", "CREATE INDEX IF NOT EXISTS ix_user_company_id ON user (company_id)")
            add_column("vehicle", "company_id", "company_id INTEGER REFERENCES company_settings(id)", "CREATE INDEX IF NOT EXISTS ix_vehicle_company_id ON vehicle (company_id)")
            add_column("fuel_card", "company_id", "company_id INTEGER REFERENCES company_settings(id)", "CREATE INDEX IF NOT EXISTS ix_fuel_card_company_id ON fuel_card (company_id)")
            add_column("company_settings", "active", "active BOOLEAN NOT NULL DEFAULT 1")
            add_column("fuel_station_company", "brand_name", "brand_name VARCHAR(120) NOT NULL DEFAULT ''", "CREATE INDEX IF NOT EXISTS ix_fuel_station_company_brand_name ON fuel_station_company (brand_name)")
            db.session.commit()
        if not CompanySettings.query.first(): db.session.add(CompanySettings())
        if not db.session.get(AppSetting, "theme"): db.session.add(AppSetting(key="theme", value="bootstrap"))
        if not db.session.get(AppSetting, "invoice_split_enabled"): db.session.add(AppSetting(key="invoice_split_enabled", value="1"))
        if not User.query.first():
            u=User(email="admin@example.com", name="Administrator", role="admin"); u.set_password("admin123!"); db.session.add(u)
        db.session.commit()
        from .models import FuelStationCompany
        from .integrations.ure import infer_brand
        for station in FuelStationCompany.query.filter(db.or_(FuelStationCompany.brand_name.is_(None), FuelStationCompany.brand_name == "")).all():
            station.brand_name = infer_brand(station.company_name)
        db.session.commit()
        default_company=CompanySettings.query.order_by(CompanySettings.id).first()
        if default_company:
            User.query.filter(User.company_id.is_(None)).update({User.company_id:default_company.id})
            db.session.execute(text("UPDATE vehicle SET company_id = (SELECT company_id FROM user WHERE user.id = vehicle.owner_id) WHERE company_id IS NULL"))
            db.session.execute(text("UPDATE fuel_card SET company_id = :cid WHERE company_id IS NULL"), {"cid":default_company.id})
            db.session.commit()
    return app