From b29be4f1750fd20bd2eada88e751dfae85817882 Mon Sep 17 00:00:00 2001 From: Scott Date: Sat, 4 Dec 2021 21:23:11 -0800 Subject: Various changes. --- text_utils.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'text_utils.py') diff --git a/text_utils.py b/text_utils.py index 36cfe2f..9a9eb54 100644 --- a/text_utils.py +++ b/text_utils.py @@ -3,6 +3,7 @@ """Utilities for dealing with "text".""" from collections import defaultdict +import logging import math import sys from typing import List, NamedTuple, Optional @@ -10,6 +11,9 @@ from typing import List, NamedTuple, Optional from ansi import fg, reset +logger = logging.getLogger(__file__) + + class RowsColumns(NamedTuple): rows: int columns: int @@ -18,8 +22,15 @@ class RowsColumns(NamedTuple): def get_console_rows_columns() -> RowsColumns: """Returns the number of rows/columns on the current console.""" - from exec_utils import cmd - rows, columns = cmd("stty size").split() + from exec_utils import cmd_with_timeout + try: + rows, columns = cmd_with_timeout( + "stty size", + timeout_seconds=5.0, + ).split() + except Exception as e: + logger.exception(e) + raise Exception('Can\'t determine console size?!') return RowsColumns(int(rows), int(columns)) -- cgit v1.3