From a63134245bbb7d4be3f95513efa538c33ada684b Mon Sep 17 00:00:00 2001 From: JrD <421273918@qq.com> Date: Sun, 12 Jul 2020 20:41:09 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E7=BB=93=E6=9E=9C=E5=88=A4?= =?UTF-8?q?=E6=96=AD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dbexport.py | 24 +++++++++++++----------- 1 file changed, 13 insertions(+), 11 deletions(-) diff --git a/dbexport.py b/dbexport.py index e897297..110ab94 100644 --- a/dbexport.py +++ b/dbexport.py @@ -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__':