90 lines
2.3 KiB
Python
90 lines
2.3 KiB
Python
from sqlalchemy.orm import relationship, Mapped, mapped_column, joinedload
|
|
from sqlalchemy import ForeignKey, Integer, Date
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from datetime import date
|
|
from typing import Optional
|
|
|
|
from .base import Base, CommonMixin
|
|
from .meal import Meal
|
|
|
|
|
|
class Diary(Base, CommonMixin):
|
|
"""Diary represents user diary for given day"""
|
|
|
|
meals: Mapped[list[Meal]] = relationship(lazy="selectin")
|
|
date: Mapped[date] = mapped_column(Date)
|
|
user_id: Mapped[int] = mapped_column(Integer, ForeignKey("user.id"))
|
|
|
|
@property
|
|
def calories(self) -> float:
|
|
"""calories.
|
|
|
|
:rtype: float
|
|
"""
|
|
return sum(meal.calories for meal in self.meals)
|
|
|
|
@property
|
|
def protein(self) -> float:
|
|
"""protein.
|
|
|
|
:rtype: float
|
|
"""
|
|
return sum(meal.protein for meal in self.meals)
|
|
|
|
@property
|
|
def carb(self) -> float:
|
|
"""carb.
|
|
|
|
:rtype: float
|
|
"""
|
|
return sum(meal.carb for meal in self.meals)
|
|
|
|
@property
|
|
def fat(self) -> float:
|
|
"""fat.
|
|
|
|
:rtype: float
|
|
"""
|
|
return sum(meal.fat for meal in self.meals)
|
|
|
|
@classmethod
|
|
async def get_diary(
|
|
cls, session: AsyncSession, user_id: int, date: date
|
|
) -> "Optional[Diary]":
|
|
"""get_diary."""
|
|
query = select(cls).where(cls.user_id == user_id).where(cls.date == date)
|
|
return await session.scalar(query)
|
|
|
|
@classmethod
|
|
async def create(cls, session: AsyncSession, user_id: int, date: date) -> "Diary":
|
|
diary = Diary(
|
|
date=date,
|
|
user_id=user_id,
|
|
)
|
|
session.add(diary)
|
|
|
|
try:
|
|
await session.flush()
|
|
except Exception:
|
|
raise RuntimeError()
|
|
|
|
diary = await cls.get_by_id(session, user_id, diary.id)
|
|
|
|
if not diary:
|
|
raise RuntimeError()
|
|
await Meal.create(session, diary.id)
|
|
return diary
|
|
|
|
@classmethod
|
|
async def get_by_id(
|
|
cls, session: AsyncSession, user_id: int, id: int
|
|
) -> "Optional[Diary]":
|
|
"""get_by_id."""
|
|
query = (
|
|
select(cls)
|
|
.where(cls.user_id == user_id)
|
|
.where(cls.id == id)
|
|
.options(joinedload(cls.meals))
|
|
)
|
|
return await session.scalar(query)
|