提交 551e0aa9 编写于 作者: D D'Arcy J.M. Cain

Fix parameter handling.

Fix a bug where cs.execute('select %d + %d', (1, 2)) would get interpreted
as cs.executemany('select %d + %d', (1, 2))
上级 37d67eb1
......@@ -60,6 +60,7 @@ import exceptions
import types
import DateTime
import time
import types
### module constants
......@@ -175,9 +176,14 @@ class pgdbCursor:
self.rowcount = -1
def execute(self, operation, params = None):
if type(params) == types.TupleType or type(params) == types.ListType:
# "The parameters may also be specified as list of
# tuples to e.g. insert multiple rows in a single
# operation, but this kind of usage is depreciated:
if params and type(params) == types.ListType and \
type(params[0]) == types.TupleType:
self.executemany(operation, params)
else:
# not a list of tuples
self.executemany(operation, (params,))
def executemany(self, operation, param_seq):
......@@ -190,7 +196,7 @@ class pgdbCursor:
try:
for params in param_seq:
if params != None:
sql = operation % params
sql = _quoteparams(operation, params)
else:
sql = operation
rows = self.__source.execute(sql)
......@@ -251,6 +257,34 @@ class pgdbCursor:
def setoutputsize(self, size, col = 0):
pass
def _quote(x):
if type(x) == types.StringType:
x = "'" + string.replace(
string.replace(str(x), '\\', '\\\\'), "'", "''") + "'"
elif type(x) in (types.IntType, types.LongType, types.FloatType):
pass
elif x is None:
x = 'NULL'
elif hasattr(x, '__pg_repr__'):
x = x.__pg_repr__()
else:
raise InterfaceError, 'do not know how to handle type %s' % type(x)
return x
def _quoteparams(s, params):
if hasattr(params, 'has_key'):
x = {}
for k, v in params.items():
x[k] = _quote(v)
params = x
else:
params = tuple(map(_quote, params))
return s % params
### connection object
class pgdbCnx:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册