buitu0 commited on
Commit
90dfcaa
·
verified ·
1 Parent(s): 19a5800

Delete printer.py

Browse files
Files changed (1) hide show
  1. printer.py +0 -46
printer.py DELETED
@@ -1,46 +0,0 @@
1
- from typing import Any
2
-
3
- from rich.console import Console, Group
4
- from rich.live import Live
5
- from rich.spinner import Spinner
6
-
7
-
8
- class Printer:
9
- """
10
- Simple wrapper to stream status updates. Used by the financial bot
11
- manager as it orchestrates planning, search and writing.
12
- """
13
-
14
- def __init__(self, console: Console) -> None:
15
- self.live = Live(console=console)
16
- self.items: dict[str, tuple[str, bool]] = {}
17
- self.hide_done_ids: set[str] = set()
18
- self.live.start()
19
-
20
- def end(self) -> None:
21
- self.live.stop()
22
-
23
- def hide_done_checkmark(self, item_id: str) -> None:
24
- self.hide_done_ids.add(item_id)
25
-
26
- def update_item(
27
- self, item_id: str, content: str, is_done: bool = False, hide_checkmark: bool = False
28
- ) -> None:
29
- self.items[item_id] = (content, is_done)
30
- if hide_checkmark:
31
- self.hide_done_ids.add(item_id)
32
- self.flush()
33
-
34
- def mark_item_done(self, item_id: str) -> None:
35
- self.items[item_id] = (self.items[item_id][0], True)
36
- self.flush()
37
-
38
- def flush(self) -> None:
39
- renderables: list[Any] = []
40
- for item_id, (content, is_done) in self.items.items():
41
- if is_done:
42
- prefix = "✅ " if item_id not in self.hide_done_ids else ""
43
- renderables.append(prefix + content)
44
- else:
45
- renderables.append(Spinner("dots", text=content))
46
- self.live.update(Group(*renderables))