| 32 | |
| 33 | @dataclass |
| 34 | class ExceptHookHandler: |
| 35 | lock: threading.Lock = field(default_factory=threading.Lock) |
| 36 | messages: list[str] = field(default_factory=list) |
| 37 | |
| 38 | def show(self) -> int: |
| 39 | count = 0 |
| 40 | with self.lock: |
| 41 | if not self.messages: |
| 42 | return 0 |
| 43 | reader.restore() |
| 44 | for tb in self.messages: |
| 45 | count += 1 |
| 46 | if tb: |
| 47 | print(tb) |
| 48 | self.messages.clear() |
| 49 | reader.scheduled_commands.append("ctrl-c") |
| 50 | reader.prepare() |
| 51 | return count |
| 52 | |
| 53 | def add(self, s: str) -> None: |
| 54 | with self.lock: |
| 55 | self.messages.append(s) |
| 56 | |
| 57 | def exception(self, args: ExceptHookArgs) -> None: |
| 58 | lines = traceback.format_exception( |
| 59 | args.exc_type, |
| 60 | args.exc_value, |
| 61 | args.exc_traceback, |
| 62 | colorize=reader.can_colorize, |
| 63 | ) # type: ignore[call-overload] |
| 64 | pre = f"\nException in {args.thread.name}:\n" if args.thread else "\n" |
| 65 | tb = pre + "".join(lines) |
| 66 | self.add(tb) |
| 67 | |
| 68 | def __call__(self) -> int: |
| 69 | return self.show() |
| 70 | |
| 71 | |
| 72 | handler = ExceptHookHandler() |
no test coverage detected
searching dependent graphs…