Coverage for opt/mealie/lib/python3.12/site-packages/mealie/services/scraper/scraped_extras.py: 34%

1from dataclasses import dataclass 

2 

3from slugify import slugify 

4 

5from mealie.repos.repository_factory import AllRepositories 

6from mealie.schema.recipe import TagOut 

7from mealie.schema.recipe.recipe_category import TagSave 

8 

9 

10class NoContextException(Exception): 

11 pass 

12 

13 

14@dataclass(slots=True) 

15class ScraperContext: 

16 repos: AllRepositories 

17 

18 

19class ScrapedExtras: 

20 def __init__(self) -> None: 

21 self._tags: list[str] = [] 

22 

23 def set_tags(self, tags: list[str]) -> None: 

24 self._tags = tags 

25 

26 def use_tags(self, ctx: ScraperContext) -> list[TagOut]: 

27 if not self._tags: 

28 return [] 

29 

30 repo = ctx.repos.tags 

31 

32 tags = [] 

33 seen_tag_slugs: set[str] = set() 

34 for tag in self._tags: 

35 slugify_tag = slugify(tag) 

36 if slugify_tag in seen_tag_slugs: 

37 continue 

38 

39 seen_tag_slugs.add(slugify_tag) 

40 

41 # Check if tag exists 

42 if db_tag := repo.get_one(slugify_tag, "slug"): 

43 tags.append(db_tag) 

44 continue 

45 

46 save_data = TagSave(name=tag, group_id=ctx.repos.group_id) 

47 db_tag = repo.create(save_data) 

48 

49 tags.append(db_tag) 

50 

51 return tags