修改为单次存储

This commit is contained in:
shmilylty
2019-08-28 17:15:30 +08:00
parent e74a738565
commit c9208b61dd
+14 -12
View File
@@ -70,20 +70,22 @@ class Database(object):
:param list results: 结果列表 :param list results: 结果列表
:param str module_name: 模块名 :param str module_name: 模块名
""" """
logger.log('DEBUG', f'正在将{module_name}模块发现{table_name}的子域结果存入数据库') logger.log('DEBUG', f'正在将{module_name}模块发现{table_name}的'
f'子域结果存入数据库')
table_name = table_name.replace('.', '_') table_name = table_name.replace('.', '_')
if results: if results:
try: for result in results:
self.conn.bulk_query( try:
f'insert into "{table_name}" (' self.conn.bulk_query(
f'id, url, subdomain, port, ips, status, reason, valid,' f'insert into "{table_name}" (id, url, subdomain, '
f'title, banner, module, source, elapsed, count)' f'port, ips, status, reason, valid, title, banner, '
f'values (:id, :url, :subdomain, :port, :ips, :status,' f'module, source, elapsed, count)'
f':reason, :valid, :title, :banner, :module, :source,' f'values (:id, :url, :subdomain, :port, :ips, :status,'
f':elapsed, :count)', f':reason, :valid, :title, :banner, :module, :source,'
results) f':elapsed, :count)',
except Exception as e: result)
logger.log('ERROR', e) except Exception as e:
logger.log('ERROR', e)
def copy_table(self, table_name, bak_table_name): def copy_table(self, table_name, bak_table_name):
""" """