From eda47b7bf41b562d9aa60c3ebb726dbbe13efe97 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bernhard=20Gro=C3=9Fer?= <30469627+boerniee@users.noreply.github.com> Date: Wed, 24 Feb 2021 16:56:44 +0100 Subject: [PATCH] #183 added backend validation for mealplanning timespan --- mealie/services/meal_services.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/mealie/services/meal_services.py b/mealie/services/meal_services.py index a723b3fde..d681112b1 100644 --- a/mealie/services/meal_services.py +++ b/mealie/services/meal_services.py @@ -2,7 +2,7 @@ from datetime import date, timedelta from typing import List, Optional from db.database import db -from pydantic import BaseModel +from pydantic import BaseModel, validator from sqlalchemy.orm.session import Session 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): meals = [] for x, meal in enumerate(self.meals):