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

增加结果判断

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) database = Database(db)
domains = utils.get_domains(target) domains = utils.get_domains(target)
datas = [] datas = []
for domain in domains: if domains:
table_name = domain_to_table(domain) for domain in domains:
rows = database.export_data(table_name, alive, limit) table_name = domain_to_table(domain)
if rows is None: rows = database.export_data(table_name, alive, limit)
continue if rows is None:
format = utils.check_format(format, len(rows)) continue
if show: format = utils.check_format(format, len(rows))
print(rows.dataset) if show:
data = rows.as_dict() print(rows.dataset)
datas.extend(data) data = rows.as_dict()
datas.extend(data)
database.close() database.close()
utils.export_all(alive, format, path, datas) if datas:
utils.export_all(alive, format, path, datas)
if __name__ == '__main__': if __name__ == '__main__':