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

get()增加重试功能

Browse Source
This commit is contained in:
g1879 2020-09-01 23:27:11 +08:00
parent 0dcfcf5cd8
commit 37bf43ac36
3 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
DrissionPage/driver_page.py
View File

@ -65,12 +65,12 @@ class DriverPage(object):
times: int = 0, times: int = 0,
interval: float = 1, interval: float = 1,
show_errmsg: bool = False, ): show_errmsg: bool = False, ):
""" """尝试连接，重试若干次 \n
:param to_url: 要访问的url :param to_url: 要访问的url
:param times: 重试次数 :param times: 重试次数
:param interval: 重试间隔（秒） :param interval: 重试间隔（秒）
:param show_errmsg: 是否抛出异常 :param show_errmsg: 是否抛出异常
:return: :return: 是否成功
""" """
self.driver.get(to_url) self.driver.get(to_url)
is_ok = self.check_page() is_ok = self.check_page()

20
DrissionPage/mix_page.py
View File

@ -235,12 +235,30 @@ class MixPage(Null, SessionPage, DriverPage):
return super().chrome_downloading(path) return super().chrome_downloading(path)
# ----------------以下为共用函数----------------------- # ----------------以下为共用函数-----------------------
def _try_to_get(self,
to_url: str,
times: int = 0,
interval: float = 1,
show_errmsg: bool = False,
**kwargs):
"""尝试连接，重试若干次 \n
:param to_url: 要访问的url
:param times: 重试次数
:param interval: 重试间隔（秒）
:param show_errmsg: 是否抛出异常
:param kwargs: 连接参数
:return: s模式为HTMLResponse对象，d模式为bool
"""
if self._mode == 'd':
return super(SessionPage, self)._try_to_get(to_url, times, interval, show_errmsg)
elif self._mode == 's':
return super()._try_to_get(to_url, times, interval, show_errmsg, **kwargs)
def get(self, def get(self,
url: str, url: str,
go_anyway=False, go_anyway=False,
show_errmsg: bool = False, show_errmsg: bool = False,
retry: int = 0, retry: int = 2,
interval: float = 1, interval: float = 1,
**kwargs) -> Union[bool, None]: **kwargs) -> Union[bool, None]:
"""跳转到一个url \n """跳转到一个url \n

2
DrissionPage/session_page.py
View File

@ -148,7 +148,7 @@ class SessionPage(object):
interval: float = 1, interval: float = 1,
show_errmsg: bool = False, show_errmsg: bool = False,
**kwargs) -> HTMLResponse: **kwargs) -> HTMLResponse:
"""尝试连接，重试若干次 """尝试连接，重试若干次 \n
:param to_url: 要访问的url :param to_url: 要访问的url
:param times: 重试次数 :param times: 重试次数
:param interval: 重试间隔（秒） :param interval: 重试间隔（秒）