From 1f62caa4f4e07e62aa095fb3ae52420b5351b257 Mon Sep 17 00:00:00 2001 From: Jing Ling Date: Sat, 16 Jan 2021 16:33:15 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=8F=AA=E5=90=AF=E7=94=A8?= =?UTF-8?q?=E9=83=A8=E5=88=86=E6=A8=A1=E5=9D=97=E4=B8=8D=E8=83=BD=E4=BD=BF?= =?UTF-8?q?=E7=94=A8=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config/setting.py | 5 ++--- modules/collect.py | 5 +++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/config/setting.py b/config/setting.py index 0c8c7b6..eafc572 100644 --- a/config/setting.py +++ b/config/setting.py @@ -30,10 +30,9 @@ result_save_path = None # 子域结果保存文件路径(默认None) # 收集模块设置 save_module_result = False # 保存各模块发现结果为json文件(默认False) enable_all_module = True # 启用所有收集模块(默认True) -enable_partial_module = [] # 启用部分收集模块 必须禁用enable_all_module才能生效 +enable_partial_module = ['modules.search.baidu'] # 启用部分收集模块 必须禁用enable_all_module才能生效 # 只使用ask和baidu搜索引擎收集子域的示例 -# enable_partial_module = [('modules.search', 'ask') -# ('modules.search', 'baidu')] +# enable_partial_module = ['modules.search.ask', 'modules.search.baidu'] # 爆破模块设置 brute_concurrent_num = 2000 # 爆破时并发查询数量(默认2000，最大推荐10000) diff --git a/modules/collect.py b/modules/collect.py index cec28de..d4112b3 100644 --- a/modules/collect.py +++ b/modules/collect.py @@ -23,7 +23,7 @@ class Collect(object): module_path = settings.module_dir.joinpath(module) for path in module_path.rglob('*.py'): import_module = f'modules.{module}.{path.stem}' - self.modules.append([import_module, path.stem]) + self.modules.append(import_module) else: self.modules = settings.enable_partial_module @@ -31,7 +31,8 @@ class Collect(object): """ Import do function """ - for module, name in self.modules: + for module in self.modules: + name = module.split('.')[-1] import_object = importlib.import_module(module) func = getattr(import_object, 'run') self.collect_funcs.append([func, name])