teilchensammler-cli/src/teilchensammler_cli/models.py

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,
    )