river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 12:57:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

处理resp.json()异常

Browse Source
This commit is contained in:
shmilylty
2019-09-16 16:41:48 +08:00
parent a9ff58061e
commit f7c03656bf
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/modules/datasets/ipv4info.py
+8 -3
View File
@@ -1,5 +1,6 @@
import config import config
from common.query import Query from common.query import Query
from config import logger
class IPv4InfoAPI(Query): class IPv4InfoAPI(Query):
@@ -26,14 +27,18 @@ class IPv4InfoAPI(Query):
return return
if resp.status_code != 200: if resp.status_code != 200:
break # 请求不正常通常网络是有问题，不再继续请求下去 break # 请求不正常通常网络是有问题，不再继续请求下去
data = resp.json() try:
subdomains = self.match(self.domain, str(data)) json = resp.json()
except Exception as e:
logger.log('DEBUG', e.args)
break
subdomains = self.match(self.domain, str(json))
if not subdomains: if not subdomains:
break break
# 合并搜索子域名搜索结果 # 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains) self.subdomains = self.subdomains.union(subdomains)
# 不直接使用subdomains是因为可能里面会出现不符合标准的子域名 # 不直接使用subdomains是因为可能里面会出现不符合标准的子域名
subdomains = data.get('Subdomains') subdomains = json.get('Subdomains')
if subdomains: if subdomains:
# ipv4info子域查询接口每次最多返回300个 用来判断是否还有下一页 # ipv4info子域查询接口每次最多返回300个 用来判断是否还有下一页
if len(subdomains) < 300: if len(subdomains) < 300: