first commit

2026-07-13 13:19:26 +02:00
commit bd562b3da6
50 changed files with 2221 additions and 0 deletions
@@ -0,0 +1,99 @@
+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