Menu

[r5]: / trunk / mysql / MySQLdb.py  Maximize  Restore  History

Download this file

115 lines (92 with data), 3.1 kB

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import _mysql
from _mysql import *
from DateTime import Date, Time, Timestamp
from types import StringType, ListType, TupleType
threadsafety = 1
apllevel = "1.1"
class Cursor:
def __init__(self, connection, name=''):
self.connection = connection
self.name = name
self.description = None
self.rowcount = -1
self.result = None
self.arraysize = None
self.warnings = 1
def setinputsizes(self, size): pass
def setoutputsizes(self, size): pass
def execute(self, query, args=None):
from types import ListType, TupleType
from string import rfind, join, split, atoi
db = self.connection.db
if not args:
db.query(query)
elif type(args) is not ListType:
db.query(query % escape_row(args))
else:
p = rfind(query, '(')
if p == -1: raise ProgrammingError, "can't find values"
n = len(args)-1
q = [query % escape_row(args[0])]
qv = query[p:]
for a in args[1:]: q.append(qv % escape_row(a))
q = join(q, ',\n')
print q
db.query(q)
self.result = db.store_result()
if self.result:
self.description = self.result.describe()
self.rowcount = self.result.num_rows()
else:
self.description = None
self.rowcount = -1
if self.warnings:
w = db.info()
if w:
warnings = atoi(split(w)[-1])
if warnings:
raise Warning, w
def fetchone(self):
try:
return self.result.fetch_row()
except AttributeError:
raise ProgrammingError, "no query executed yet"
def fetchmany(self, size=None):
size = size or self.inputsizes or 1
rows = []
for i in range(size):
row = self.fetchone()
if not row: break
rows.append(row)
return rows
def fetchall(self):
rows = []
while 1:
row = self.fetchone()
if not row: break
rows.append(row)
return rows
def nextset(self): pass
def Raw(s): return s
STRING = FIELD_TYPE.STRING
NUMBER = FIELD_TYPE.LONG
TIME = FIELD_TYPE.TIME
TIMESTAMP = FIELD_TYPE.TIMESTAMP
ROW_ID = FIELD_TYPE.LONG
class Connection:
CursorClass = Cursor
def __init__(self, dsn=None, user=None, password=None,
host=None, database=None, **kwargs):
newargs = {}
if user: newargs['user'] = user
if password: newargs['passwd'] = password
if host: newargs['host'] = host
if database: newargs['db'] = database
newargs.update(kwargs)
self.db = apply(_mysql.connect, (), newargs)
def close(self):
self.db.close()
def commit(self): pass
def cursor(self, name=''):
return self.CursorClass(self, name)
Connect = connect = Connection