river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-25 20:37:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

模块优化

Browse Source
This commit is contained in:
Jing Ling
2020-08-10 18:19:11 +08:00
parent 4f462fcd60
commit 8a035c76ca
42 changed files with 49 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files
common/module.py
+13 -2
View File
@@ -208,8 +208,19 @@ class Module(object):
logger.log('TRACE', f'{module} module does not use proxy')
return self.proxy
def match_subdomains(self, html, distinct=True, fuzzy=True):
return utils.match_subdomains(self.domain, html, distinct, fuzzy)
def match_subdomains(self, resp, distinct=True, fuzzy=True):
if not resp:
return set()
elif isinstance(resp, str):
return utils.match_subdomains(self.domain, resp, distinct, fuzzy)
elif hasattr(resp, 'text'):
return utils.match_subdomains(self.domain, resp.text, distinct, fuzzy)
else:
return set()
def collect_subdomains(self, resp):
subdomains = self.match_subdomains(resp)
return self.subdomains.union(subdomains)
def save_json(self):
"""