Fixed ZeroDivisionError in async_alert() when shutil.get_terminal_size().columns returned 0.

Author: kmvanbrunt

CHANGELOG.md

@@ -1,3 +1,8 @@
+## 2.5.4 (TBD)
+* Bug Fixes
+  * Fixed `ZeroDivisionError` in `async_alert()` when `shutil.get_terminal_size().columns`
+    returned 0.
+
 ## 2.5.3 (November 5, 2024)
 * Enhancements
   * Changed `CommandSet._cmd` to a read-only property which never returns `None` because it

cmd2/cmd2.py

@@ -5335,9 +5335,12 @@ def async_alert(self, alert_msg: str, new_prompt: Optional[str] = None) -> None:
             if update_terminal:
                 import shutil
 
+                # Prior to Python 3.11, this can return 0 for things like pseudo terminals.
+                terminal_columns = shutil.get_terminal_size().columns or 80
+
                 # Print a string which replaces the onscreen prompt and input lines with the alert.
                 terminal_str = ansi.async_alert_str(
-                    terminal_columns=shutil.get_terminal_size().columns,
+                    terminal_columns=terminal_columns,
                     prompt=rl_get_display_prompt(),
                     line=readline.get_line_buffer(),
                     cursor_offset=rl_get_point(),

cmd2/utils.py

@@ -874,7 +874,8 @@ def align_text(
     )
 
     if width is None:
-        width = shutil.get_terminal_size().columns
+        # Prior to Python 3.11, this can return 0 for things like pseudo terminals.
+        width = shutil.get_terminal_size().columns or 80
 
     if width < 1:
         raise ValueError("width must be at least 1")

tests/test_utils.py

@@ -624,7 +624,9 @@ def test_align_text_term_width():
     text = 'foo'
     fill_char = ' '
 
-    term_width = shutil.get_terminal_size().columns
+    # Prior to Python 3.11, this can return 0 for things like pseudo terminals.
+    # So copy the behavior of cu.align_text() and fall back to 80 if needed.
+    term_width = shutil.get_terminal_size().columns or 80
     expected_fill = (term_width - ansi.style_aware_wcswidth(text)) * fill_char
 
     aligned = cu.align_text(text, cu.TextAlignment.LEFT, fill_char=fill_char)