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
+13 -14
View File
@@ -70,22 +70,21 @@ 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}" ('
f'insert into "{table_name}" (id, url, subdomain, ' f'id, url, subdomain, port, ips, status, reason, valid,'
f'port, ips, status, reason, valid, title, banner, ' f'title, banner, module, source, elapsed, count)'
f'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)', results)
result) except Exception as e:
except Exception as e: logger.log('ERROR', e)
logger.log('ERROR', e)
def copy_table(self, table_name, bak_table_name): def copy_table(self, table_name, bak_table_name):
""" """