fooder-api/fooder/domain/meal.py

from typing import Optional

from sqlalchemy import ForeignKey, Integer, select
from sqlalchemy.exc import IntegrityError
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.orm import Mapped, joinedload, mapped_column, relationship

from .base import Base, CommonMixin
from .entry import Entry
from .preset import Preset


class Meal(Base, CommonMixin):
    """Meal."""

    name: Mapped[str]
    order: Mapped[int]
    diary_id: Mapped[int] = mapped_column(Integer, ForeignKey("diary.id"))
    entries: Mapped[list[Entry]] = relationship(
        lazy="selectin", order_by=Entry.last_changed
    )

    @property
    def calories(self) -> float:
        """calories.

        :rtype: float
        """
        return sum(entry.calories for entry in self.entries)

    @property
    def protein(self) -> float:
        """protein.

        :rtype: float
        """
        return sum(entry.protein for entry in self.entries)

    @property
    def carb(self) -> float:
        """carb.

        :rtype: float
        """
        return sum(entry.carb for entry in self.entries)

    @property
    def fat(self) -> float:
        """fat.

        :rtype: float
        """
        return sum(entry.fat for entry in self.entries)

    @property
    def fiber(self) -> float:
        """fiber.

        :rtype: float
        """
        return sum(entry.fiber for entry in self.entries)

    @classmethod
    async def create(
        cls,
        session: AsyncSession,
        diary_id: int,
        name: Optional[str] = None,
    ) -> "Meal":
        # check if order already exists in diary
        query = (
            select(cls.order).where(cls.diary_id == diary_id).order_by(cls.order.desc())
        )
        existing_meal = await session.scalar(query)
        order = existing_meal + 1 if existing_meal else 1

        if name is None:
            name = f"Meal {order}"
        meal = Meal(diary_id=diary_id, name=name, order=order)
        session.add(meal)

        try:
            await session.flush()
        except IntegrityError:
            raise AssertionError("diary does not exist")

        db_meal = await cls._get_by_id(session, meal.id)
        if not db_meal:
            raise RuntimeError()
        return db_meal

    @classmethod
    async def create_from_preset(
        cls,
        session: AsyncSession,
        diary_id: int,
        name: Optional[str],
        preset: Preset,
    ) -> "Meal":
        # check if order already exists in diary
        query = (
            select(cls.order).where(cls.diary_id == diary_id).order_by(cls.order.desc())
        )
        existing_meal = await session.scalar(query)
        order = existing_meal + 1 if existing_meal else 1

        if name is None:
            name = preset.name or f"Meal {order}"

        meal = Meal(diary_id=diary_id, name=name, order=order)
        session.add(meal)

        try:
            await session.flush()
        except IntegrityError:
            raise AssertionError("diary does not exist")

        for entry in preset.entries:
            await Entry.create(session, meal.id, entry.product_id, entry.grams)

        db_meal = await cls._get_by_id(session, meal.id)
        if not db_meal:
            raise RuntimeError()
        return db_meal

    @classmethod
    async def _get_by_id(cls, session: AsyncSession, id: int) -> "Optional[Meal]":
        """get_by_id."""
        query = select(cls).where(cls.id == id).options(joinedload(cls.entries))
        return await session.scalar(query.order_by(cls.id))

    @classmethod
    async def get_by_id(
        cls, session: AsyncSession, user_id: int, id: int
    ) -> "Optional[Meal]":
        """get_by_id."""
        from .diary import Diary

        query = (
            select(cls)
            .where(cls.id == id)
            .join(Diary)
            .where(Diary.user_id == user_id)
            .options(joinedload(cls.entries))
        )
        return await session.scalar(query.order_by(cls.id))

    async def delete(self, session: AsyncSession) -> None:
        """delete."""
        for entry in self.entries:
            await session.delete(entry)
        await session.delete(self)
        await session.flush()