From 806768b1bf175c407dcf4b91d69fa5bb657759c2 Mon Sep 17 00:00:00 2001 From: frederik Date: Tue, 29 Apr 2025 00:28:59 +0200 Subject: [PATCH] add extra args --- botasaurus_driver/core/config.py | 7 +++++-- botasaurus_driver/driver.py | 2 ++ 2 files changed, 7 insertions(+), 2 deletions(-) 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,