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

1from functools import cached_property 

2 

3from fastapi import HTTPException 

4from pydantic import UUID4 

5 

6from mealie.repos.all_repositories import get_repositories 

7from mealie.routes._base import BaseUserController, controller 

8from mealie.routes._base.mixins import HttpRepo 

9from mealie.routes._base.routers import UserAPIRouter 

10from mealie.schema.recipe import RecipeShareTokenSummary 

11from mealie.schema.recipe.recipe_share_token import RecipeShareToken, RecipeShareTokenCreate, RecipeShareTokenSave 

12 

13router = UserAPIRouter(prefix="/shared/recipes", tags=["Shared: Recipes"]) 

14 

15 

16@controller(router) 

17class RecipeSharedController(BaseUserController): 

18 @cached_property 

19 def repo(self): 

20 return self.repos.recipe_share_tokens 

21 

22 @cached_property 

23 def mixins(self): 

24 return HttpRepo[RecipeShareTokenSave, RecipeShareToken, RecipeShareTokenCreate](self.repo, self.logger) 

25 

26 @router.get("", response_model=list[RecipeShareTokenSummary]) 

27 def get_all(self, recipe_id: UUID4 | None = None): 

28 if recipe_id: 

29 return self.repo.multi_query({"recipe_id": recipe_id}, override_schema=RecipeShareTokenSummary) 

30 else: 

31 return self.repo.get_all(override=RecipeShareTokenSummary) 

32 

33 @router.post("", response_model=RecipeShareToken, status_code=201) 

34 def create_one(self, data: RecipeShareTokenCreate) -> RecipeShareToken: 

35 # check if recipe group id is the same as the user group id 

36 group_repos = get_repositories(self.repos.session, group_id=self.group_id, household_id=None) 

37 recipe = group_repos.recipes.get_one(data.recipe_id, "id") 

38 if recipe is None or recipe.group_id != self.group_id: 

39 raise HTTPException(status_code=404, detail="Recipe not found in your group") 

40 

41 save_data = RecipeShareTokenSave(**data.model_dump(), group_id=self.group_id) 

42 return self.mixins.create_one(save_data) 

43 

44 @router.get("/{item_id}", response_model=RecipeShareToken) 

45 def get_one(self, item_id: UUID4): 

46 return self.mixins.get_one(item_id) 

47 

48 @router.delete("/{item_id}") 

49 def delete_one(self, item_id: UUID4) -> None: 

50 return self.mixins.delete_one(item_id)