river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-25 20:37:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化

Browse Source
This commit is contained in:
Jing Ling
2020-09-16 00:28:23 +08:00
parent 899e243a3a
commit a1631bf6fc
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/search/zoomeye_api.py
+5 -4
View File
@@ -23,16 +23,15 @@ class ZoomEyeAPI(Search):
data = {'username': self.user, 'password': self.pwd} data = {'username': self.user, 'password': self.pwd}
resp = self.post(url=url, json=data) resp = self.post(url=url, json=data)
if not resp: if not resp:
logger.log('FATAL', f'{self.source} module login failed,' logger.log('ALERT', f'{self.source} module login failed')
f' can not get access token') return None
exit(1)
data = resp.json() data = resp.json()
if resp.status_code == 200: if resp.status_code == 200:
logger.log('DEBUG', f'{self.source} module login success') logger.log('DEBUG', f'{self.source} module login success')
return data.get('access_token') return data.get('access_token')
else: else:
logger.log('ALERT', data.get('message')) logger.log('ALERT', data.get('message'))
exit(1) return None
def search(self): def search(self):
""" """
@@ -40,6 +39,8 @@ class ZoomEyeAPI(Search):
""" """
page_num = 1 page_num = 1
access_token = self.login() access_token = self.login()
if not access_token:
return
while True: while True:
time.sleep(self.delay) time.sleep(self.delay)
self.header = self.get_header() self.header = self.get_header()