Fix PyAutoGUI targeting issue on Linux

mdmintz · mdmintz · commit f46cca90af3b · 2025-07-01T18:00:52.000-04:00
diff --git a/seleniumbase/core/browser_launcher.py b/seleniumbase/core/browser_launcher.py
@@ -580,13 +580,16 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
     headless = False
     headed = None
     xvfb = None
+    xvfb_metrics = None
     binary_location = None
     if hasattr(sb_config, "headless"):
         headless = sb_config.headless
     if hasattr(sb_config, "headed"):
         headed = sb_config.headed
     if hasattr(sb_config, "xvfb"):
         xvfb = sb_config.xvfb
+    if hasattr(sb_config, "xvfb_metrics"):
+        xvfb_metrics = sb_config.xvfb_metrics
     if hasattr(sb_config, "binary_location"):
         binary_location = sb_config.binary_location
 
@@ -599,6 +602,7 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
             headless=headless,
             headed=headed,
             xvfb=xvfb,
+            xvfb_metrics=xvfb_metrics,
             browser_executable_path=binary_location,
         )
     )
@@ -963,6 +967,11 @@ def __install_pyautogui_if_missing():
                         backend="xvfb",
                         use_xauth=True,
                     )
+                    if "--debug-display" in sys.argv:
+                        print(
+                            "Starting VDisplay from browser_launcher: (%s, %s)"
+                            % (xvfb_width, xvfb_height)
+                        )
                     _xvfb_display.start()
                     sb_config._virtual_display = _xvfb_display
                     sb_config.headless_active = True
diff --git a/seleniumbase/core/sb_cdp.py b/seleniumbase/core/sb_cdp.py
@@ -1439,6 +1439,10 @@ def __install_pyautogui_if_missing(self):
                         shared_utils.is_linux()
                         and (not sb_config.headed or sb_config.xvfb)
                         and not driver.config.headless
+                        and (
+                            not hasattr(sb_config, "_virtual_display")
+                            or not sb_config._virtual_display
+                        )
                     ):
                         from sbvirtualdisplay import Display
                         xvfb_width = 1366
@@ -1466,6 +1470,11 @@ def __install_pyautogui_if_missing(self):
                                 backend="xvfb",
                                 use_xauth=True,
                             )
+                            if "--debug-display" in sys.argv:
+                                print(
+                                    "Starting VDisplay from sb_cdp: (%s, %s)"
+                                    % (xvfb_width, xvfb_height)
+                                )
                             xvfb_display.start()
 
     def __get_configured_pyautogui(self, pyautogui_copy):
diff --git a/seleniumbase/fixtures/base_case.py b/seleniumbase/fixtures/base_case.py
@@ -14164,7 +14164,13 @@ def __activate_virtual_display(self):
                     backend="xvfb",
                     use_xauth=True,
                 )
+                if "--debug-display" in sys.argv:
+                    print(
+                        "Starting VDisplay from base_case: (%s, %s)"
+                        % (self._xvfb_width, self._xvfb_height)
+                    )
                 self._xvfb_display.start()
+                sb_config._virtual_display = self._xvfb_display
                 if "DISPLAY" not in os.environ.keys():
                     print(
                         "\nX11 display failed! Will use regular xvfb!"
diff --git a/seleniumbase/undetected/cdp_driver/cdp_util.py b/seleniumbase/undetected/cdp_driver/cdp_util.py
@@ -51,7 +51,14 @@ def __activate_virtual_display_as_needed(
     headless, headed, xvfb, xvfb_metrics
 ):
     """This is only needed on Linux."""
-    if IS_LINUX and (not headed or xvfb):
+    if (
+        IS_LINUX
+        and (not headed or xvfb)
+        and (
+            not hasattr(sb_config, "_virtual_display")
+            or not sb_config._virtual_display
+        )
+    ):
         from sbvirtualdisplay import Display
         pip_find_lock = fasteners.InterProcessLock(
             constants.PipInstall.FINDLOCK
@@ -87,6 +94,11 @@ def __activate_virtual_display_as_needed(
                         backend="xvfb",
                         use_xauth=True,
                     )
+                    if "--debug-display" in sys.argv:
+                        print(
+                            "Starting VDisplay from cdp_util: (%s, %s)"
+                            % (_xvfb_width, _xvfb_height)
+                        )
                     _xvfb_display.start()
                     if "DISPLAY" not in os.environ.keys():
                         print(
