Fix issue with getting console width

Author: mdmintz

seleniumbase/console_scripts/sb_print.py

@@ -130,7 +130,7 @@ def main():
     used_width = None  # code_width and few spaces on right for padding
     magic_syntax = None  # the syntax generated by rich.syntax.Syntax()
     try:
-        console_width = os.get_terminal_size()[0]
+        console_width = os.get_terminal_size().columns
         if console_width:
             console_width = int(console_width)
     except Exception:

seleniumbase/fixtures/shared_utils.py

@@ -1,22 +1,15 @@
 """
 This module contains shared utility methods.
 """
-import fasteners
 import subprocess
 import sys
-import time
-from selenium.common.exceptions import ElementNotVisibleException
-from selenium.common.exceptions import NoAlertPresentException
-from selenium.common.exceptions import NoSuchAttributeException
-from selenium.common.exceptions import NoSuchElementException
-from selenium.common.exceptions import NoSuchFrameException
-from selenium.common.exceptions import NoSuchWindowException
-from seleniumbase.common.exceptions import TextNotVisibleException
 from seleniumbase.fixtures import constants
 from seleniumbase import config as sb_config
 
 
 def pip_install(package, version=None):
+    import fasteners
+
     pip_install_lock = fasteners.InterProcessLock(
         constants.PipInstall.LOCKFILE
     )
@@ -32,11 +25,47 @@ def pip_install(package, version=None):
             )
 
 
+def is_windows():
+    platform = sys.platform
+    if "win32" in platform or "win64" in platform or "x64" in platform:
+        return True
+    else:
+        return False
+
+
+def get_terminal_width():
+    import os
+
+    width = 80  # default
+    try:
+        width = os.get_terminal_size().columns
+    except Exception:
+        try:
+            if is_windows():
+                raise Exception("Don't even try 'tput cols' on Windows!")
+            width = int(subprocess.check_output(["tput", "cols"]))
+        except Exception:
+            try:
+                import shutil
+
+                width = shutil.get_terminal_size((80, 20)).columns
+            except Exception:
+                pass
+    return width
+
+
 def format_exc(exception, message):
     """
     Formats an exception message to make the output cleaner.
     """
+    from selenium.common.exceptions import ElementNotVisibleException
+    from selenium.common.exceptions import NoAlertPresentException
+    from selenium.common.exceptions import NoSuchAttributeException
+    from selenium.common.exceptions import NoSuchElementException
+    from selenium.common.exceptions import NoSuchFrameException
+    from selenium.common.exceptions import NoSuchWindowException
     from seleniumbase.common.exceptions import NoSuchFileException
+    from seleniumbase.common.exceptions import TextNotVisibleException
 
     if exception == Exception:
         exc = Exception
@@ -101,6 +130,8 @@ def check_if_time_limit_exceeded():
         and sb_config.time_limit
         and not sb_config.recorder_mode
     ):
+        import time
+
         time_limit = sb_config.time_limit
         now_ms = int(time.time() * 1000)
         if now_ms > sb_config.start_time_ms + sb_config.time_limit_ms:

seleniumbase/plugins/sb_manager.py

@@ -95,6 +95,7 @@ def SB(
     from seleniumbase import config as sb_config
     from seleniumbase.config import settings
     from seleniumbase.fixtures import constants
+    from seleniumbase.fixtures import shared_utils
 
     sb_config_backup = sb_config
     sys_argv = sys.argv
@@ -692,6 +693,7 @@ def SB(
     else:
         sb.headless_active = False
     test_name = None
+    terminal_width = shared_utils.get_terminal_width()
     if test:
         import colorama
         import os
@@ -703,7 +705,6 @@ def SB(
         stack_base = traceback.format_stack()[0].split(os.sep)[-1]
         test_name = stack_base.split(", in ")[0].replace('", line ', ":")
         test_name += ":SB"
-        terminal_width = os.get_terminal_size()[0]
         start_text = "=== {%s} starts ===" % test_name
         remaining_spaces = terminal_width - len(start_text)
         left_space = ""
@@ -764,7 +765,6 @@ def SB(
             if not test_passed:
                 result = "failed"
                 c1 = colorama.Fore.RED
-            terminal_width = os.get_terminal_size()[0]
             end_text = (
                 "=== {%s} %s in %.2fs ==="
                 % (test_name, result, run_time)