docekr

app/database_init.py

@@ -0,0 +1,75 @@
+from __future__ import annotations
+
+from sqlalchemy import inspect
+
+from .extensions import db
+from .models import AppSetting, CompanySettings, User
+
+
+DEFAULT_SETTINGS = {
+    "theme": "bootstrap",
+    "invoice_split_enabled": "1",
+}
+
+
+def create_schema() -> None:
+    """Tworzy wszystkie tabele, ograniczenia i indeksy zdefiniowane w modelach."""
+    db.metadata.create_all(bind=db.engine)
+
+
+def seed_initial_data() -> None:
+    """Dodaje minimalne dane startowe do świeżej bazy."""
+    if not CompanySettings.query.first():
+        db.session.add(CompanySettings())
+
+    for key, value in DEFAULT_SETTINGS.items():
+        if db.session.get(AppSetting, key) is None:
+            db.session.add(AppSetting(key=key, value=value))
+
+    if not User.query.first():
+        admin = User(
+            email="admin@example.com",
+            name="Administrator",
+            role="admin",
+        )
+        admin.set_password("admin123!")
+        db.session.add(admin)
+
+    db.session.commit()
+
+
+def initialize_database() -> None:
+    """Zakłada świeżą bazę: schemat, indeksy, constraints i dane startowe."""
+    create_schema()
+    seed_initial_data()
+
+
+def schema_summary() -> dict[str, list[str]]:
+    """Zwraca utworzone tabele i indeksy; przydatne do logów i testów."""
+    inspector = inspect(db.engine)
+    tables = sorted(inspector.get_table_names())
+    indexes = sorted(
+        f"{table}.{index['name']}"
+        for table in tables
+        for index in inspector.get_indexes(table)
+        if index.get("name")
+    )
+    return {"tables": tables, "indexes": indexes}
+
+
+def main() -> None:
+    from . import create_app
+
+    app = create_app()
+    with app.app_context():
+        initialize_database()
+        summary = schema_summary()
+
+    print(
+        "Baza danych została zainicjalizowana "
+        f"({len(summary['tables'])} tabel, {len(summary['indexes'])} indeksów)."
+    )
+
+
+if __name__ == "__main__":
+    main()