- try:
- rows, columns = cmd(
- "stty size",
- timeout_seconds=1.0,
- ).split()
- except Exception as e:
- logger.exception(e)
- raise Exception('Can\'t determine console size?!') from e
- return RowsColumns(int(rows), int(columns))
+ rows: Optional[str] = os.environ.get('LINES', None)
+ cols: Optional[str] = os.environ.get('COLUMNS', None)
+ if not rows or not cols:
+ try:
+ rows, cols = cmd(
+ "stty size",
+ timeout_seconds=1.0,
+ ).split()
+ except Exception:
+ rows = None
+ cols = None
+
+ if rows is None:
+ try:
+ rows = cmd(
+ "tput rows",
+ timeout_seconds=1.0,
+ )
+ except Exception:
+ rows = None
+
+ if cols is None:
+ try:
+ cols = cmd(
+ "tput cols",
+ timeout_seconds=1.0,
+ )
+ except Exception:
+ cols = None
+
+ if not rows or not cols:
+ raise Exception('Can\'t determine console size?!')
+ return RowsColumns(int(rows), int(cols))