Coverage for opt/mealie/lib/python3.12/site-packages/mealie/routes/unit_and_foods/foods.py: 100%

1from functools import cached_property 

2 

3from fastapi import APIRouter, Depends, HTTPException 

4from pydantic import UUID4 

5 

6from mealie.routes._base.base_controllers import BaseUserController 

7from mealie.routes._base.controller import controller 

8from mealie.routes._base.mixins import HttpRepo 

9from mealie.routes._base.routers import MealieCrudRoute 

10from mealie.schema import mapper 

11from mealie.schema.recipe.recipe_ingredient import ( 

12 CreateIngredientFood, 

13 IngredientFood, 

14 IngredientFoodPagination, 

15 MergeFood, 

16 SaveIngredientFood, 

17) 

18from mealie.schema.response.pagination import PaginationQuery 

19from mealie.schema.response.responses import SuccessResponse 

20 

21router = APIRouter(prefix="/foods", tags=["Recipes: Foods"], route_class=MealieCrudRoute) 

22 

23 

24@controller(router) 

25class IngredientFoodsController(BaseUserController): 

26 @cached_property 

27 def repo(self): 

28 return self.repos.ingredient_foods 

29 

30 @cached_property 

31 def mixins(self): 

32 return HttpRepo[SaveIngredientFood, IngredientFood, CreateIngredientFood]( 

33 self.repo, 

34 self.logger, 

35 self.registered_exceptions, 

36 ) 

37 

38 @router.get("", response_model=IngredientFoodPagination) 

39 def get_all(self, q: PaginationQuery = Depends(PaginationQuery), search: str | None = None): 

40 response = self.repo.page_all( 

41 pagination=q, 

42 override=IngredientFood, 

43 search=search, 

44 ) 

45 

46 response.set_pagination_guides(router.url_path_for("get_all"), q.model_dump()) 

47 return response 

48 

49 @router.post("", response_model=IngredientFood, status_code=201) 

50 def create_one(self, data: CreateIngredientFood): 

51 save_data = mapper.cast(data, SaveIngredientFood, group_id=self.group_id) 

52 return self.mixins.create_one(save_data) 

53 

54 @router.put("/merge", response_model=SuccessResponse) 

55 def merge_one(self, data: MergeFood): 

56 try: 

57 self.repo.merge(data.from_food, data.to_food) 

58 return SuccessResponse.respond("Successfully merged foods") 

59 except Exception as e: 

60 self.logger.error(e) 

61 raise HTTPException(500, "Failed to merge foods") from e 

62 

63 @router.get("/{item_id}", response_model=IngredientFood) 

64 def get_one(self, item_id: UUID4): 

65 return self.mixins.get_one(item_id) 

66 

67 @router.put("/{item_id}", response_model=IngredientFood) 

68 def update_one(self, item_id: UUID4, data: CreateIngredientFood): 

69 data = mapper.cast(data, SaveIngredientFood, group_id=self.group_id) 

70 return self.mixins.update_one(data, item_id) 

71 

72 @router.delete("/{item_id}", response_model=IngredientFood) 

73 def delete_one(self, item_id: UUID4): 

74 return self.mixins.delete_one(item_id)