mirror of
https://github.com/shmilylty/OneForAll.git
synced 2026-08-26 12:57:50 +08:00
增加请求异常处理
This commit is contained in:
@@ -38,7 +38,11 @@ class Github(Search):
|
|||||||
'login': self.email,
|
'login': self.email,
|
||||||
'password': self.password
|
'password': self.password
|
||||||
}
|
}
|
||||||
|
try:
|
||||||
resp = self.session.post(self.post_url, data=post_data)
|
resp = self.session.post(self.post_url, data=post_data)
|
||||||
|
except Exception as e:
|
||||||
|
logger.log('ERROR', e)
|
||||||
|
return False
|
||||||
if resp.status_code != 200:
|
if resp.status_code != 200:
|
||||||
return False
|
return False
|
||||||
match = re.search(r'"user-login" content="(.*?)"', resp.text)
|
match = re.search(r'"user-login" content="(.*?)"', resp.text)
|
||||||
@@ -51,7 +55,11 @@ class Github(Search):
|
|||||||
|
|
||||||
:return: 获取失败返回None,成功返回token
|
:return: 获取失败返回None,成功返回token
|
||||||
"""
|
"""
|
||||||
|
try:
|
||||||
resp = self.session.get(self.login_url)
|
resp = self.session.get(self.login_url)
|
||||||
|
except Exception as e:
|
||||||
|
logger.log('ERROR', e)
|
||||||
|
return None
|
||||||
if resp.status_code != 200:
|
if resp.status_code != 200:
|
||||||
return None
|
return None
|
||||||
match = re.search(
|
match = re.search(
|
||||||
@@ -68,13 +76,17 @@ class Github(Search):
|
|||||||
self.session.proxies = self.get_proxy(self.source)
|
self.session.proxies = self.get_proxy(self.source)
|
||||||
self.session.verify = self.verify
|
self.session.verify = self.verify
|
||||||
if not self.login_github():
|
if not self.login_github():
|
||||||
logger.log('ERROR', f'{self.session}模块登录失败')
|
logger.log('ERROR', f'{self.source}模块登录失败')
|
||||||
return
|
return
|
||||||
page_num = 1
|
page_num = 1
|
||||||
while True:
|
while True:
|
||||||
time.sleep(self.delay)
|
time.sleep(self.delay)
|
||||||
params = {'p': page_num, 'q': f'"{self.domain}"', 'type': 'Code'}
|
params = {'p': page_num, 'q': f'"{self.domain}"', 'type': 'Code'}
|
||||||
|
try:
|
||||||
resp = self.session.get(self.addr, params=params)
|
resp = self.session.get(self.addr, params=params)
|
||||||
|
except Exception as e:
|
||||||
|
logger.log('ERROR', e)
|
||||||
|
break
|
||||||
if resp.status_code != 200:
|
if resp.status_code != 200:
|
||||||
logger.log('ERROR', f'{self.session}模块搜索出错')
|
logger.log('ERROR', f'{self.session}模块搜索出错')
|
||||||
break
|
break
|
||||||
|
|||||||
Reference in New Issue
Block a user