mirror of
https://github.com/shmilylty/OneForAll.git
synced 2026-08-26 04:47:48 +08:00
set使用update
This commit is contained in:
@@ -30,7 +30,7 @@ class Ask(Search):
|
||||
subdomains = self.match_subdomains(resp, fuzzy=False)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
self.page_num += 1
|
||||
if '>Next<' not in resp.text:
|
||||
break
|
||||
|
||||
@@ -25,7 +25,7 @@ class Baidu(Search):
|
||||
for find_res in bs.find_all('a', {'class': 'c-showurl'}):
|
||||
url = find_res.get('href')
|
||||
subdomains = self.match_location(url)
|
||||
subdomains_all = subdomains_all.union(subdomains)
|
||||
subdomains_all.update(subdomains)
|
||||
return subdomains_all
|
||||
|
||||
def search(self, domain, filtered_subdomain=''):
|
||||
@@ -54,7 +54,7 @@ class Baidu(Search):
|
||||
subdomains = self.match_subdomains(resp, fuzzy=False)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
self.page_num += self.per_page_num
|
||||
# 搜索页面没有出现下一页时停止搜索
|
||||
if '&pn={next_pn}&'.format(next_pn=self.page_num) not in resp.text:
|
||||
|
||||
@@ -36,7 +36,7 @@ class Bing(Search):
|
||||
subdomains = self.match_subdomains(resp, fuzzy=False)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
# 搜索页面没有出现下一页时停止搜索
|
||||
if '<div class="sw_next">' not in resp.text:
|
||||
break
|
||||
|
||||
@@ -36,7 +36,7 @@ class BingAPI(Search):
|
||||
subdomains = self.match_subdomains(resp)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
self.page_num += self.per_page_num
|
||||
if self.page_num >= self.limit_num: # 搜索条数限制
|
||||
break
|
||||
|
||||
@@ -40,7 +40,7 @@ class FoFa(Search):
|
||||
subdomains = self.match_subdomains(resp)
|
||||
if not subdomains: # 搜索没有发现子域名则停止搜索
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
size = resp_json.get('size')
|
||||
if size < 10000:
|
||||
break
|
||||
|
||||
@@ -37,7 +37,7 @@ class Gitee(Search):
|
||||
subdomains = self.match_subdomains(soup, fuzzy=False)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
if '<li class="disabled"><a href="###">' in resp.text:
|
||||
break
|
||||
page_num += 1
|
||||
|
||||
@@ -63,7 +63,7 @@ class GithubAPI(Search):
|
||||
subdomains = self.match_subdomains(resp)
|
||||
if not subdomains:
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
page += 1
|
||||
try:
|
||||
resp_json = resp.json()
|
||||
|
||||
@@ -40,7 +40,7 @@ class Google(Search):
|
||||
subdomains = self.match_subdomains(resp, fuzzy=False)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
page_num += per_page_num
|
||||
if 'start=' + str(page_num) not in resp.text:
|
||||
break
|
||||
|
||||
@@ -35,7 +35,7 @@ class GoogleAPI(Search):
|
||||
subdomains = self.match_subdomains(resp)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
self.page_num += self.per_page_num
|
||||
if self.page_num > 100: # 免费的API只能查询前100条结果
|
||||
break
|
||||
|
||||
@@ -26,7 +26,7 @@ class ShodanAPI(Search):
|
||||
subdomains = self.match_subdomains(resp)
|
||||
if not subdomains: # 搜索没有发现子域名则停止搜索
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
page += 1
|
||||
|
||||
def run(self):
|
||||
|
||||
@@ -31,7 +31,7 @@ class So(Search):
|
||||
subdomains = self.match_subdomains(resp, fuzzy=False)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
page_num += 1
|
||||
# 搜索页面没有出现下一页时停止搜索
|
||||
if '<a id="snext"' not in resp.text:
|
||||
|
||||
@@ -28,7 +28,7 @@ class Sogou(Search):
|
||||
subdomains = self.match_subdomains(resp, fuzzy=False)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
self.page_num += 1
|
||||
# 搜索页面没有出现下一页时停止搜索
|
||||
if '<a id="sogou_next"' not in resp.text:
|
||||
|
||||
@@ -38,7 +38,7 @@ class Yahoo(Search):
|
||||
subdomains = self.match_subdomains(text, fuzzy=False)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
if '>Next</a>' not in resp.text: # 搜索页面没有出现下一页时停止搜索
|
||||
break
|
||||
self.page_num += self.per_page_num
|
||||
|
||||
@@ -36,7 +36,7 @@ class Yandex(Search):
|
||||
subdomains = self.match_subdomains(resp, fuzzy=False)
|
||||
if not self.check_subdomains(subdomains):
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
if '>next</a>' not in resp.text: # 搜索页面没有出现下一页时停止搜索
|
||||
break
|
||||
self.page_num += 1
|
||||
|
||||
@@ -50,7 +50,7 @@ class ZoomEyeAPI(Search):
|
||||
subdomains = self.match_subdomains(resp)
|
||||
if not subdomains: # 搜索没有发现子域名则停止搜索
|
||||
break
|
||||
self.subdomains = self.subdomains.union(subdomains)
|
||||
self.subdomains.update(subdomains)
|
||||
page_num += 1
|
||||
if page_num > 500:
|
||||
break
|
||||
|
||||
Reference in New Issue
Block a user