sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-04-28 21:03:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

minor refactoring - #297

Browse Source
This commit is contained in:
Bernardo Damele 2013-02-06 17:09:43 +00:00
parent b272b0574d
commit e439c3d3f5
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lib/utils/api.py
View File

@ -75,6 +75,9 @@ class Database(object):
self.cursor.close() self.cursor.close()
self.connection.close() self.connection.close()
def commit(self):
self.cursor.commit()
def execute(self, statement, arguments=None): def execute(self, statement, arguments=None):
if arguments: if arguments:
self.cursor.execute(statement, arguments) self.cursor.execute(statement, arguments)
@ -186,6 +189,10 @@ class StdDbOut(object):
output = conf.database_cursor.execute("SELECT id, value FROM data WHERE taskid = ? AND status = ? AND content_type = ? LIMIT 0,1", output = conf.database_cursor.execute("SELECT id, value FROM data WHERE taskid = ? AND status = ? AND content_type = ? LIMIT 0,1",
(self.taskid, status, content_type)) (self.taskid, status, content_type))
# Delete partial output from IPC database if we have got a complete output
if status == CONTENT_STATUS.COMPLETE and len(output) > 0:
conf.database_cursor.execute("DELETE FROM data WHERE id=?", (output[0][0],))
if status == CONTENT_STATUS.IN_PROGRESS: if status == CONTENT_STATUS.IN_PROGRESS:
if len(output) == 0: if len(output) == 0:
conf.database_cursor.execute("INSERT INTO data VALUES(NULL, ?, ?, ?, ?)", conf.database_cursor.execute("INSERT INTO data VALUES(NULL, ?, ?, ?, ?)",
@ -195,10 +202,6 @@ class StdDbOut(object):
conf.database_cursor.execute("UPDATE data SET value = ? WHERE id = ?", conf.database_cursor.execute("UPDATE data SET value = ? WHERE id = ?",
(jsonize(new_value), output[0][0])) (jsonize(new_value), output[0][0]))
else: else:
if len(output) > 0:
conf.database_cursor.execute("DELETE FROM data WHERE taskid = ? AND status = %s AND content_type = ?" % CONTENT_STATUS.IN_PROGRESS,
(self.taskid, content_type))
conf.database_cursor.execute("INSERT INTO data VALUES(NULL, ?, ?, ?, ?)", conf.database_cursor.execute("INSERT INTO data VALUES(NULL, ?, ?, ?, ?)",
(self.taskid, status, content_type, jsonize(value))) (self.taskid, status, content_type, jsonize(value)))
else: else: