gh-93044: No longer convert the database argument of sqlite3.connect() to bytes (GH-93046)

Just pass it to the factory as is.
This commit is contained in:
Serhiy Storchaka 2022-05-21 14:35:46 +03:00 committed by GitHub
parent b96e20c1d9
commit 14c0d33016
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 19 additions and 7 deletions

View File

@ -676,6 +676,19 @@ class OpenTests(unittest.TestCase):
with managed_connect(f"file:{TESTFN}?mode=ro", uri=True) as cx:
cx.execute(self._sql)
def test_factory_database_arg(self):
def factory(database, *args, **kwargs):
nonlocal database_arg
database_arg = database
return sqlite.Connection(":memory:", *args, **kwargs)
for database in (TESTFN, os.fsencode(TESTFN),
FakePath(TESTFN), FakePath(os.fsencode(TESTFN))):
database_arg = None
with sqlite.connect(database, factory=factory):
pass
self.assertEqual(database_arg, database)
def test_database_keyword(self):
with sqlite.connect(database=":memory:") as cx:
self.assertEqual(type(cx), sqlite.Connection)

View File

@ -0,0 +1,2 @@
No longer convert the database argument of :func:`sqlite3.connect` to bytes
before passing it to the factory.

View File

@ -43,9 +43,7 @@ pysqlite_connect(PyObject *module, PyObject *const *args, Py_ssize_t nargs, PyOb
if (!args) {
goto exit;
}
if (!PyUnicode_FSConverter(args[0], &database)) {
goto exit;
}
database = args[0];
if (!noptargs) {
goto skip_optional_pos;
}
@ -294,4 +292,4 @@ skip_optional:
exit:
return return_value;
}
/*[clinic end generated code: output=43aa4f4356f9269d input=a9049054013a1b77]*/
/*[clinic end generated code: output=a7cfa6dc9d54273c input=a9049054013a1b77]*/

View File

@ -46,7 +46,7 @@ module _sqlite3
/*[clinic input]
_sqlite3.connect as pysqlite_connect
database: object(converter='PyUnicode_FSConverter')
database: object
timeout: double = 5.0
detect_types: int = 0
isolation_level: object = NULL
@ -66,7 +66,7 @@ pysqlite_connect_impl(PyObject *module, PyObject *database, double timeout,
int detect_types, PyObject *isolation_level,
int check_same_thread, PyObject *factory,
int cached_statements, int uri)
/*[clinic end generated code: output=450ac9078b4868bb input=ea6355ba55a78e12]*/
/*[clinic end generated code: output=450ac9078b4868bb input=e16914663ddf93ce]*/
{
if (isolation_level == NULL) {
isolation_level = PyUnicode_FromString("");
@ -81,7 +81,6 @@ pysqlite_connect_impl(PyObject *module, PyObject *database, double timeout,
timeout, detect_types,
isolation_level, check_same_thread,
factory, cached_statements, uri);
Py_DECREF(database); // needed bco. the AC FSConverter
Py_DECREF(isolation_level);
return res;
}