Commit c6a2320e authored by Victor Stinner's avatar Victor Stinner Committed by GitHub

bpo-37406: sqlite3 raises TypeError for wrong operation type (GH-14386)

The sqlite3 module now raises TypeError, rather than ValueError, if
operation argument type is not str: execute(), executemany() and
calling a connection.
parent ed076ed4
...@@ -230,7 +230,7 @@ class CursorTests(unittest.TestCase): ...@@ -230,7 +230,7 @@ class CursorTests(unittest.TestCase):
""") """)
def CheckExecuteWrongSqlArg(self): def CheckExecuteWrongSqlArg(self):
with self.assertRaises(ValueError): with self.assertRaises(TypeError):
self.cu.execute(42) self.cu.execute(42)
def CheckExecuteArgInt(self): def CheckExecuteArgInt(self):
...@@ -377,7 +377,7 @@ class CursorTests(unittest.TestCase): ...@@ -377,7 +377,7 @@ class CursorTests(unittest.TestCase):
self.cu.executemany("insert into test(income) values (?)", mygen()) self.cu.executemany("insert into test(income) values (?)", mygen())
def CheckExecuteManyWrongSqlArg(self): def CheckExecuteManyWrongSqlArg(self):
with self.assertRaises(ValueError): with self.assertRaises(TypeError):
self.cu.executemany(42, [(3,)]) self.cu.executemany(42, [(3,)])
def CheckExecuteManySelect(self): def CheckExecuteManySelect(self):
......
...@@ -261,7 +261,7 @@ class RegressionTests(unittest.TestCase): ...@@ -261,7 +261,7 @@ class RegressionTests(unittest.TestCase):
Call a connection with a non-string SQL request: check error handling Call a connection with a non-string SQL request: check error handling
of the statement constructor. of the statement constructor.
""" """
self.assertRaises(sqlite.Warning, self.con, 1) self.assertRaises(TypeError, self.con, 1)
def CheckCollation(self): def CheckCollation(self):
def collation_cb(a, b): def collation_cb(a, b):
......
The sqlite3 module now raises TypeError, rather than ValueError, if
operation argument type is not str: execute(), executemany() and calling a
connection.
...@@ -1223,7 +1223,7 @@ PyObject* pysqlite_connection_call(pysqlite_Connection* self, PyObject* args, Py ...@@ -1223,7 +1223,7 @@ PyObject* pysqlite_connection_call(pysqlite_Connection* self, PyObject* args, Py
if (!_PyArg_NoKeywords(MODULE_NAME ".Connection", kwargs)) if (!_PyArg_NoKeywords(MODULE_NAME ".Connection", kwargs))
return NULL; return NULL;
if (!PyArg_ParseTuple(args, "O", &sql)) if (!PyArg_ParseTuple(args, "U", &sql))
return NULL; return NULL;
_pysqlite_drop_unused_statement_references(self); _pysqlite_drop_unused_statement_references(self);
......
...@@ -384,12 +384,7 @@ _pysqlite_query_execute(pysqlite_Cursor* self, int multiple, PyObject* args) ...@@ -384,12 +384,7 @@ _pysqlite_query_execute(pysqlite_Cursor* self, int multiple, PyObject* args)
if (multiple) { if (multiple) {
/* executemany() */ /* executemany() */
if (!PyArg_ParseTuple(args, "OO", &operation, &second_argument)) { if (!PyArg_ParseTuple(args, "UO", &operation, &second_argument)) {
goto error;
}
if (!PyUnicode_Check(operation)) {
PyErr_SetString(PyExc_ValueError, "operation parameter must be str");
goto error; goto error;
} }
...@@ -406,12 +401,7 @@ _pysqlite_query_execute(pysqlite_Cursor* self, int multiple, PyObject* args) ...@@ -406,12 +401,7 @@ _pysqlite_query_execute(pysqlite_Cursor* self, int multiple, PyObject* args)
} }
} else { } else {
/* execute() */ /* execute() */
if (!PyArg_ParseTuple(args, "O|O", &operation, &second_argument)) { if (!PyArg_ParseTuple(args, "U|O", &operation, &second_argument)) {
goto error;
}
if (!PyUnicode_Check(operation)) {
PyErr_SetString(PyExc_ValueError, "operation parameter must be str");
goto error; goto error;
} }
......
...@@ -59,6 +59,8 @@ int pysqlite_statement_create(pysqlite_Statement* self, pysqlite_Connection* con ...@@ -59,6 +59,8 @@ int pysqlite_statement_create(pysqlite_Statement* self, pysqlite_Connection* con
self->st = NULL; self->st = NULL;
self->in_use = 0; self->in_use = 0;
assert(PyUnicode_Check(sql));
sql_cstr = PyUnicode_AsUTF8AndSize(sql, &sql_cstr_len); sql_cstr = PyUnicode_AsUTF8AndSize(sql, &sql_cstr_len);
if (sql_cstr == NULL) { if (sql_cstr == NULL) {
rc = PYSQLITE_SQL_WRONG_TYPE; rc = PYSQLITE_SQL_WRONG_TYPE;
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment