#183 added backend validation for mealplanning timespan

This commit is contained in:
Bernhard Großer 2021-02-24 16:56:44 +01:00
commit eda47b7bf4

View file

@ -2,7 +2,7 @@ from datetime import date, timedelta
from typing import List, Optional from typing import List, Optional
from db.database import db from db.database import db
from pydantic import BaseModel from pydantic import BaseModel, validator
from sqlalchemy.orm.session import Session from sqlalchemy.orm.session import Session
from services.recipe_services import Recipe from services.recipe_services import Recipe
@ -41,6 +41,12 @@ class MealPlan(BaseModel):
} }
} }
@validator('endDate')
def endDate_after_startDate(cls, v, values, **kwargs):
if 'startDate' in values and v < values['startDate']:
raise ValueError('EndDate should be greater than StartDate')
return v
def process_meals(self, session: Session): def process_meals(self, session: Session):
meals = [] meals = []
for x, meal in enumerate(self.meals): for x, meal in enumerate(self.meals):