def _one_statement(self, method_name, stream, bypass_types):
"""
Route to the correct optimized function for one statement queries
:param method_name: the name of the method
:type method_name: string
:param stream: the stream of tokens
:type stream: iterable of tokens
:param bypass_types: set if parsing should bypass types
:type bypass_types: boolean
"""
sql = Tokens2Unicode(stream)
if self._paramstyle:
sql = self._paramstyle(sql)
# Insert statement (return last row id)
if IsType('INSERT')(stream):
return self._one_statement_INSERT(method_name, sql, bypass_types)
## Update statement (return affected row count)
#if IsType('UPDATE')(stream):
# return self._one_statement_UPDATE(method_name, sql, bypass_types)
# One-value function (a row of a cell)
if getlimit(stream) == 1:
columns = getcolumns(stream)
# Value function (one row, one field)
if len(columns) == 1 and columns[0] != '*':
return self._one_statement_value(method_name, sql, bypass_types)
# Register function (one row, several fields)
return self._one_statement_register(method_name, sql, bypass_types)
# Table function (several rows)
return self._one_statement_table(method_name, sql, bypass_types)
def test_getlimit3(self):
limit = getlimit(tokenize(self.sql3))
self.assertEqual(limit, 1)
