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

pep8

Browse Source
This commit is contained in:
shmilylty
2019-08-16 19:13:19 +08:00
parent 8d46ce0862
commit f52768632a
50 changed files with 161 additions and 162 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/modules/datasets/binaryedge_api.py
+2 -2
View File
@@ -24,8 +24,8 @@ class BinaryEdgeAPI(Query):
resp = self.get(url)
if not resp:
return
subdomains_find = self.match(self.domain, str(resp.json()))
self.subdomains = self.subdomains.union(subdomains_find)
subdomains = self.match(self.domain, str(resp.json()))
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/bufferover.py
+2 -2
View File
@@ -21,9 +21,9 @@ class BufferOver(Query):
resp = self.get(self.addr, params)
if not resp:
return
subdomains_find = self.match(self.domain, resp.text)
subdomains = self.match(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/chinaz.py
+2 -2
View File
@@ -21,9 +21,9 @@ class Chinaz(Query):
resp = self.get(self.addr)
if not resp:
return
subdomains_find = self.match(self.domain, resp.text)
subdomains = self.match(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/chinaz_api.py
+2 -2
View File
@@ -23,9 +23,9 @@ class ChinazAPI(Query):
resp = self.get(self.addr, params)
if not resp:
return
subdomains_find = self.match(self.domain, str(resp.json()))
subdomains = self.match(self.domain, str(resp.json()))
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/circl_api.py
+2 -2
View File
@@ -23,9 +23,9 @@ class CirclAPI(Query):
resp = self.get(self.addr + self.domain, auth=(self.user, self.pwd))
if not resp:
return
subdomains_find = self.match(self.domain, str(resp.json()))
subdomains = self.match(self.domain, str(resp.json()))
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/dnsdb.py
+4 -4
View File
@@ -54,13 +54,13 @@ class DNSdb(Query):
resp = self.get(url)
if not resp:
return
subdomains_find = self.match(self.domain, resp.text)
subdomains = self.match(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
else:
subdomains_find = self.match(self.domain, resp.text)
subdomains = self.match(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/dnsdb_api.py
+2 -2
View File
@@ -26,9 +26,9 @@ class DNSdbAPI(Query):
if not resp:
return
if resp.status_code == 200:
subdomains_find = utils.match_subdomain(self.domain, resp.text)
subdomains = utils.match_subdomain(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/dnsdumpster.py
+3 -3
View File
@@ -28,10 +28,10 @@ class DNSdumpster(Query):
resp = self.post(self.addr, data)
if not resp:
return
subdomains_find = utils.match_subdomain(self.domain, resp.text)
if subdomains_find:
subdomains = utils.match_subdomain(self.domain, resp.text)
if subdomains:
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/hackertarget.py
+3 -3
View File
@@ -21,10 +21,10 @@ class HackerTarget(Query):
if not resp:
return
if resp.status_code == 200:
subdomains_find = utils.match_subdomain(self.domain, resp.text)
if subdomains_find:
subdomains = utils.match_subdomain(self.domain, resp.text)
if subdomains:
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/ipv4info.py
+5 -5
View File
@@ -26,14 +26,14 @@ class IPv4InfoAPI(Query):
return
if resp.status_code != 200:
break # 请求不正常通常网络是有问题，不再继续请求下去
resp_json = resp.json()
subdomains_find = self.match(self.domain, str(resp_json))
if not subdomains_find:
data = resp.json()
subdomains = self.match(self.domain, str(data))
if not subdomains:
break
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
# 不直接使用subdomains是因为可能里面会出现不符合标准的子域名
subdomains = resp_json.get('Subdomains')
subdomains = data.get('Subdomains')
if subdomains:
# ipv4info子域查询接口每次最多返回300个 用来判断是否还有下一页
if len(subdomains) < 300:
oneforall/modules/datasets/netcraft.py
+3 -3
View File
@@ -46,11 +46,11 @@ class NetCraft(Query):
resp = self.get(self.addr + last, params)
if not resp:
return
subdomains_find = self.match(self.domain, resp.text)
if not subdomains_find: # 搜索没有发现子域名则停止搜索
subdomains = self.match(self.domain, resp.text)
if not subdomains: # 搜索没有发现子域名则停止搜索
break
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
if 'Next page' not in resp.text: # 搜索页面没有出现下一页时停止搜索
break
last = re.search(r'&last=.*' + self.domain, resp.text).group(0)
oneforall/modules/datasets/ptrarchive.py
+3 -3
View File
@@ -25,10 +25,10 @@ class PTRArchive(Query):
if not resp:
return
if resp.status_code == 200:
subdomains_find = utils.match_subdomain(self.domain, resp.text)
if subdomains_find:
subdomains = utils.match_subdomain(self.domain, resp.text)
if subdomains:
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/riddler.py
+2 -2
View File
@@ -21,9 +21,9 @@ class Riddler(Query):
resp = self.get(self.addr, params)
if not resp:
return
subdomains_find = self.match(self.domain, resp.text)
subdomains = self.match(self.domain, resp.text)
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/robtex.py
+3 -3
View File
@@ -32,10 +32,10 @@ class Robtex(Query):
resp = self.get(url)
if not resp:
return
subdomains_find = self.match(self.domain, resp.text)
if subdomains_find:
subdomains = self.match(self.domain, resp.text)
if subdomains:
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/securitytrails_api.py
+3 -3
View File
@@ -26,10 +26,10 @@ class SecurityTrailsAPI(Query):
if not resp:
return
prefixs = resp.json()['subdomains']
subdomains_find = [f'{prefix}.{self.domain}' for prefix in prefixs]
if subdomains_find:
subdomains = [f'{prefix}.{self.domain}' for prefix in prefixs]
if subdomains:
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""
oneforall/modules/datasets/sitedossier.py
+3 -3
View File
@@ -26,11 +26,11 @@ class SiteDossier(Query):
resp = self.get(url)
if not resp:
return
subdomains_find = self.match(self.domain, resp.text)
if not subdomains_find: # 搜索没有发现子域名则停止搜索
subdomains = self.match(self.domain, resp.text)
if not subdomains: # 搜索没有发现子域名则停止搜索
break
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
# 搜索页面没有出现下一页时停止搜索
if 'Show next 100 items' not in resp.text:
break
oneforall/modules/datasets/threatcrowd.py
+2 -2
View File
@@ -25,9 +25,9 @@ class ThreatCrowd(Query):
return
if not resp:
return
subdomains_find = self.match(self.domain, str(resp.json()))
subdomains = self.match(self.domain, str(resp.json()))
# 合并搜索子域名搜索结果
self.subdomains = self.subdomains.union(subdomains_find)
self.subdomains = self.subdomains.union(subdomains)
def run(self):
"""