def _convertResultSet(description, resultset, qualified=False):
"""internal function that turns a result set into a list of dictionaries."""
if qualified:
fldnames=[x[0] for x in description]
else:
fldnames=[_strip_tablename(x[0]) for x in description]
return [dict(izip(fldnames, row)) for row in resultset]
def iterfetch(resultSpec, sqlTemplate, *values, **kwargs):
"""
a method that executes sql and returns rows of tuples of PyDO
objects and scalar values, ordered according to a result set
specification.
resultSpec is a list that may contain:
* PyDO classes;
* 2-tuples of (PyDO class, alias string), which indicate that
the table represented by the PyDO class is to be referenced by
the given alias;
* strings, which represent arbitrary SQL expressions that may
occur in a SQL column-list specification.
sqlTemplate is a string that may contain interpolation variables
in the style of string.Template. In particular, two variables are
supplied to this template automatically:
$COLUMNS -- a list of columns computed from the supplied resultSpec;
$TABLES -- a list of tables similarly computed.
Additional interpolation variables may be passed in as keyword
arguments. Bind variables to the SQL may also be passed in,
through positional arguments; if there is only one positional
argument, and it is a dictionary, it will be used instead of a
list of values, under the assumption that either the 'named' or
'pyformat' paramstyle is being used.
For each element E in the resultSpec, the result row contains one
element F. If E is a PyDO class, F will either be an instance of
E, or, if all its corresponding columns were null for that row and
E has a uniqueness constraint (which in PyDO is implicitly a not
null constraint), None. If E is a string, F will be whatever the
cursor returned for that column.
"""
resultSpec=list(_processResultSpec(resultSpec))
objs=[x for x in resultSpec if not isinstance(x, basestring)]
# check that all objs have the same connectionAlias
caliases=tuple(frozenset(o.connectionAlias for o in objs))
if len(caliases)>1:
raise ValueError, \
"objects passed to fetch must have same connection alias"
elif len(caliases)==0:
raise ValueError, "must supply some object in result spec"
dbi=objs[0].getDBI()
tables = ', '.join(x.getTable() for x in objs)
# if an item has no uniqueness constraints, it really could
# be all null; otherwise, take all-nullness to mean that
# we're dealing with a join with a no matching row for that
# table. "noneable" means here, "we can represent it as None"
noneable=[o for o in objs if o.getUniquenessConstraints()]
allcols=[]
for item in resultSpec:
if hasattr(item, 'getColumns'):
allcols.append(sorted(item.getColumns(True)))
else:
allcols.append(item)
columns=', '.join(iflatten(allcols))
sql=string.Template(sqlTemplate).substitute(kwargs,
TABLES=tables,
COLUMNS=columns)
c=dbi.cursor()
if len(values)==1 and isinstance(values[0], dict):
values=values[0]
if dbi.verbose:
debug('SQL: %s', sql)
debug('bind variables: %s', values)
c.execute(sql, values)
result=c.fetchall()
c.close()
if not result:
raise StopIteration
retrow=[]
for row in result:
del retrow[:]
p=0
for o, cols in izip(resultSpec, allcols):
if isinstance(o, basestring):
retrow.append(row[p])
p+=1
else:
assert isinstance(o, TableAlias) or (isclass(o) and issubclass(o, PyDO))
d={}
for col in cols:
d[_strip_tablename(col)]=row[p]
p+=1
if o in noneable:
for v in d.itervalues():
if v is not None:
notnull=True
break
else:
notnull=False
else:
notnull=True
if notnull:
retrow.append(o(**d))
else:
retrow.append(None)
yield tuple(retrow)
def iterfetch(resultSpec, sqlTemplate, *values, **kwargs):
"""
a method that executes sql and returns rows of tuples of PyDO
objects and scalar values, ordered according to a result set
specification.
resultSpec is a list that may contain:
* PyDO classes;
* 2-tuples of (PyDO class, alias string), which indicate that
the table represented by the PyDO class is to be referenced by
the given alias;
* strings, which represent arbitrary SQL expressions that may
occur in a SQL column-list specification.
sqlTemplate is a string that may contain interpolation variables
in the style of string.Template. In particular, two variables are
supplied to this template automatically:
$COLUMNS -- a list of columns computed from the supplied resultSpec;
$TABLES -- a list of tables similarly computed.
Additional interpolation variables may be passed in as keyword
arguments. Bind variables to the SQL may also be passed in,
through positional arguments; if there is only one positional
argument, and it is a dictionary, it will be used instead of a
list of values, under the assumption that either the 'named' or
'pyformat' paramstyle is being used.
For each element E in the resultSpec, the result row contains one
element F. If E is a PyDO class, F will either be an instance of
E, or, if all its corresponding columns were null for that row and
E has a uniqueness constraint (which in PyDO is implicitly a not
null constraint), None. If E is a string, F will be whatever the
cursor returned for that column.
"""
resultSpec = list(_processResultSpec(resultSpec))
objs = [x for x in resultSpec if not isinstance(x, basestring)]
# check that all objs have the same connectionAlias
caliases = tuple(frozenset(o.connectionAlias for o in objs))
if len(caliases) != 1:
raise ValueError, \
"objects passed to fetch must have same connection alias"
dbi = objs[0].getDBI()
tables = ', '.join(x.getTable() for x in objs)
# if an item has no uniqueness constraints, it really could
# be all null; otherwise, take all-nullness to mean that
# we're dealing with a join with a no matching row for that
# table. "noneable" means here, "we can represent it as None"
noneable = [o for o in objs if o.getUniquenessConstraints()]
allcols = []
for item in resultSpec:
if hasattr(item, 'getColumns'):
allcols.append(sorted(item.getColumns(True)))
else:
allcols.append(item)
columns = ', '.join(iflatten(allcols))
sql = string.Template(sqlTemplate).substitute(kwargs,
TABLES=tables,
COLUMNS=columns)
c = dbi.cursor()
if len(values) == 1 and isinstance(values, dict):
values = values[0]
if dbi.verbose:
debug('SQL: %s', sql)
debug('bind variables: %s', values)
c.execute(sql, values)
result = c.fetchall()
c.close()
if not result:
raise StopIteration
retrow = []
for row in result:
del retrow[:]
p = 0
for o, cols in izip(resultSpec, allcols):
if isinstance(o, basestring):
retrow.append(row[p])
p += 1
else:
assert isinstance(o, TableAlias) or (isclass(o)
and issubclass(o, PyDO))
d = {}
for col in cols:
d[_strip_tablename(col)] = row[p]
p += 1
if o in noneable:
for v in d.itervalues():
if v is not None:
notnull = True
break
else:
notnull = False
else:
notnull = True
if notnull:
retrow.append(o(**d))
else:
retrow.append(None)
yield tuple(retrow)
