Merge pull request #65 from JrDw0/patch-2

优化
This commit is contained in:
Jing Ling
2020-04-10 02:45:05 +08:00
committed by GitHub
+3 -2
View File
@@ -1,6 +1,7 @@
import requests import requests
import api import api
import json import json
from common.utils import match_subdomain
from common.search import Search from common.search import Search
from config import logger from config import logger
@@ -61,7 +62,7 @@ class GithubAPI(Search):
if resp.status_code != 200: if resp.status_code != 200:
logger.log('ERROR', f'{self.source}模块搜索出错') logger.log('ERROR', f'{self.source}模块搜索出错')
break break
subdomains = self.match(self.domain, resp.text) subdomains = match_subdomain(self.domain, resp.text)
if not subdomains: if not subdomains:
break break
self.subdomains = self.subdomains.union(subdomains) self.subdomains = self.subdomains.union(subdomains)
@@ -72,7 +73,7 @@ class GithubAPI(Search):
logger.log('ERROR', e.args) logger.log('ERROR', e.args)
break break
total_count = resp_json.get('total_count') total_count = resp_json.get('total_count')
if isinstance(total_count, int): if not isinstance(total_count, int):
break break
if page * 100 > total_count: if page * 100 > total_count:
break break