Skip to content

Commit f2ce01f

Browse files
committed
Refactor CDP Mode
1 parent b65b725 commit f2ce01f

File tree

3 files changed

+6
-18
lines changed

3 files changed

+6
-18
lines changed

seleniumbase/undetected/cdp_driver/browser.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -633,7 +633,7 @@ async def start(self=None) -> Browser:
633633
""" % (dashes, message, dashes)
634634
)
635635
self.connection = Connection(
636-
self.info.webSocketDebuggerUrl, _owner=self
636+
self.info.webSocketDebuggerUrl, browser=self
637637
)
638638
if self.config.autodiscover_targets:
639639
logger.info("Enabling autodiscover targets")
@@ -807,7 +807,7 @@ async def update_targets(self):
807807
f"/{t.target_id}"
808808
),
809809
target=t,
810-
_owner=self,
810+
browser=self,
811811
)
812812
)
813813
await asyncio.sleep(0)

seleniumbase/undetected/cdp_driver/cdp_util.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -719,18 +719,6 @@ def start_sync(*args, **kwargs) -> Browser:
719719
loop = kwargs["loop"]
720720
else:
721721
loop = asyncio.new_event_loop()
722-
if "user_data_dir" in kwargs and kwargs["user_data_dir"]:
723-
headless = False
724-
if "headless" in kwargs:
725-
headless = kwargs["headless"]
726-
decoy_args = kwargs
727-
decoy_args["headless"] = True
728-
driver = loop.run_until_complete(start(**decoy_args))
729-
kwargs["headless"] = headless
730-
kwargs["user_data_dir"] = driver.config.user_data_dir
731-
time.sleep(0.2)
732-
driver.stop() # Due to Chrome-130, must stop & start
733-
time.sleep(0.1)
734722
return loop.run_until_complete(start(*args, **kwargs))
735723

736724

seleniumbase/undetected/cdp_driver/connection.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -184,13 +184,13 @@ def __init__(
184184
self,
185185
websocket_url=None,
186186
target=None,
187-
_owner=None,
187+
browser=None,
188188
**kwargs,
189189
):
190190
super().__init__()
191191
self._target = target
192192
self.__count__ = itertools.count(0)
193-
self._owner = _owner
193+
self.browser = browser
194194
self.websocket_url: str = websocket_url
195195
self.websocket = None
196196
self.mapper = {}
@@ -426,8 +426,8 @@ async def send(
426426
await self.aopen()
427427
if not self.websocket or self.websocket.state is State.CLOSED:
428428
return
429-
if self._owner:
430-
browser = self._owner
429+
if self.browser:
430+
browser = self.browser
431431
if browser.config:
432432
if browser.config.expert:
433433
await self._prepare_expert()

0 commit comments

Comments
 (0)