import logging import re import uuid from natsort import natsorted from sqlmodel import ( Field, # pyright: ignore[reportUnknownVariableType] SQLModel, ) logger = logging.getLogger(__name__) class TeilchenCreate(SQLModel): description: str | None name: str = Field(index=True) number: int = Field(default=1) text: str tags: str | None class Teilchen(TeilchenCreate, table=True): id: uuid.UUID = Field(default_factory=uuid.uuid7, primary_key=True) def make_teilchen_input(text: str) -> TeilchenCreate | None: if not text: logger.error("Empty text.") return None name_end = text.find(".") name = text[0:name_end] if not name: logger.error("Could not extract name.") return None description_start = text.find('"', name_end + 1) description_end = text.find('"', description_start + 1) description = text[description_start:description_end] if not description: logger.warning("Could not extract description.") tags = re.findall(r"#\w+", text.lower()) if not tags: logger.warning("No tags found in text.") return TeilchenCreate( name=name, description=description, tags=" ".join(natsorted(tags)), text=text, )