sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-02-02 20:54:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

Consistency fix (down below we use direct SQL)

Browse Source
This commit is contained in:
Miroslav Stampar 2014-01-02 10:59:53 +01:00
parent 0b4fcb6845
commit e0143e397a
1 changed files with 16 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
lib/utils/api.py
View File

@ -54,20 +54,6 @@ class DataStore(object):
# API objects # API objects
class Database(object): class Database(object):
filepath = None filepath = None
LOGS_TABLE = ("CREATE TABLE logs("
"id INTEGER PRIMARY KEY AUTOINCREMENT, "
"taskid INTEGER, time TEXT, "
"level TEXT, message TEXT"
")")
DATA_TABLE = ("CREATE TABLE data("
"id INTEGER PRIMARY KEY AUTOINCREMENT, "
"taskid INTEGER, status INTEGER, "
"content_type INTEGER, value TEXT"
")")
ERRORS_TABLE = ("CREATE TABLE errors("
"id INTEGER PRIMARY KEY AUTOINCREMENT, "
"taskid INTEGER, error TEXT"
")")
def __init__(self, database=None): def __init__(self, database=None):
self.database = self.filepath if database is None else database self.database = self.filepath if database is None else database
@ -96,9 +82,22 @@ class Database(object):
return self.cursor.fetchall() return self.cursor.fetchall()
def init(self): def init(self):
self.execute(self.LOGS_TABLE) self.execute("CREATE TABLE logs("
self.execute(self.DATA_TABLE) "id INTEGER PRIMARY KEY AUTOINCREMENT, "
self.execute(self.ERRORS_TABLE) "taskid INTEGER, time TEXT, "
"level TEXT, message TEXT"
")")
self.execute("CREATE TABLE data("
"id INTEGER PRIMARY KEY AUTOINCREMENT, "
"taskid INTEGER, status INTEGER, "
"content_type INTEGER, value TEXT"
")")
self.execute("CREATE TABLE errors("
"id INTEGER PRIMARY KEY AUTOINCREMENT, "
"taskid INTEGER, error TEXT"
")")
class Task(object): class Task(object):