From a1631bf6fc04ed706fa13e075eff9705ea2fdbaf Mon Sep 17 00:00:00 2001 From: Jing Ling Date: Wed, 16 Sep 2020 00:28:23 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- modules/search/zoomeye_api.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/modules/search/zoomeye_api.py b/modules/search/zoomeye_api.py index d63b93e..afef3ee 100644 --- a/modules/search/zoomeye_api.py +++ b/modules/search/zoomeye_api.py @@ -23,16 +23,15 @@ class ZoomEyeAPI(Search): data = {'username': self.user, 'password': self.pwd} resp = self.post(url=url, json=data) if not resp: - logger.log('FATAL', f'{self.source} module login failed,' - f' can not get access token') - exit(1) + logger.log('ALERT', f'{self.source} module login failed') + return None data = resp.json() if resp.status_code == 200: logger.log('DEBUG', f'{self.source} module login success') return data.get('access_token') else: logger.log('ALERT', data.get('message')) - exit(1) + return None def search(self): """ @@ -40,6 +39,8 @@ class ZoomEyeAPI(Search): """ page_num = 1 access_token = self.login() + if not access_token: + return while True: time.sleep(self.delay) self.header = self.get_header()