diff --git a/src/teilchensammler_cli/models.py b/src/teilchensammler_cli/models.py index 1cc6633..484e915 100644 --- a/src/teilchensammler_cli/models.py +++ b/src/teilchensammler_cli/models.py @@ -4,14 +4,14 @@ import uuid from sqlmodel import ( Field, SQLModel, + Session, + select, + Sequence, ) logger = logging.getLogger(__name__) -TeilchenDatum = tuple[int, str, str, str, str] - - class TeilchenCreate(SQLModel): description: str | None name: str = Field(index=True) @@ -57,12 +57,12 @@ async def make_teilchen_input(text: str) -> TeilchenCreate | None: ) -async def load_initial_data() -> list[TeilchenDatum]: - return [ - (0, "Name0", "Description0", "9000", "#tag0 #tag00 #tag000"), - (1, "Name1", "Description1", "9001", "#tag1 #tag11 #tag111"), - (2, "Name2", "Description2", "9002", "#tag2 #tag22 #tag222"), - (3, "Name3", "Description3", "9003", "#tag3 #tag33 #tag333"), - (4, "Name4", "Description4", "9004", "#tag4 #tag44 #tag444"), - (5, "Name5", "Description5", "9005", "#tag5 #tag55 #tag555"), - ] +async def load_initial_data() -> Sequence[Teilchen]: + from .database import engine + + with Session(engine) as session: + statement = select( + Teilchen.id, Teilchen.name, Teilchen.description, Teilchen.number, Teilchen.tags + ) + all_teilchen = session.exec(statement).all() + return all_teilchen