python · ZackerySpytz · Apr 17, 2020 · pitrou · Apr 18, 2020
diff --git a/Doc/library/dbm.rst b/Doc/library/dbm.rst
@@ -237,6 +237,13 @@ supported.
 
       Close the ``gdbm`` database.
 
+   .. method:: gdbm.count()
+
+      Count the number of records in the database.  This method requires GNU
+      DBM version 1.11 or greater.
+
+      .. versionadded:: 3.9
+
 :mod:`dbm.ndbm` --- Interface based on ndbm
 -------------------------------------------
 

diff --git a/Lib/test/test_dbm_gnu.py b/Lib/test/test_dbm_gnu.py
@@ -4,19 +4,18 @@
 import os
 from test.support import TESTFN, TESTFN_NONASCII, unlink
 
+try:
+    from _gdbm import _GDBM_VERSION as gdbm_version
+except ImportError:
+    gdbm_version = None
 
 filename = TESTFN
 
 class TestGdbm(unittest.TestCase):
     @staticmethod
     def setUpClass():
-        if support.verbose:
-            try:
-                from _gdbm import _GDBM_VERSION as version
-            except ImportError:
-                pass
-            else:
-                print(f"gdbm version: {version}")
+        if support.verbose and gdbm_version:
+            print(f"gdbm version: {gdbm_version}")
 
     def setUp(self):
         self.g = None
@@ -162,6 +161,21 @@ def test_nonexisting_file(self):
         self.assertIn(nonexisting_file, str(cm.exception))
         self.assertEqual(cm.exception.filename, nonexisting_file)
 
+    @unittest.skipUnless(gdbm_version and gdbm_version >= (1, 11, 0),
+                         'needs >= 1.11')
+    def test_count(self):
+        self.g = gdbm.open(filename, 'c')
+        self.assertEqual(self.g.count(), 0)
+
+        self.g['a'] = 'a'
+        self.assertEqual(self.g.count(), 1)
+        self.g['b'] = 'b'
+        self.assertEqual(self.g.count(), 2)
+        del self.g['a']
+        self.assertEqual(self.g.count(), 1)
+        del self.g['b']
+        self.assertEqual(self.g.count(), 0)
+
 
 if __name__ == '__main__':
     unittest.main()
diff --git a/Misc/NEWS.d/next/Library/2020-04-17-08-28-08.bpo-32494.k4zxIG.rst b/Misc/NEWS.d/next/Library/2020-04-17-08-28-08.bpo-32494.k4zxIG.rst
@@ -0,0 +1,2 @@
+Add :meth:`gdbm.count` to wrap the ``gdbm_count()`` function added in GDBM
+1.11.
diff --git a/Modules/_gdbmmodule.c b/Modules/_gdbmmodule.c
@@ -486,6 +486,32 @@ _gdbm_gdbm_sync_impl(dbmobject *self)
     Py_RETURN_NONE;
 }
 
+#if GDBM_VERSION_MAJOR >= 1 && GDBM_VERSION_MINOR >= 11
+/*[clinic input]
+_gdbm.gdbm.count
+
+Count the number of records in the database.
+[clinic start generated code]*/
+
+static PyObject *
+_gdbm_gdbm_count_impl(dbmobject *self)
+/*[clinic end generated code: output=8b62b02ae4c2fcee input=1b9d3f0877315a04]*/
+{
+    gdbm_count_t count;
+    check_dbmobject_open(self);
+    if (gdbm_count(self->di_dbm, &count) == -1) {
+        if (errno != 0) {
+            PyErr_SetFromErrno(DbmError);
+        }
+        else {
+            PyErr_SetString(DbmError, gdbm_strerror(gdbm_errno));
+        }
+        return NULL;
+    }
+    return PyLong_FromUnsignedLongLong(count);
+}
+#endif
+
 static PyObject *
 dbm__enter__(PyObject *self, PyObject *args)
 {
@@ -509,6 +535,7 @@ static PyMethodDef dbm_methods[] = {
     _GDBM_GDBM_SYNC_METHODDEF
     _GDBM_GDBM_GET_METHODDEF
     _GDBM_GDBM_SETDEFAULT_METHODDEF
+    _GDBM_GDBM_COUNT_METHODDEF
     {"__enter__", dbm__enter__, METH_NOARGS, NULL},
     {"__exit__",  dbm__exit__, METH_VARARGS, NULL},
     {NULL,              NULL}           /* sentinel */

diff --git a/Modules/clinic/_gdbmmodule.c.h b/Modules/clinic/_gdbmmodule.c.h
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		Add :meth:`gdbm.count` to wrap the ``gdbm_count()`` function added in GDBM
		1.11.