diff --git a/backend/db/schemas.py b/backend/db/schemas.py index a03415915bd617e5ea153030b54a8e841cee363d..0777b3a630414a37958b339dcdd32d94f9738226 100644 --- a/backend/db/schemas.py +++ b/backend/db/schemas.py @@ -68,6 +68,7 @@ class NewsBase(BaseModel): class News(NewsBase): """Database news base schema""" + id: int published_at: datetime = Field(..., title="Publication date of the news") class Config: @@ -99,13 +100,21 @@ class OpeningHours(OpeningHoursBase): orm_mode = True -class Closure(BaseModel): - """ Closure schema """ +class ClosureBase(BaseModel): + """ Closure schema base """ place: str = Field(..., title="Name of the restaurant") beginning_date: datetime = Field(..., title="Beginning date of closure") end_date: datetime = Field(..., title="Ending date of closure") +class Closure(ClosureBase): + """ Closure schema """ + id: int + + class Config: + orm_mode = True + + class Restaurant(BaseModel): """Restaurant schema for reading""" name: str