Coverage for opt/mealie/lib/python3.12/site-packages/mealie/services/exporter/exporter.py: 100%

1import datetime 1 ctx1a

2import shutil 1 ctx1a

3import zipfile 1 ctx1a

4from pathlib import Path 1 ctx1a

5from uuid import UUID, uuid4 1 ctx1a

6 

7from mealie.pkgs.stats.fs_stats import pretty_size 1 ctx1a

8from mealie.repos.all_repositories import AllRepositories 1 ctx1a

9from mealie.schema.group.group_exports import GroupDataExport 1 ctx1a

10from mealie.schema.user import GroupInDB 1 ctx1a

11 

12from .._base_service import BaseService 1 ctx1a

13from ._abc_exporter import ABCExporter 1 ctx1a

14 

15 

16class Exporter(BaseService): 1 ctx1a

17 def __init__(self, group_id: UUID, temp_zip: Path, exporters: list[ABCExporter]) -> None: 1 ctx1a

18 super().__init__() (empty)

19 

20 self.group_id = group_id (empty)

21 self.temp_path = temp_zip (empty)

22 self.exporters = exporters (empty)

23 

24 def run(self, db: AllRepositories) -> GroupDataExport: 1 ctx1a

25 # Create Zip File 

26 self.temp_path.touch() (empty)

27 

28 # Open Zip File 

29 with zipfile.ZipFile(self.temp_path, "w") as zip: (empty)

30 for exporter in self.exporters: (empty)

31 exporter.export(zip) (empty)

32 

33 export_id = uuid4() (empty)

34 

35 export_path = GroupInDB.get_export_directory(self.group_id) / f"{export_id}.zip" (empty)

36 

37 shutil.copy(self.temp_path, export_path) (empty)

38 

39 group_data_export = GroupDataExport( (empty)

40 id=export_id, 

41 group_id=self.group_id, 

42 path=str(export_path), 

43 name="Data Export", 

44 size=pretty_size(export_path.stat().st_size), 

45 filename=export_path.name, 

46 expires=datetime.datetime.now(datetime.UTC) + datetime.timedelta(days=1), 

47 ) 

48 

49 db.group_exports.create(group_data_export) (empty)

50 

51 return group_data_export (empty)