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

Update dbexport.py

Browse Source
This commit is contained in:
JrDw0
2020-05-11 10:28:52 +08:00
committed by GitHub
parent 69744cf7cb
commit 7dfa8874ba
1 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dbexport.py
+13 -14
View File
@@ -2,7 +2,7 @@
# coding=utf-8 # coding=utf-8
""" """
OneForAll数据库导出模块 OneForAll export from database module
:copyright: Copyright (c) 2019, Jing Ling. All rights reserved. :copyright: Copyright (c) 2019, Jing Ling. All rights reserved.
:license: GNU General Public License v3.0, see LICENSE for more details. :license: GNU General Public License v3.0, see LICENSE for more details.
@@ -17,25 +17,24 @@ from config.log import logger
def export(table, db=None, alive=False, limit=None, path=None, format='csv', show=False): def export(table, db=None, alive=False, limit=None, path=None, format='csv', show=False):
""" """
OneForAll数据库导出模块 OneForAll export from database module
Example: Example:
python3 dbexport.py --table name --format csv --dir= ./result.csv python3 dbexport.py --table name --format csv --dir= ./result.csv
python3 dbexport.py --db result.db --table name --show False python3 dbexport.py --db result.db --table name --show False
Note: Note:
参数alive可选值True，False分别表示导出存活，全部子域结果 --alive True/False Only export alive subdomains or not (default False)
参数format可选格式有'txt', 'rst', 'csv', 'tsv', 'json', 'yaml', 'html', --format rst/csv/tsv/json/yaml/html/jira/xls/xlsx/dbf/latex/ods (result format)
'jira', 'xls', 'xlsx', 'dbf', 'latex', 'ods' --path Result directory (default directory is ./results)
参数path默认None使用OneForAll结果目录自动生成路径
:param str table: 要导出的表 :param str table: Table to be exported
:param str db: 要导出的数据库路径(默认为results/result.sqlite3) :param str db: Database path to be exported (default ./results/result.sqlite3)
:param bool alive: 只导出存活的子域结果(默认False) :param bool alive: Only export the results of alive subdomains (default False)
:param str limit: 导出限制条件(默认None) :param str limit: Export limit (default None)
:param str format: 导出文件格式(默认csv) :param str format: Result format (default csv)
:param str path: 导出文件路径(默认None) :param str path: Result directory (default None)
:param bool show: 终端显示导出数据(默认False) :param bool show: Displays the exported data in terminal (default False)
""" """
database = Database(db) database = Database(db)
@@ -47,7 +46,7 @@ def export(table, db=None, alive=False, limit=None, path=None, format='csv', sho
data = rows.export(format) data = rows.export(format)
database.close() database.close()
utils.save_data(path, data) utils.save_data(path, data)
logger.log('INFOR', f'{table}主域的子域结果 {path}') logger.log('INFOR', f'{table}\'s subdomains result: {path}')
data_dict = rows.as_dict() data_dict = rows.as_dict()
return data_dict return data_dict