from fastapi import HTTPException from ..model.entry import Entry, CreateEntryPayload, UpdateEntryPayload from ..domain.entry import Entry as DBEntry from ..domain.meal import Meal as DBMeal from .base import AuthorizedController class CreateEntry(AuthorizedController): async def call(self, content: CreateEntryPayload) -> Entry: async with self.async_session.begin() as session: meal = await DBMeal.get_by_id(session, self.user.id, content.meal_id) if meal is None: raise HTTPException(status_code=404, detail="meal not found") try: entry = await DBEntry.create( session, content.meal_id, content.product_id, content.grams ) return Entry.from_orm(entry) except AssertionError as e: raise HTTPException(status_code=400, detail=e.args[0]) class UpdateEntry(AuthorizedController): async def call(self, entry_id: int, content: UpdateEntryPayload) -> Entry: async with self.async_session.begin() as session: entry = await DBEntry.get_by_id(session, self.user.id, entry_id) if entry is None: raise HTTPException(status_code=404, detail="entry not found") try: await entry.update( session, content.meal_id, content.product_id, content.grams ) return Entry.from_orm(entry) except AssertionError as e: raise HTTPException(status_code=400, detail=e.args[0]) class DeleteEntry(AuthorizedController): async def call(self, entry_id: int) -> Entry: async with self.async_session.begin() as session: entry = await DBEntry.get_by_id(session, self.user.id, entry_id) if entry is None: raise HTTPException(status_code=404, detail="entry not found") try: await entry.delete(session) except AssertionError as e: raise HTTPException(status_code=400, detail=e.args[0])