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

update

Browse Source
This commit is contained in:
Miroslav Stampar 2010-03-26 21:30:36 +00:00
parent 1ec5221d82
commit 4ca1adba2c
1 changed files with 41 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
lib/core/replication.py
View File

@ -26,6 +26,11 @@ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
import sqlite3 import sqlite3
class Replication: class Replication:
"""
This class holds all methods/classes used for database
replication purposes.
"""
def __init__(self, dbpath): def __init__(self, dbpath):
self.dbpath = dbpath self.dbpath = dbpath
self.connection = sqlite3.connect(dbpath) self.connection = sqlite3.connect(dbpath)
@ -33,6 +38,11 @@ class Replication:
self.cursor = self.connection.cursor() self.cursor = self.connection.cursor()
class DataType: class DataType:
"""
Using this class we define auxiliary objects
used for representing sqlite data types.
"""
def __init__(self, name): def __init__(self, name):
self.name = name self.name = name
@ -43,6 +53,10 @@ class Replication:
return "<DataType: %s>" % self return "<DataType: %s>" % self
class Table: class Table:
"""
This class defines methods used to manipulate table objects.
"""
def __init__(self, parent, name, columns=None, create=True, typeless=False): def __init__(self, parent, name, columns=None, create=True, typeless=False):
self.parent = parent self.parent = parent
self.name = name self.name = name
@ -55,18 +69,39 @@ class Replication:
self.parent.cursor.execute('CREATE TABLE %s (%s)' % (self.name, ','.join(colname for colname in self.columns))) self.parent.cursor.execute('CREATE TABLE %s (%s)' % (self.name, ','.join(colname for colname in self.columns)))
def insert(self, rows): def insert(self, rows):
"""
This function is used for inserting row(s) into current table.
"""
self.parent.cursor.executemany('INSERT INTO %s VALUES (?,?,?,?,?)' % self.name, rows) self.parent.cursor.executemany('INSERT INTO %s VALUES (?,?,?,?,?)' % self.name, rows)
def select(self, condition=None):
"""
This function is used for selecting row(s) from current table.
"""
stmt = 'SELECT * FROM %s' % self.name
if condition:
stmt += 'WHERE %s' % condition
return self.parent.cursor.execute(stmt)
NULL = DataType('NULL') # sqlite data types
NULL = DataType('NULL')
INTEGER = DataType('INTEGER') INTEGER = DataType('INTEGER')
REAL = DataType('REAL') REAL = DataType('REAL')
TEXT = DataType('TEXT') TEXT = DataType('TEXT')
BLOB = DataType('BLOB') BLOB = DataType('BLOB')
def createTable(self, name, columns): def createTable(self, tblname, columns=None):
return Table(self, name, columns) """
This function creates Table instance with current connection settings.
"""
return Table(self, tblname, columns)
def dropTable(self, tblname):
"""
This function drops table with given name using current connection.
"""
self.cursor.execute('DROP TABLE IF EXISTS %s' % tblname)
def __del__(self): def __del__(self):
self.cursor.close() self.cursor.close()
self.connection.close() self.connection.close()