| from app.logger import on_flush |
| import os |
| import shutil |
|
|
|
|
| class TerminalService: |
| def __init__(self, server): |
| self.server = server |
| self.cols = None |
| self.rows = None |
| self.subscriptions = set() |
| on_flush(self.send_messages) |
|
|
| def get_terminal_size(self): |
| try: |
| size = os.get_terminal_size() |
| return (size.columns, size.lines) |
| except OSError: |
| try: |
| size = shutil.get_terminal_size() |
| return (size.columns, size.lines) |
| except OSError: |
| return (80, 24) |
|
|
| def update_size(self): |
| columns, lines = self.get_terminal_size() |
| changed = False |
|
|
| if columns != self.cols: |
| self.cols = columns |
| changed = True |
|
|
| if lines != self.rows: |
| self.rows = lines |
| changed = True |
|
|
| if changed: |
| return {"cols": self.cols, "rows": self.rows} |
|
|
| return None |
|
|
| def subscribe(self, client_id): |
| self.subscriptions.add(client_id) |
|
|
| def unsubscribe(self, client_id): |
| self.subscriptions.discard(client_id) |
|
|
| def send_messages(self, entries): |
| if not len(entries) or not len(self.subscriptions): |
| return |
|
|
| new_size = self.update_size() |
|
|
| for client_id in self.subscriptions.copy(): |
| if client_id not in self.server.sockets: |
| |
| self.unsubscribe(client_id) |
| continue |
|
|
| self.server.send_sync("logs", {"entries": entries, "size": new_size}, client_id) |
|
|
