29 |
# (which it does NOT by default) |
# (which it does NOT by default) |
30 |
|
|
31 |
|
|
32 |
import string |
__all__ = ['DataObject_SQL','DataObject_Object'] |
|
import sys |
|
|
from gnue.common.datasources import GDataObjects, GConditions |
|
|
from gnue.common.apps import GDebug |
|
|
from gnue.common.datasources.drivers import DBSIG2 |
|
33 |
|
|
|
import MySQLdb |
|
34 |
|
|
35 |
|
from gnue.common.datasources.drivers import DBSIG2 |
36 |
|
|
37 |
|
class _Base: |
38 |
|
_escapeSingleQuote = '\\' |
39 |
|
_dateTimeFormat = "'%Y-%m-%d %H:%M:%S'" |
40 |
|
|
41 |
class MySQL_RecordSet(DBSIG2.RecordSet): |
class DataObject_SQL(_Base, DBSIG2.DataObject_SQL): |
42 |
pass |
pass |
|
|
|
|
|
|
|
class MySQL_ResultSet(DBSIG2.ResultSet): |
|
|
def __init__(self, dataObject, cursor=None, defaultValues={}, masterRecordSet=None): |
|
|
DBSIG2.ResultSet.__init__(self, dataObject, \ |
|
|
cursor, defaultValues, masterRecordSet) |
|
|
self._recordSetClass = MySQL_RecordSet |
|
|
|
|
|
# Compensate for bug in python mysql drivers older than 0.9.2a2 |
|
|
if MySQLdb.__version__ >= '0.9.2a2': |
|
|
self.fetchBugFix = self._cursor.fetchmany |
|
|
else: |
|
|
self.__done = 0 |
|
|
self.fetchBugFix = self.__mySqlNeedsLotsOfTLC |
|
|
|
|
43 |
|
|
44 |
# Compensate for MySQ bug |
class DataObject_Object(_Base, DBSIG2.DataObject_Object): |
45 |
def __mySqlNeedsLotsOfTLC(self): |
pass |
|
if self.__done: |
|
|
return None |
|
|
|
|
|
self.__done = 1 |
|
|
return self._cursor.fetchall() |
|
|
|
|
|
|
|
|
def _loadNextRecord(self): |
|
|
if self._cursor: |
|
|
rs = None |
|
|
|
|
|
try: |
|
|
# See __init__ for details |
|
|
rsets = self.fetchBugFix() |
|
|
|
|
|
except self._dataObject._DatabaseError, err: |
|
|
raise GDataObjects.ConnectionError, err |
|
|
|
|
|
if rsets and len(rsets): |
|
|
for rs in(rsets): |
|
|
if rs: |
|
|
i = 0 |
|
|
dict = {} |
|
|
for f in (rs): |
|
|
dict[string.lower(self._fieldNames[i])] = f |
|
|
i += 1 |
|
|
self._cachedRecords.append (self._recordSetClass(parent=self, \ |
|
|
initialData=dict)) |
|
|
else: |
|
|
return 0 |
|
|
return 1 |
|
|
else: |
|
|
return 0 |
|
|
else: |
|
|
return 0 |
|
|
|
|
|
class MySQL_DataObject(DBSIG2.DataObject): |
|
|
def __init__(self): |
|
|
DBSIG2.DataObject.__init__(self) |
|
|
self._DatabaseError = MySQLdb.DatabaseError |
|
|
self._resultSetClass = MySQL_ResultSet |
|
|
def connect(self, connectData={}): |
|
|
GDebug.printMesg(1,"Mysql database driver initializing") |
|
|
|
|
|
try: |
|
|
self._dataConnection = MySQLdb.connect(user=connectData['_username'], |
|
|
passwd=connectData['_password'], |
|
|
host=connectData['host'], |
|
|
db=connectData['dbname']) |
|
|
except self._DatabaseError, value: |
|
|
raise GDataObjects.LoginError, value |
|
|
|
|
|
self._beginTransaction() |
|
|
self._postConnect() |
|
|
|
|
|
|
|
|
def _postConnect(self): |
|
|
self.triggerExtensions = TriggerExtensions(self._dataConnection) |
|
|
|
|
|
|
|
|
def _beginTransaction(self): |
|
|
try: |
|
|
self._dataConnection.begin() |
|
|
except: |
|
|
pass |
|
|
|
|
|
|
|
|
# |
|
|
# Schema (metadata) functions |
|
|
# |
|
|
|
|
|
# Return a list of the types of Schema objects this driver provides |
|
|
def getSchemaTypes(self): |
|
|
return [('table',_('Tables'),1)] |
|
|
|
|
|
# Return a list of Schema objects |
|
|
def getSchemaList(self, type=None): |
|
|
|
|
|
# TODO: This excludes any system tables and views. Should it? |
|
|
statement = "SHOW TABLES" |
|
|
|
|
|
cursor = self._dataConnection.cursor() |
|
|
cursor.execute(statement) |
|
|
|
|
|
list = [] |
|
|
for rs in cursor.fetchall(): |
|
|
list.append(GDataObjects.Schema(attrs={'id':rs[0], 'name':rs[0], |
|
|
'type':'table', |
|
|
'primarykey': self.__getPrimaryKey(rs[0])}, |
|
|
getChildSchema=self.__getFieldSchema)) |
|
|
|
|
|
cursor.close() |
|
|
return list |
|
|
|
|
|
|
|
|
# Find a schema object with specified name |
|
|
def getSchemaByName(self, name, type=None): |
|
|
statement = "DESCRIBE %s" % (name) |
|
|
|
|
|
cursor = self._dataConnection.cursor() |
|
|
cursor.execute(statement) |
|
|
|
|
|
rs = cursor.fetchone() |
|
|
if rs: |
|
|
schema = GDataObjects.Schema(attrs={'id':name, 'name':name, |
|
|
'type':'table', |
|
|
'primarykey': self.__getPrimaryKey(name,cursor)}, |
|
|
getChildSchema=self.__getFieldSchema) |
|
|
else: |
|
|
schema = None |
|
|
|
|
|
cursor.close() |
|
|
return schema |
|
|
|
|
|
|
|
|
def __getPrimaryKey(self, id, cursor=None): |
|
|
statement = "DESCRIBE %s" % id |
|
|
if not cursor: |
|
|
cursor = self._dataConnection.cursor() |
|
|
close_cursor = 1 |
|
|
else: |
|
|
close_cursor = 0 |
|
|
cursor.execute(statement) |
|
|
|
|
|
lst = [] |
|
|
for rs in cursor.fetchall(): |
|
|
if rs[3] == 'PRI': |
|
|
lst.append(rs[0]) |
|
|
|
|
|
if close_cursor: |
|
|
cursor.close() |
|
|
|
|
|
return tuple(lst) |
|
|
|
|
|
# Get fields for a table |
|
|
def __getFieldSchema(self, parent): |
|
|
|
|
|
statement = "DESCRIBE %s" % parent.id |
|
|
|
|
|
cursor = self._dataConnection.cursor() |
|
|
cursor.execute(statement) |
|
|
|
|
|
list = [] |
|
|
for rs in cursor.fetchall(): |
|
|
|
|
|
nativetype = string.split(string.replace(rs[1],')',''),'(') |
|
|
|
|
|
|
|
|
attrs={'id': "%s.%s" % (parent.id, rs[0]), 'name': rs[0], |
|
|
'type':'field', 'nativetype': nativetype[0], |
|
|
'required': rs[2] != 'YES'} |
|
|
|
|
|
if nativetype[0] in ('int','integer','bigint','mediumint', |
|
|
'smallint','tinyint','float','real', |
|
|
'double','decimal'): |
|
|
attrs['datatype']='number' |
|
|
elif nativetype[0] in ('date','time','timestamp','datetime'): |
|
|
attrs['datatype']='date' |
|
|
else: |
|
|
attrs['datatype']='text' |
|
|
|
|
|
try: |
|
|
if len(nativetype) == 2: |
|
|
try: |
|
|
ln, prec = nativetype[1].split(',') |
|
|
except: |
|
|
ln = nativetype[1] |
|
|
prec = None |
|
|
attrs['length'] = int(ln.split()[0]) |
|
|
if prec != None: |
|
|
attrs['precision'] = int(prec) |
|
|
except ValueError: |
|
|
GDebug.printMesg(1,'WARNING: mysql native type error: %s' % nativetype) |
|
|
|
|
|
if rs[4] not in ('NULL', '0000-00-00 00:00:00','', None): |
|
|
attrs['defaulttype'] = 'constant' |
|
|
attrs['defaultval'] = rs[4] |
|
|
|
|
|
if rs[5] == 'auto_increment': |
|
|
attrs['defaulttype'] = 'serial' |
|
|
|
|
|
|
|
|
list.append(GDataObjects.Schema(attrs=attrs)) |
|
|
|
|
|
cursor.close() |
|
|
return list |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class MySQL_DataObject_Object(MySQL_DataObject, \ |
|
|
DBSIG2.DataObject_Object): |
|
|
|
|
|
def __init__(self): |
|
|
MySQL_DataObject.__init__(self) |
|
|
|
|
|
def _buildQuery(self, conditions={},forDetail=None,additionalSQL=""): |
|
|
return DBSIG2.DataObject_Object._buildQuery(self, conditions,forDetail,additionalSQL) |
|
|
|
|
|
|
|
|
class MySQL_DataObject_SQL(MySQL_DataObject, \ |
|
|
DBSIG2.DataObject_SQL): |
|
|
def __init__(self): |
|
|
# Call DBSIG init first because MySQL_DataObject needs to overwrite |
|
|
# some of its values |
|
|
DBSIG2.DataObject_SQL.__init__(self) |
|
|
MySQL_DataObject.__init__(self) |
|
|
|
|
|
def _buildQuery(self, conditions={}): |
|
|
return DBSIG2.DataObject_SQL._buildQuery(self, conditions) |
|
|
|
|
|
|
|
|
# |
|
|
# Extensions to Trigger Namespaces |
|
|
# |
|
|
class TriggerExtensions: |
|
|
|
|
|
def __init__(self, connection): |
|
|
self.__connection = connection |
|
|
|
|
|
# Return the current date, according to database |
|
|
# def getDate(self): |
|
|
# pass |
|
|
|
|
|
# Return a sequence number from sequence 'name' |
|
|
# def getSequence(self, name): |
|
|
# pass |
|
|
|
|
|
# Run the SQL statement 'statement' |
|
|
# def sql(self, statement): |
|
|
# pass |
|
|
|
|
|
|
|
|
|
|
|
###################################### |
|
|
# |
|
|
# The following hashes describe |
|
|
# this driver's characteristings. |
|
|
# |
|
|
###################################### |
|
|
|
|
|
# |
|
|
# All datasouce "types" and corresponding DataObject class |
|
|
# |
|
|
supportedDataObjects = { |
|
|
'object': MySQL_DataObject_Object, |
|
|
'sql': MySQL_DataObject_SQL |
|
|
} |
|
46 |
|
|
47 |
|
|