[Python-checkins] gh-92206: Improve scoping of sqlite3 register cursor helper (#92212)
JelleZijlstra
webhook-mailer at python.org
Tue May 3 08:33:28 EDT 2022
https://github.com/python/cpython/commit/1d4a9a45b7ac8c1c5fecc363c988be59500f1ed7
commit: 1d4a9a45b7ac8c1c5fecc363c988be59500f1ed7
branch: main
author: Erlend Egeberg Aasland <erlend.aasland at protonmail.com>
committer: JelleZijlstra <jelle.zijlstra at gmail.com>
date: 2022-05-03T06:33:24-06:00
summary:
gh-92206: Improve scoping of sqlite3 register cursor helper (#92212)
files:
M Modules/_sqlite/connection.c
M Modules/_sqlite/connection.h
M Modules/_sqlite/cursor.c
diff --git a/Modules/_sqlite/connection.c b/Modules/_sqlite/connection.c
index fe244c8b22771..75d29704695c6 100644
--- a/Modules/_sqlite/connection.c
+++ b/Modules/_sqlite/connection.c
@@ -371,32 +371,6 @@ connection_dealloc(pysqlite_Connection *self)
Py_DECREF(tp);
}
-/*
- * Registers a cursor with the connection.
- *
- * 0 => error; 1 => ok
- */
-int pysqlite_connection_register_cursor(pysqlite_Connection* connection, PyObject* cursor)
-{
- PyObject* weakref;
-
- weakref = PyWeakref_NewRef((PyObject*)cursor, NULL);
- if (!weakref) {
- goto error;
- }
-
- if (PyList_Append(connection->cursors, weakref) != 0) {
- Py_CLEAR(weakref);
- goto error;
- }
-
- Py_DECREF(weakref);
-
- return 1;
-error:
- return 0;
-}
-
/*[clinic input]
_sqlite3.Connection.cursor as pysqlite_connection_cursor
diff --git a/Modules/_sqlite/connection.h b/Modules/_sqlite/connection.h
index 2b946ff3c7369..629fe3d3a95a1 100644
--- a/Modules/_sqlite/connection.h
+++ b/Modules/_sqlite/connection.h
@@ -99,7 +99,6 @@ typedef struct
PyObject* NotSupportedError;
} pysqlite_Connection;
-int pysqlite_connection_register_cursor(pysqlite_Connection* connection, PyObject* cursor);
int pysqlite_check_thread(pysqlite_Connection* self);
int pysqlite_check_connection(pysqlite_Connection* con);
diff --git a/Modules/_sqlite/cursor.c b/Modules/_sqlite/cursor.c
index 861704f95cc83..84f481792ddd3 100644
--- a/Modules/_sqlite/cursor.c
+++ b/Modules/_sqlite/cursor.c
@@ -35,6 +35,28 @@ class _sqlite3.Cursor "pysqlite_Cursor *" "clinic_state()->CursorType"
[clinic start generated code]*/
/*[clinic end generated code: output=da39a3ee5e6b4b0d input=3c5b8115c5cf30f1]*/
+/*
+ * Registers a cursor with the connection.
+ *
+ * 0 => error; 1 => ok
+ */
+static int
+register_cursor(pysqlite_Connection *connection, PyObject *cursor)
+{
+ PyObject *weakref = PyWeakref_NewRef((PyObject *)cursor, NULL);
+ if (weakref == NULL) {
+ return 0;
+ }
+
+ if (PyList_Append(connection->cursors, weakref) < 0) {
+ Py_CLEAR(weakref);
+ return 0;
+ }
+
+ Py_DECREF(weakref);
+ return 1;
+}
+
/*[clinic input]
_sqlite3.Cursor.__init__ as pysqlite_cursor_init
@@ -70,7 +92,7 @@ pysqlite_cursor_init_impl(pysqlite_Cursor *self,
return -1;
}
- if (!pysqlite_connection_register_cursor(connection, (PyObject*)self)) {
+ if (!register_cursor(connection, (PyObject *)self)) {
return -1;
}
More information about the Python-checkins
mailing list