blackteay/DrissionPage
1
0
Fork 0
mirror of https://gitee.com/g1879/DrissionPage.git synced 2024-12-10 04:00:23 +08:00
Code Issues Projects Releases Wiki Activity

继续修改抓包功能，未完成

Browse Source
This commit is contained in:
g1879 2023-04-26 19:01:00 +08:00
parent 283bbfe06f
commit f817fcee5c
3 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
DrissionPage/chromium_base.py
View File

@ -1063,6 +1063,7 @@ class NetworkListener(object):
self._count = None self._count = None
self._caught = 0 # 已获取到的数量 self._caught = 0 # 已获取到的数量
self._all_tabs = False # 是否监听所有tab
def set_targets(self, targets, is_regex=False, count=None): def set_targets(self, targets, is_regex=False, count=None):
"""指定要等待的数据包 """指定要等待的数据包
@ -1092,12 +1093,23 @@ class NetworkListener(object):
else: else:
self.stop() self.stop()
def start(self):
driver = self._page.browser_driver if self._all_tabs else self._page.driver
driver.set_listener('Fetch.requestPaused', self._request_paused)
patterns = []
for i in self._targets:
patterns.append({'requestStage': 'Request', 'urlPattern': i})
patterns.append({'requestStage': 'Response', 'urlPattern': i})
if patterns:
driver.call_method('Fetch.enable', patterns=patterns)
else:
driver.call_method('Fetch.enable')
def stop(self): def stop(self):
"""停止监听数据包""" """停止监听数据包"""
self._page.run_cdp('Network.disable') driver = self._page.browser_driver if self._all_tabs else self._page.driver
self._page.driver.Network.requestWillBeSent = None driver.set_listener('Fetch.requestPaused', None)
self._page.driver.Network.responseReceived = None driver.call_method('Fetch.disable')
self._page.driver.Network.loadingFinished = None
def listen(self, timeout=None, any_one=False): def listen(self, timeout=None, any_one=False):
"""等待指定数据包加载完成 """等待指定数据包加载完成
@ -1126,6 +1138,9 @@ class NetworkListener(object):
self._caught = 0 self._caught = 0
return r return r
def _request_paused(self, **kwargs):
pass
def _request_will_sent(self, **kwargs): def _request_will_sent(self, **kwargs):
"""接收到请求时的回调函数""" """接收到请求时的回调函数"""
for target in self._targets: for target in self._targets:

2
DrissionPage/chromium_driver.pyi
View File

@ -53,7 +53,7 @@ class ChromiumDriver(object):
def stop(self) -> bool: ... def stop(self) -> bool: ...
def set_listener(self, event: str, callback: Callable) -> Union[Callable, None, bool]: ... def set_listener(self, event: str, callback: Union[Callable, None]) -> Union[Callable, None, bool]: ...
def get_listener(self, event: str) -> Union[Callable, None]: ... def get_listener(self, event: str) -> Union[Callable, None]: ...

4
DrissionPage/web_page.pyi
View File

@ -26,8 +26,8 @@ class WebPage(SessionPage, ChromiumPage, BasePage):
def __init__(self, def __init__(self,
mode: str = 'd', mode: str = 'd',
timeout: float = None, timeout: float = None,
driver_or_options: Union[ChromiumDriver, ChromiumOptions, DriverOptions] = None, driver_or_options: Union[ChromiumDriver, ChromiumOptions, DriverOptions, bool] = None,
session_or_options: Union[Session, SessionOptions] = None) -> None: session_or_options: Union[Session, SessionOptions, bool] = None) -> None:
self._mode: str = ... self._mode: str = ...
self._has_driver: bool = ... self._has_driver: bool = ...
self._has_session: bool = ... self._has_session: bool = ...