gh-100450: Add sqlite.Row.__contains__ method

Lib/test/test_sqlite3/test_factory.py

@@ -240,6 +240,14 @@ def test_sqlite_row_as_sequence(self):
         self.assertEqual(list(reversed(row)), list(reversed(as_tuple)))
         self.assertIsInstance(row, Sequence)
 
+    def test_sqlite_row_contains(self):
+        """Checks if the row object can be used with `in`"""
+        self.con.row_factory = sqlite.Row
+        row = self.con.execute("select 1 as a, 2 as b").fetchone()
+        self.assertIn('a', row)
+        self.assertIn('b', row)
+        self.assertNotIn('c', row)
+
     def test_fake_cursor_class(self):
         # Issue #24257: Incorrect use of PyObject_IsInstance() caused
         # segmentation fault.

Misc/NEWS.d/next/Library/2022-12-23-14-40-16.gh-issue-100450.sevifI.rst

@@ -0,0 +1 @@
+Add ``sqlite.Row.__contains__`` method.

Modules/_sqlite/row.c

@@ -96,6 +96,24 @@ PyObject* pysqlite_row_item(pysqlite_Row* self, Py_ssize_t idx)
    return Py_XNewRef(item);
 }
 
+static int
+pysqlite_row_contains(pysqlite_Row* self, PyObject *arg)
+{
+    Py_ssize_t nitems;
+    Py_ssize_t i;
+    PyObject *key;
+    int cmp = 0;
+
+    nitems = PyTuple_Size(self->description);
+
+    for (i = 0; cmp == 0 && i < nitems; i++) {
+        key = PyTuple_GET_ITEM(PyTuple_GET_ITEM(self->description, i), 0);
+        cmp = PyObject_RichCompareBool(arg, key, Py_EQ);
+    }
+
+    return cmp;
+}
+
 static int
 equal_ignore_case(PyObject *left, PyObject *right)
 {
@@ -249,6 +267,7 @@ static PyType_Slot row_slots[] = {
     {Py_mp_subscript, pysqlite_row_subscript},
     {Py_sq_length, pysqlite_row_length},
     {Py_sq_item, pysqlite_row_item},
+    {Py_sq_contains, pysqlite_row_contains},
     {Py_tp_new, pysqlite_row_new},
     {Py_tp_traverse, row_traverse},
     {Py_tp_clear, row_clear},