river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-10-27 10:55:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加结果判断

Browse Source
This commit is contained in:
JrD
2020-07-12 20:41:09 +08:00
parent 9a67e4932f
commit a63134245b
1 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dbexport.py
+2
View File
@@ -44,6 +44,7 @@ def export(target, db=None, alive=False, limit=None, path=None, format='csv', sh
database = Database(db) database = Database(db)
domains = utils.get_domains(target) domains = utils.get_domains(target)
datas = [] datas = []
if domains:
for domain in domains: for domain in domains:
table_name = domain_to_table(domain) table_name = domain_to_table(domain)
rows = database.export_data(table_name, alive, limit) rows = database.export_data(table_name, alive, limit)
@@ -55,6 +56,7 @@ def export(target, db=None, alive=False, limit=None, path=None, format='csv', sh
data = rows.as_dict() data = rows.as_dict()
datas.extend(data) datas.extend(data)
database.close() database.close()
if datas:
utils.export_all(alive, format, path, datas) utils.export_all(alive, format, path, datas)