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-04-10 00:24:55 +08:00
parent dd7935b69c
commit 686325b386
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/common/utils.py
+3 -1
View File
@@ -40,7 +40,7 @@ def match_subdomain(domain, text, distinct=True):
:rtype: set or list :rtype: set or list
""" """
regexp = r'(?:[a-z0-9](?:[a-z0-9\-]{0,61}[a-z0-9])?\.){0,}' \ regexp = r'(?:[a-z0-9](?:[a-z0-9\-]{0,61}[a-z0-9])?\.){0,}' \
+ domain.replace('.', r'\.') + domain.replace('.', r'\.') + '$'
result = re.findall(regexp, text, re.I) result = re.findall(regexp, text, re.I)
if not result: if not result:
return set() return set()
@@ -321,6 +321,8 @@ def export_all(format, path, datas):
logger.log('INFOR', f'所有主域的子域结果 {path}') logger.log('INFOR', f'所有主域的子域结果 {path}')
row_list = list() row_list = list()
for row in datas: for row in datas:
row.pop('header')
row.pop('response')
keys = row.keys() keys = row.keys()
values = row.values() values = row.values()
if format in {'xls', 'xlsx'}: if format in {'xls', 'xlsx'}: