修复#23问题

This commit is contained in:
Jing Ling
2019-09-26 22:14:08 +08:00
parent 6ef7c96a93
commit 0a049e991d
+5 -5
View File
@@ -24,19 +24,19 @@ class CensysAPI(Query):
data = { data = {
'query': f'parsed.names: {self.domain}', 'query': f'parsed.names: {self.domain}',
'page': 1, 'page': 1,
'fields': ['parsed.subject_dn'], 'fields': ['parsed.subject_dn', 'parsed.names'],
'flatten': True} 'flatten': True}
resp = self.post(self.addr, json=data, auth=(self.id, self.secret)) resp = self.post(self.addr, json=data, auth=(self.id, self.secret))
if not resp: if not resp:
return return
data = resp.json() json = resp.json()
status = data.get('status') status = json.get('status')
if status != 'ok': if status != 'ok':
logger.log('ALERT', status) logger.log('ALERT', status)
return return
subdomains = self.match(self.domain, str(data)) subdomains = self.match(self.domain, str(json))
self.subdomains = self.subdomains.union(subdomains) self.subdomains = self.subdomains.union(subdomains)
pages = data.get('metadata').get('pages') pages = json.get('metadata').get('pages')
for page in range(2, pages + 1): for page in range(2, pages + 1):
time.sleep(self.delay) time.sleep(self.delay)
data['page'] = page data['page'] = page