fooder-api/fooder/controller/diary.py

from datetime import date

from fastapi import HTTPException

from ..domain.diary import Diary as DBDiary
from ..model.diary import Diary
from .base import AuthorizedController


class GetDiary(AuthorizedController):
    async def call(self, date: date) -> Diary:
        async with self.async_session() as session:
            diary = await DBDiary.get_diary(session, self.user.id, date)

            if diary is not None:
                return Diary.from_orm(diary)
            else:
                try:
                    await DBDiary.create(session, self.user.id, date)
                    await session.commit()
                    return Diary.from_orm(
                        await DBDiary.get_diary(session, self.user.id, date)
                    )
                except AssertionError as e:
                    raise HTTPException(status_code=400, detail=e.args[0])