gh-107122: Add clear method to dbm.ndbm module

Author: corona10

Doc/library/dbm.rst

@@ -313,6 +313,11 @@ to locate the appropriate header file to simplify building this module.
 
  Close the ``ndbm`` database.
 
+ .. method:: ndbm.clear()
+ Remove all items from the ``ndbm`` database.
+
+ .. versionadded:: 3.13
+
 
 :mod:`dbm.dumb` --- Portable DBM implementation
 -----------------------------------------------

Lib/test/test_dbm.py

@@ -155,6 +155,21 @@ def test_keys(self):
  self.assertNotIn(b'xxx', d)
  self.assertRaises(KeyError, lambda: d[b'xxx'])
 
+ def test_clear(self):
+ with dbm.open(_fname, 'c') as d:
+ self.assertEqual(d.keys(), [])
+ a = [(b'a', b'b'), (b'12345678910', b'019237410982340912840198242')]
+ for k, v in a:
+ d[k] = v
+ for k, _ in a:
+ self.assertIn(k, d)
+ self.assertEqual(len(d), len(a))
+
+ d.clear()
+ self.assertEqual(len(d), 0)
+ for k, _ in a:
+ self.assertNotIn(k, d)
+
  def setUp(self):
  self.addCleanup(setattr, dbm, '_defaultmod', dbm._defaultmod)
  dbm._defaultmod = self.module

Misc/NEWS.d/next/Core and Builtins/2023-07-23-21-16-54.gh-issue-107122.VNuNcq.rst

@@ -0,0 +1 @@
+Add :meth:`clear` to :mod:`dbm.ndbm`. Patch By Dong-hee Na.

Modules/_dbmmodule.c

@@ -414,6 +414,34 @@ _dbm_dbm_setdefault_impl(dbmobject *self, PyTypeObject *cls, const char *key,
  return default_value;
 }
 
+/*[clinic input]
+_dbm.dbm.clear
+ cls: defining_class
+ /
+Remove all items from the database.
+
+[clinic start generated code]*/
+
+static PyObject *
+_dbm_dbm_clear_impl(dbmobject *self, PyTypeObject *cls)
+/*[clinic end generated code: output=8d126b9e1d01a434 input=43aa6ca1acb7f5f5]*/
+{
+ _dbm_state *state = PyType_GetModuleState(cls);
+ assert(state != NULL);
+ check_dbmobject_open(self, state->dbm_error);
+ for (datum key = dbm_firstkey(self->di_dbm); key.dptr; key = dbm_nextkey(self->di_dbm)) {
+ if (dbm_delete(self->di_dbm, key) < 0) {
+ dbm_clearerr(self->di_dbm);
+ PyErr_SetString(state->dbm_error, "cannot delete item from database");
+ return NULL;
+ }
+ }
+ if (self->di_size > 0) {
+ self->di_size = 0;
+ }
+ Py_RETURN_NONE;
+}
+
 static PyObject *
 dbm__enter__(PyObject *self, PyObject *args)
 {
@@ -431,6 +459,7 @@ static PyMethodDef dbm_methods[] = {
  _DBM_DBM_KEYS_METHODDEF
  _DBM_DBM_GET_METHODDEF
  _DBM_DBM_SETDEFAULT_METHODDEF
+ _DBM_DBM_CLEAR_METHODDEF
  {"__enter__", dbm__enter__, METH_NOARGS, NULL},
  {"__exit__", dbm__exit__, METH_VARARGS, NULL},
  {NULL, NULL} /* sentinel */