frontpage construction

src/wce_crm/backend/backend.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
 
+import dataclasses as dc
 from typing import Any, TypedDict, cast
 
 import polars as pl
+import sqlalchemy as sql
 
 from wce_crm import db
 
@@ -176,3 +178,32 @@ def get_initial_recording(
         raise KeyError(f"Database ID {id_} not found")
 
     return row._asdict()
+
+
+@dc.dataclass(slots=True)
+class FrontpageCompany:
+    erfassung_id: int
+    ma_id: int
+    name: str
+
+
+def get_company_list() -> list[FrontpageCompany]:
+    stmt = sql.select(
+        db.grunderfassung_unternehmen.c.erfassung_id,
+        db.grunderfassung_unternehmen.c.Partnersuche__un_suche,
+    )
+
+    with db.ENGINE.connect() as conn:
+        res = conn.execute(stmt)
+
+    front_page_companies: list[FrontpageCompany] = []
+
+    for entry in res:
+        erfassung_id = entry[0]
+        ma_id = entry[1]
+        comp_info = comp_search_get_info(ma_id)
+        name = comp_info["ma_unternehmensname"]
+
+        front_page_companies.append(FrontpageCompany(erfassung_id, ma_id, name))
+
+    return front_page_companies