From 96ae36c0bc8e5aee035483aa5ed10596617c3912 Mon Sep 17 00:00:00 2001 From: Nima Date: Wed, 27 May 2020 23:54:36 +0430 Subject: [PATCH] bug fixed. - getter added for category. --- foods/models.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/foods/models.py b/foods/models.py index 6af6d51..c1b06a7 100644 --- a/foods/models.py +++ b/foods/models.py @@ -41,7 +41,7 @@ def recipe(self): return None else: payload = json.loads(self.Recipe) - payload['category'] = self.Category + payload['category'] = self.get_category() payload['date_created'] = self.CreatedAt.strftime('%Y-%m-%d %H:%M:%S') return payload @@ -55,7 +55,7 @@ def simple_view(self): """ return { 'id': self.id, - 'category': self.Category, + 'category': self.get_category(), 'image': self.Image, 'thumbnail': self.Thumbnail, 'title': self.Title, @@ -71,6 +71,9 @@ def __str__(self): def get_calorie(self): return self.Calories + def get_category(self): + return self.Category.strip().lower() + # for admin integration class FoodModelView(ModelView):