river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-25 20:37:48 +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
+13 -11
View File
@@ -44,18 +44,20 @@ def export(target, db=None, alive=False, limit=None, path=None, format='csv', sh
database = Database(db)
domains = utils.get_domains(target)
datas = []
for domain in domains:
table_name = domain_to_table(domain)
rows = database.export_data(table_name, alive, limit)
if rows is None:
continue
format = utils.check_format(format, len(rows))
if show:
print(rows.dataset)
data = rows.as_dict()
datas.extend(data)
if domains:
for domain in domains:
table_name = domain_to_table(domain)
rows = database.export_data(table_name, alive, limit)
if rows is None:
continue
format = utils.check_format(format, len(rows))
if show:
print(rows.dataset)
data = rows.as_dict()
datas.extend(data)
database.close()
utils.export_all(alive, format, path, datas)
if datas:
utils.export_all(alive, format, path, datas)
if __name__ == '__main__':