Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.11/site-packages/unblob/ui.py: 63%

1from typing import Protocol 

2 

3from rich import progress 

4from rich.style import Style 

5 

6from .models import TaskResult 

7 

8 

9class ProgressReporter(Protocol): 

10 def __enter__(self): ... 

11 

12 def __exit__(self, _exc_type, _exc_value, _tb): ... 

13 

14 def update(self, result: TaskResult): ... 

15 

16 

17class NullProgressReporter: 

18 def __enter__(self): 

19 pass 

20 

21 def __exit__(self, _exc_type, _exc_value, _tb): 

22 pass 

23 

24 def update(self, result: TaskResult): 

25 pass 

26 

27 

28class RichConsoleProgressReporter: 

29 def __init__(self): 

30 self._progress = progress.Progress( 

31 progress.TextColumn( 

32 "Extraction progress: {task.percentage:>3.0f}%", 

33 style=Style(color="#00FFC8"), 

34 ), 

35 progress.BarColumn( 

36 complete_style=Style(color="#00FFC8"), style=Style(color="#002060") 

37 ), 

38 ) 

39 self._overall_progress_task = self._progress.add_task("Extraction progress:") 

40 

41 def __enter__(self): 

42 self._progress.start() 

43 

44 def __exit__(self, _exc_type, _exc_value, _tb): 

45 self._progress.remove_task(self._overall_progress_task) 

46 self._progress.stop() 

47 

48 def update(self, result: TaskResult): 

49 if (total := self._progress.tasks[0].total) is not None: 

50 self._progress.update( 

51 self._overall_progress_task, 

52 advance=1, 

53 total=total + len(result.subtasks), 

54 )