Add a category filter to Recipe

backend/models/filters.py

@@ -11,3 +11,12 @@ class Length(Enum):
     UNDER_30 = "under_30"
     UNDER_60 = "under_60"
     OVER_60 = "over_60"
+
+
+class Category(Enum):
+    VEGAN = "vegan"
+    VEGETARIAN = "vegetarian"
+    FISH = "fish"
+    MEAT = "meat"
+    SIDE = "side"
+    DESSERT = "dessert"

backend/models/recipe.py

@@ -2,7 +2,7 @@
 from enum import Enum
 from uuid import uuid4
 from backend.services.dynamodb import DynamoDBClient
-from backend.models.filters import Difficulty, Length
+from backend.models.filters import Category, Difficulty, Length
 
 from typing import Any, Dict, List, Optional
 
@@ -46,6 +46,7 @@ def __init__(
         tags: List[str],
         difficulty: Difficulty,
         length: Length,
+        category: Category,
         id: Optional[str] = None,
     ) -> None:
         self.id = id
@@ -55,6 +56,7 @@ def __init__(
         self.source = source
         self.difficulty = difficulty
         self.length = length
+        self.category = category
 
         if self.id is None:
             self.id = uuid4()
@@ -74,6 +76,7 @@ def to_dict(self) -> Dict[str, str | Dict[str, str] | List[str]]:
             "tags": self.tags,
             "difficulty": self.difficulty.value,
             "length": self.length.value,
+            "category": self.category.value,
         }
 
     @classmethod
@@ -106,6 +109,7 @@ def from_dict(cls, input: Dict[str, Any]) -> "Recipe":
                 ingredients=input["ingredients"],
                 difficulty=Difficulty(input["difficulty"]),
                 length=Length(input["length"]),
+                category=Category(input["category"]),
                 id=id,
             )
         except:

backend/tests/fixtures.py

@@ -19,4 +19,5 @@
     "source": SOURCE_WEBSITE_DICT,
     "difficulty": "easy",
     "length": "under_30",
+    "category": "vegetarian",
 }

backend/tests/models/test_recipe.py

@@ -34,6 +34,7 @@ def test_recipe_saves_id(self, dynamodb_mock: Mock):
             tags=[],
             difficulty="easy",
             length="under_30",
+            category="vegetarian",
         )
         assert recipe.id == "id"
 
@@ -48,6 +49,7 @@ def test_recipe_generates_id_if_none(self, uuid_mock: Mock, dynamodb_mock: Mock)
             tags=[],
             difficulty="easy",
             length="under_30",
+            category="vegetarian",
         )
         assert recipe.id == "generated_id"
 

scripts/post_test_recipe.py

@@ -17,6 +17,7 @@
         "source": {"type": "website", "address": "address"},
         "difficulty": "easy",
         "length": "under_30",
+        "category": "vegetarian",
     },
 )