From a7b0def3c2b74f24fbd35fe8caf09e3ac7edb4df Mon Sep 17 00:00:00 2001
From: Jing Ling <admin@hackfun.org>
Date: Wed, 12 Feb 2020 10:44:15 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E8=B7=AF=E5=BE=84=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 oneforall/common/utils.py | 12 ++++++------
 oneforall/takeover.py     | 12 ++++++------
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/oneforall/common/utils.py b/oneforall/common/utils.py
index 4a3d9d5..6503b0b 100644
--- a/oneforall/common/utils.py
+++ b/oneforall/common/utils.py
@@ -147,16 +147,16 @@ def get_semaphore():
         return 800
 
 
-def check_path(path, table, format):
+def check_path(path, name, format):
     """
     检查结果输出目录路径
 
     :param path: 保存路径
-    :param table: 导出表名
+    :param name: 导出名字
     :param format: 保存格式
     :return: 目录路径
     """
-    default_path = config.result_save_path.joinpath(f'{table}.{format}')
+    default_path = config.result_save_path.joinpath(f'{name}.{format}')
     if path is None:
         path = default_path
     try:
@@ -296,9 +296,9 @@ def export_all(format, datas):
     :param datas: 待导出的结果数据
     """
     format = check_format(format, len(datas))
-    dpath = check_dpath()
     timestamp = get_timestamp()
-    fpath = dpath.joinpath(f'all_subdomain_{timestamp}.{format}')
+    name = f'all_subdomain_{timestamp}'
+    path = check_path(None, name, format)
     row_list = list()
     for row in datas:
         row.pop('header')
@@ -314,7 +314,7 @@ def export_all(format, datas):
         row_list.append(Record(keys, values))
     rows = RecordCollection(iter(row_list))
     content = rows.export(format)
-    save_data(fpath, content)
+    save_data(path, content)
 
 
 def get_timestamp():
diff --git a/oneforall/takeover.py b/oneforall/takeover.py
index 035726f..85ceb4e 100644
--- a/oneforall/takeover.py
+++ b/oneforall/takeover.py
@@ -56,12 +56,12 @@ class Takeover(Module):
     Note:
         参数format可选格式有'txt', 'rst', 'csv', 'tsv', 'json', 'yaml', 'html',
                           'jira', 'xls', 'xlsx', 'dbf', 'latex', 'ods'
-        参数path为None默认使用OneForAll结果目录
+        参数path默认None使用OneForAll结果目录生成路径
 
     :param any target:  单个子域或者每行一个子域的文件路径(必需参数)
     :param int thread:  线程数(默认100)
     :param str format:  导出格式(默认csv)
-    :param str path:   导出目录(默认None)
+    :param str path:    导出路径(默认None)
     """
     def __init__(self, target, thread=100, path=None, format='csv'):
         Module.__init__(self)
@@ -83,9 +83,7 @@ class Takeover(Module):
             data = str(self.results)
         else:
             data = self.results.export(self.format)
-        timestamp = utils.get_timestamp()
-        path = self.path.joinpath(f'takeover_{timestamp}.{self.format}')
-        utils.save_data(path, data)
+        utils.save_data(self.path, data)
 
     def compare(self, subdomain, cname, responses):
         domain_resp = self.get('http://' + subdomain, check=False)
@@ -136,7 +134,9 @@ class Takeover(Module):
         logger.log('INFOR', f'开始执行{self.source}模块')
         self.subdomains = utils.get_domains(self.target)
         self.format = utils.check_format(self.format, len(self.subdomains))
-        self.path = utils.check_dpath(self.path)
+        timestamp = utils.get_timestamp()
+        name = f'all_subdomain_{timestamp}'
+        self.path = utils.check_path(self.path, name, self.format)
         if self.subdomains:
             logger.log('INFOR', f'正在检查子域接管风险')
             self.fingerprints = get_fingerprint()