Coverage for opt/mealie/lib/python3.12/site-packages/mealie/services/group_services/labels_service.py: 89%

1from mealie.repos.all_repositories import get_repositories 1 ctx1j

2from mealie.repos.repository_factory import AllRepositories 1 ctx1j

3from mealie.schema.household.group_shopping_list import ShoppingListMultiPurposeLabelCreate 1 ctx1j

4from mealie.schema.labels.multi_purpose_label import ( 1 ctx1j

5 MultiPurposeLabelCreate, 

6 MultiPurposeLabelOut, 

7 MultiPurposeLabelSave, 

8) 

9from mealie.schema.response.pagination import PaginationQuery 1 ctx1j

10 

11 

12class MultiPurposeLabelService: 1 ctx1j

13 def __init__(self, repos: AllRepositories): 1 ctx1j

14 self.repos = repos 9 ctx1abcdefghi

15 self.labels = repos.group_multi_purpose_labels 9 ctx1abcdefghi

16 

17 def _update_shopping_list_label_references(self, new_labels: list[MultiPurposeLabelOut]) -> None: 1 ctx1j

18 # remove the households filter so we get all lists 

19 household_repos = get_repositories(self.repos.session, group_id=self.repos.group_id, household_id=None) 9 ctx1abcdefghi

20 shopping_lists_repo = household_repos.group_shopping_lists 9 ctx1abcdefghi

21 shopping_list_multi_purpose_labels_repo = household_repos.shopping_list_multi_purpose_labels 9 ctx1abcdefghi

22 

23 shopping_lists = shopping_lists_repo.page_all(PaginationQuery(page=1, per_page=-1)) 9 ctx1abcdefghi

24 new_shopping_list_labels: list[ShoppingListMultiPurposeLabelCreate] = [] 9 ctx1abcdefghi

25 for label in new_labels: 9 ctx1abcdefghi

26 new_shopping_list_labels.extend( 9 ctx1abcdefghi

27 [ 

28 ShoppingListMultiPurposeLabelCreate( 

29 shopping_list_id=shopping_list.id, label_id=label.id, position=len(shopping_list.label_settings) 

30 ) 

31 for shopping_list in shopping_lists.items 

32 ] 

33 ) 

34 

35 shopping_list_multi_purpose_labels_repo.create_many(new_shopping_list_labels) 9 ctx1abcdefghi

36 

37 def create_one(self, data: MultiPurposeLabelCreate) -> MultiPurposeLabelOut: 1 ctx1j

38 label = self.labels.create(data.cast(MultiPurposeLabelSave, group_id=self.repos.group_id)) 9 ctx1abcdefghi

39 self._update_shopping_list_label_references([label]) 9 ctx1abcdefghi

40 return label 9 ctx1abcdefghi

41 

42 def create_many(self, data: list[MultiPurposeLabelCreate]) -> list[MultiPurposeLabelOut]: 1 ctx1j

43 labels = self.labels.create_many( 

44 [label.cast(MultiPurposeLabelSave, group_id=self.repos.group_id) for label in data] 

45 ) 

46 self._update_shopping_list_label_references(labels) 

47 return labels