Merge pull request #1635 from seleniumbase/context-manager-improvements

SB Context Manager improvements

Author: mdmintz

examples/verify_undetected.py

@@ -11,7 +11,7 @@ def test_browser_is_undetected(self):
         self.open("https://nowsecure.nl/#relax")
         try:
             self.assert_text("OH YEAH, you passed!", "h1", timeout=6.75)
-            self.post_message("Selenium wasn't detected!", duration=1.6)
+            self.post_message("Selenium wasn't detected!", duration=2.8)
             self._print("\n Success! Website did not detect Selenium! ")
         except Exception:
             self.fail('Selenium was detected! Try using: "pytest --uc"')

mkdocs_build/requirements.txt

@@ -20,7 +20,7 @@ importlib-metadata==5.1.0
 bleach==5.0.1
 jsmin==3.0.1
 lunr==0.6.2
-nltk==3.7
+nltk==3.8
 tornado==6.2
 watchdog==2.2.0
 cairocffi==1.4.0

requirements.txt

@@ -2,7 +2,8 @@ pip>=20.3.4;python_version<"3.6"
 pip>=21.3.1;python_version>="3.6" and python_version<"3.7"
 pip>=22.3.1;python_version>="3.7"
 packaging>=20.9;python_version<"3.6"
-packaging>=21.3;python_version>="3.6"
+packaging>=21.3;python_version>="3.6" and python_version<"3.7"
+packaging>=22.0;python_version>="3.7"
 setuptools>=44.1.1;python_version<"3.6"
 setuptools>=59.6.0;python_version>="3.6" and python_version<"3.7"
 setuptools>=65.6.3;python_version>="3.7"
@@ -15,7 +16,7 @@ attrs>=21.4.0;python_version<"3.6"
 attrs>=22.1.0;python_version>="3.6"
 PyYAML>=6.0;python_version>="3.6"
 certifi>=2021.10.8;python_version<"3.6"
-certifi>=2022.9.24;python_version>="3.6"
+certifi>=2022.12.7;python_version>="3.6"
 filelock>=3.2.1;python_version<"3.6"
 filelock>=3.4.1;python_version>="3.6" and python_version<"3.7"
 filelock>=3.8.2;python_version>="3.7"
@@ -98,7 +99,7 @@ pygments==2.13.0;python_version>="3.6"
 pyreadline==2.1;platform_system=="Windows" and python_version<"3.6"
 pyreadline3==3.4.1;platform_system=="Windows" and python_version>="3.6"
 tabcompleter==1.1.0
-pdbp==1.2.5
+pdbp==1.2.6
 colorama==0.4.6;python_version<"3.6"
 colorama==0.4.5;python_version>="3.6" and python_version<"3.7"
 colorama==0.4.6;python_version>="3.7"
@@ -107,11 +108,12 @@ importlib-metadata==2.1.3;python_version<"3.6"
 importlib-metadata==4.2.0;python_version>="3.6" and python_version<"3.8"
 pycparser==2.21
 pyotp==2.3.0;python_version<"3.6"
-pyotp==2.7.0;python_version>="3.6"
+pyotp==2.7.0;python_version>="3.6" and python_version<"3.7"
+pyotp==2.8.0;python_version>="3.7"
 cffi==1.15.1
 typing-extensions==3.10.0.2;python_version<"3.6"
 typing-extensions==4.1.1;python_version>="3.6" and python_version<"3.7"
-typing-extensions==4.2.0;python_version>="3.7" and python_version<"3.9"
+typing-extensions==4.4.0;python_version>="3.7" and python_version<"3.9"
 rich==12.6.0;python_version>="3.6" and python_version<"4.0"
 
 # --- Testing Requirements --- #
@@ -123,10 +125,13 @@ coverage==6.5.0;python_version>="3.7"
 pytest-cov==2.12.1;python_version<"3.6"
 pytest-cov==4.0.0;python_version>="3.6"
 flake8==3.7.9;python_version<"3.6"
