Include my own debugger, "pdbp" (Pdb+), with SeleniumBase

Author: mdmintz

seleniumbase/__init__.py

@@ -1,6 +1,9 @@
 import collections
 import pdb
-import shutil
+try:
+    import pdbp  # (Pdb+) --- Python Debugger Plus
+except Exception:
+    pass
 import sys
 from selenium import webdriver
 from seleniumbase.__version__ import __version__
@@ -16,20 +19,20 @@
 from seleniumbase.plugins.driver_manager import Driver  # noqa
 from seleniumbase.plugins.driver_manager import DriverContext  # noqa
 
-if hasattr(pdb, "DefaultConfig"):
-    # Only load pdbpp configuration if pdbpp is installed
-    pdb.DefaultConfig.filename_color = pdb.Color.blue
-    pdb.DefaultConfig.line_number_color = pdb.Color.turquoise
-    pdb.DefaultConfig.show_hidden_frames_count = False
-    pdb.DefaultConfig.disable_pytest_capturing = True
-    pdb.DefaultConfig.enable_hidden_frames = False
-    pdb.DefaultConfig.truncate_long_lines = True
-    pdb.DefaultConfig.sticky_by_default = True
-    # Fix spacing for line numbers > 9999
-    pdb.Pdb.get_terminal_size = lambda x: (
-        shutil.get_terminal_size()[0] - 1,
-        shutil.get_terminal_size()[1],
-    )
+if sys.version_info[0] < 3 and "pdbp" in locals():
+    # With Python3, "import pdbp" is all you need
+    for key in pdbp.__dict__.keys():
+        # Replace pdb with pdbp
+        pdb.__dict__[key] = pdbp.__dict__[key]
+    if hasattr(pdb, "DefaultConfig"):
+        # Here's how to customize Pdb+ options
+        pdb.DefaultConfig.filename_color = pdb.Color.blue
+        pdb.DefaultConfig.line_number_color = pdb.Color.turquoise
+        pdb.DefaultConfig.show_hidden_frames_count = False
+        pdb.DefaultConfig.disable_pytest_capturing = True
+        pdb.DefaultConfig.enable_hidden_frames = False
+        pdb.DefaultConfig.truncate_long_lines = True
+        pdb.DefaultConfig.sticky_by_default = True
 if sys.version_info[0] >= 3:
     from seleniumbase import translate  # noqa
 if sys.version_info >= (3, 7):