river_tao/OneForAll-mirror
1
0
Fork 0
mirror of https://github.com/shmilylty/OneForAll.git synced 2026-08-26 12:57:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

修改为批量存储

Browse Source
This commit is contained in:
shmilylty
2019-10-13 17:11:18 +08:00
parent 071c115d70
commit 1cd52ce2d6
1 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oneforall/common/database.py
+6 -7
View File
@@ -70,20 +70,19 @@ 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:
for result in results:
try: try:
self.conn.bulk_query( self.conn.bulk_query(
f'insert into "{table_name}" (id, url, subdomain, ' f'insert into "{table_name}" ('
f'port, ips, status, reason, valid, title, banner, ' f'id, url, subdomain, port, ips, status, reason, valid,'
f'module, source, elapsed, count)' f'title, banner, module, source, elapsed, count)'
f'values (:id, :url, :subdomain, :port, :ips, :status,' f'values (:id, :url, :subdomain, :port, :ips, :status,'
f':reason, :valid, :title, :banner, :module, :source,' f':reason, :valid, :title, :banner, :module, :source,'
f':elapsed, :count)', f':elapsed, :count)',
result) results)
except Exception as e: except Exception as e:
logger.log('ERROR', e) logger.log('ERROR', e)