-flake8==5.0.4;python_version>="3.6"
+flake8==5.0.4;python_version>="3.6" and python_version<"3.9"
+flake8==6.0.0;python_version>="3.9"
 mccabe==0.6.1;python_version<"3.6"
 mccabe==0.7.0;python_version>="3.6"
 pyflakes==2.1.1;python_version<"3.6"
-pyflakes==2.5.0;python_version>="3.6"
+pyflakes==2.5.0;python_version>="3.6" and python_version<"3.9"
+pyflakes==3.0.1;python_version>="3.9"
 pycodestyle==2.5.0;python_version<"3.6"
-pycodestyle==2.9.1;python_version>="3.6"
+pycodestyle==2.9.1;python_version>="3.6" and python_version<"3.9"
+pycodestyle==2.10.0;python_version>="3.9"

seleniumbase/__version__.py

@@ -1,2 +1,2 @@
 # seleniumbase package
-__version__ = "4.9.7"
+__version__ = "4.9.8"

seleniumbase/core/colored_traceback.py

@@ -0,0 +1,67 @@
+import sys
+
+
+def add_hook(always=False, style="default", debug=False):
+    import os
+
+    if os.environ.get("NO_COLOR", ""):
+        return  # https://no-color.org
+    isatty = getattr(sys.stderr, "isatty", lambda: False)
+    if always or isatty():
+        colorizer = Colorizer(style, debug)
+        sys.excepthook = colorizer.colorize_traceback
+
+
+class Colorizer(object):
+    def __init__(self, style, debug=False):
+        self.style = style
+        self.debug = debug
+
+    def colorize_traceback(self, type, value, tb):
+        import traceback
+        import pygments.lexers
+
+        tb_text = "".join(traceback.format_exception(type, value, tb))
+        lexer_name = "pytb" if sys.version_info < (3, ) else "py3tb"
+        lexer = pygments.lexers.get_lexer_by_name(lexer_name)
+        tb_colored = pygments.highlight(tb_text, lexer, self.formatter)
+        self.stream.write(tb_colored)
+
+    @property
+    def formatter(self):
+        from pygments.formatters import get_formatter_by_name
+        import pygments.util
+
+        colors = _get_term_color_support()
+        if self.debug:
+            sys.stderr.write("Detected support for %s colors\n" % colors)
+        if colors == 256:
+            fmt_options = {"style": self.style}
+        elif self.style in ("light", "dark"):
+            fmt_options = {"bg": self.style}
+        else:
+            fmt_options = {"bg": "dark"}
+        fmt_alias = "terminal256" if colors == 256 else "terminal"
+        try:
+            return get_formatter_by_name(fmt_alias, **fmt_options)
+        except pygments.util.ClassNotFound as ex:
+            if self.debug:
+                sys.stderr.write(str(ex) + "\n")
+            return get_formatter_by_name(fmt_alias)
+
+    @property
+    def stream(self):
+        try:
+            import colorama
+        except ImportError:
+            return sys.stderr
+        return colorama.AnsiToWin32(sys.stderr)
+
+
+def _get_term_color_support():
+    try:
+        import curses
+    except ImportError:
+        return 16
+    curses.setupterm()
+    return curses.tigetnum("colors")

seleniumbase/core/log_helper.py

@@ -8,6 +8,10 @@
 from seleniumbase.config import settings
 from seleniumbase.fixtures import constants
 
+python3_11_or_newer = False
+if sys.version_info >= (3, 11):
+    python3_11_or_newer = True
+
 
 def log_screenshot(test_logpath, driver, screenshot=None, get=False):
     screenshot_name = settings.SCREENSHOT_NAME
@@ -75,7 +79,7 @@ def get_master_time():
 
 
 def get_browser_version(driver):
-    if sys.version_info >= (3, 11) and hasattr(sb_config, "_browser_version"):
+    if python3_11_or_newer and hasattr(sb_config, "_browser_version"):
         return sb_config._browser_version
     driver_capabilities = driver.capabilities
     if "version" in driver_capabilities: