Pre Merge pull request !51 from 有三/dev

This commit is contained in:
有三 2024-07-22 18:24:28 +00:00 committed by Gitee
commit f4aa448b11
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F

View File

@ -22,6 +22,7 @@ from .._functions.settings import Settings
from .._functions.tools import PortFinder
from .._functions.tools import raise_error
from .._pages.chromium_base import Timeout
from .._pages.chromium_frame import ChromiumFrame
from .._pages.tabs import ChromiumTab, MixTab
from .._units.downloader import DownloadManager
from .._units.setter import BrowserSetter
@ -457,6 +458,15 @@ class Chromium(object):
:param owner: 使用该驱动的对象
:return: Driver对象
"""
if isinstance(owner, ChromiumFrame):
d = self._drivers.get(tab_id, None)
if not d:
d = Driver(tab_id, 'page', self.address)
d.owner = owner
self._drivers[tab_id] = d
self._all_drivers.setdefault(tab_id, set()).add(d)
return d
d = self._drivers.pop(tab_id, None)
if not d:
d = Driver(tab_id, 'page', self.address)