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

优化

Browse Source
This commit is contained in:
Jing Ling
2020-08-01 13:36:03 +08:00
parent 368969ccc7
commit ded438f750
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/search/baidu.py
+3 -4
View File
@@ -13,11 +13,10 @@ class Baidu(Search):
self.domain = domain self.domain = domain
self.limit_num = 750 # 限制搜索条数 self.limit_num = 750 # 限制搜索条数
def redirect_match(self, domain, html): def redirect_match(self, html):
""" """
获取跳转地址并传递地址进行跳转head请求 获取跳转地址并传递地址进行跳转head请求
:param domain: 域名
:param html: 响应体 :param html: 响应体
:return: 子域 :return: 子域
""" """
@@ -52,7 +51,7 @@ class Baidu(Search):
return return
if len(domain) > 12: # 解决百度搜索结果中域名过长会显示不全的问题 if len(domain) > 12: # 解决百度搜索结果中域名过长会显示不全的问题
# 获取百度跳转URL响应头的Location字段获取直链 # 获取百度跳转URL响应头的Location字段获取直链
subdomains = self.redirect_match(domain, resp.text) subdomains = self.redirect_match(resp.text)
else: else:
subdomains = self.match_subdomains(resp.text, fuzzy=False) subdomains = self.match_subdomains(resp.text, fuzzy=False)
if not subdomains: # 搜索没有发现子域名则停止搜索 if not subdomains: # 搜索没有发现子域名则停止搜索
@@ -106,4 +105,4 @@ def do(domain): # 统一入口名字 方便多线程调用
if __name__ == '__main__': if __name__ == '__main__':
do('huayunshuzi.com') do('mi.com')