diff --git a/botasaurus_driver/core/config.py b/botasaurus_driver/core/config.py index a98ecda..536a9ef 100644 --- a/botasaurus_driver/core/config.py +++ b/botasaurus_driver/core/config.py @@ -135,6 +135,7 @@ def __init__( self, headless=False, enable_xvfb_virtual_display=False, + xvfb_extra_args=None, proxy=None, profile=None, tiny_profile=False, @@ -151,12 +152,14 @@ def __init__( host="127.0.0.1", port=None, ): + if xvfb_extra_args is None: + xvfb_extra_args = [] if tiny_profile and profile is None: raise ValueError("Profile must be given when using tiny profile") if enable_xvfb_virtual_display and headless: raise ValueError("Xvfb Virtual Display cannot be used while headless mode is enabled") - + self.xvfb_extra_args = xvfb_extra_args self.headless = headless self.proxy = proxy self.enable_xvfb_virtual_display = enable_xvfb_virtual_display # New attribute @@ -256,7 +259,7 @@ def __call__(self): from pyvirtualdisplay import Display try: - self._display = Display(visible=False, size=(1920, 1080)) + self._display = Display(visible=False, size=(1920, 1080), extra_args=self.xvfb_extra_args) self._display.start() except FileNotFoundError: print( diff --git a/botasaurus_driver/driver.py b/botasaurus_driver/driver.py index 91c8ea7..645f453 100644 --- a/botasaurus_driver/driver.py +++ b/botasaurus_driver/driver.py @@ -2065,6 +2065,7 @@ def __init__( headless=False, enable_xvfb_virtual_display=False, proxy=None, + xvfb_extra_args=None, profile=None, tiny_profile=False, block_images=False, @@ -2083,6 +2084,7 @@ def __init__( self.config = Config( headless=headless, enable_xvfb_virtual_display=enable_xvfb_virtual_display, + xvfb_extra_args=xvfb_extra_args, proxy=proxy, profile=profile, tiny_profile=tiny_profile,