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

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 CreateIngredientUnit, 

13 IngredientUnit, 

14 IngredientUnitPagination, 

15 MergeUnit, 

16 SaveIngredientUnit, 

17) 

18from mealie.schema.response.pagination import PaginationQuery 

19from mealie.schema.response.responses import SuccessResponse 

20 

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

22 

23 

24@controller(router) 

25class IngredientUnitsController(BaseUserController): 

26 @cached_property 

27 def repo(self): 

28 return self.repos.ingredient_units 

29 

30 @cached_property 

31 def mixins(self): 

32 return HttpRepo[CreateIngredientUnit, IngredientUnit, CreateIngredientUnit]( 

33 self.repo, 

34 self.logger, 

35 self.registered_exceptions, 

36 ) 

37 

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

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

40 response = self.repo.page_all( 

41 pagination=q, 

42 override=IngredientUnit, 

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=IngredientUnit, status_code=201) 

50 def create_one(self, data: CreateIngredientUnit): 

51 save_data = mapper.cast(data, SaveIngredientUnit, 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: MergeUnit): 

56 try: 

57 self.repo.merge(data.from_unit, data.to_unit) 

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

59 except Exception as e: 

60 self.logger.error(e) 

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

62 

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

64 def get_one(self, item_id: UUID4): 

65 return self.mixins.get_one(item_id) 

66 

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

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

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

70 

71 @router.delete("/{item_id}", response_model=IngredientUnit) 

72 def delete_one(self, item_id: UUID4): 

73 return self.mixins.delete_one(item_id) # type: ignore