blob: e54c69cf708294853770b2f81abe7294b10ebb7b [file] [log] [blame]
Serhiy Storchaka1009bf12015-04-03 23:53:51 +03001/*[clinic input]
2preserve
3[clinic start generated code]*/
4
Serhiy Storchaka9260e772015-04-17 21:05:18 +03005PyDoc_STRVAR(_dbm_dbm_close__doc__,
6"close($self, /)\n"
7"--\n"
8"\n"
9"Close the database.");
10
11#define _DBM_DBM_CLOSE_METHODDEF \
12 {"close", (PyCFunction)_dbm_dbm_close, METH_NOARGS, _dbm_dbm_close__doc__},
13
14static PyObject *
15_dbm_dbm_close_impl(dbmobject *self);
16
17static PyObject *
18_dbm_dbm_close(dbmobject *self, PyObject *Py_UNUSED(ignored))
19{
20 return _dbm_dbm_close_impl(self);
21}
22
23PyDoc_STRVAR(_dbm_dbm_keys__doc__,
24"keys($self, /)\n"
25"--\n"
26"\n"
27"Return a list of all keys in the database.");
28
29#define _DBM_DBM_KEYS_METHODDEF \
30 {"keys", (PyCFunction)_dbm_dbm_keys, METH_NOARGS, _dbm_dbm_keys__doc__},
31
32static PyObject *
33_dbm_dbm_keys_impl(dbmobject *self);
34
35static PyObject *
36_dbm_dbm_keys(dbmobject *self, PyObject *Py_UNUSED(ignored))
37{
38 return _dbm_dbm_keys_impl(self);
39}
40
41PyDoc_STRVAR(_dbm_dbm_get__doc__,
Serhiy Storchaka2e38cc32018-04-29 12:38:06 +030042"get($self, key, default=None, /)\n"
Serhiy Storchaka1009bf12015-04-03 23:53:51 +030043"--\n"
44"\n"
45"Return the value for key if present, otherwise default.");
46
Serhiy Storchaka9260e772015-04-17 21:05:18 +030047#define _DBM_DBM_GET_METHODDEF \
Serhiy Storchaka4a934d42018-11-27 11:27:36 +020048 {"get", (PyCFunction)(void(*)(void))_dbm_dbm_get, METH_FASTCALL, _dbm_dbm_get__doc__},
Serhiy Storchaka1009bf12015-04-03 23:53:51 +030049
50static PyObject *
Serhiy Storchaka9260e772015-04-17 21:05:18 +030051_dbm_dbm_get_impl(dbmobject *self, const char *key,
52 Py_ssize_clean_t key_length, PyObject *default_value);
Serhiy Storchaka1009bf12015-04-03 23:53:51 +030053
54static PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +020055_dbm_dbm_get(dbmobject *self, PyObject *const *args, Py_ssize_t nargs)
Serhiy Storchaka1009bf12015-04-03 23:53:51 +030056{
57 PyObject *return_value = NULL;
58 const char *key;
59 Py_ssize_clean_t key_length;
Serhiy Storchaka2e38cc32018-04-29 12:38:06 +030060 PyObject *default_value = Py_None;
Serhiy Storchaka1009bf12015-04-03 23:53:51 +030061
Sylvain74453812017-06-10 06:51:48 +020062 if (!_PyArg_ParseStack(args, nargs, "s#|O:get",
63 &key, &key_length, &default_value)) {
Victor Stinner259f0e42017-01-17 01:35:17 +010064 goto exit;
65 }
Serhiy Storchaka9260e772015-04-17 21:05:18 +030066 return_value = _dbm_dbm_get_impl(self, key, key_length, default_value);
67
68exit:
69 return return_value;
70}
71
72PyDoc_STRVAR(_dbm_dbm_setdefault__doc__,
73"setdefault($self, key, default=b\'\', /)\n"
74"--\n"
75"\n"
76"Return the value for key if present, otherwise default.\n"
77"\n"
78"If key is not in the database, it is inserted with default as the value.");
79
80#define _DBM_DBM_SETDEFAULT_METHODDEF \
Serhiy Storchaka4a934d42018-11-27 11:27:36 +020081 {"setdefault", (PyCFunction)(void(*)(void))_dbm_dbm_setdefault, METH_FASTCALL, _dbm_dbm_setdefault__doc__},
Serhiy Storchaka9260e772015-04-17 21:05:18 +030082
83static PyObject *
84_dbm_dbm_setdefault_impl(dbmobject *self, const char *key,
85 Py_ssize_clean_t key_length,
86 PyObject *default_value);
87
88static PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +020089_dbm_dbm_setdefault(dbmobject *self, PyObject *const *args, Py_ssize_t nargs)
Serhiy Storchaka9260e772015-04-17 21:05:18 +030090{
91 PyObject *return_value = NULL;
92 const char *key;
93 Py_ssize_clean_t key_length;
94 PyObject *default_value = NULL;
95
Sylvain74453812017-06-10 06:51:48 +020096 if (!_PyArg_ParseStack(args, nargs, "s#|O:setdefault",
97 &key, &key_length, &default_value)) {
Victor Stinner259f0e42017-01-17 01:35:17 +010098 goto exit;
99 }
Serhiy Storchaka9260e772015-04-17 21:05:18 +0300100 return_value = _dbm_dbm_setdefault_impl(self, key, key_length, default_value);
Serhiy Storchaka1009bf12015-04-03 23:53:51 +0300101
102exit:
103 return return_value;
104}
105
106PyDoc_STRVAR(dbmopen__doc__,
107"open($module, filename, flags=\'r\', mode=0o666, /)\n"
108"--\n"
109"\n"
110"Return a database object.\n"
111"\n"
112" filename\n"
113" The filename to open.\n"
114" flags\n"
115" How to open the file. \"r\" for reading, \"w\" for writing, etc.\n"
116" mode\n"
117" If creating a new file, the mode bits for the new file\n"
118" (e.g. os.O_RDWR).");
119
120#define DBMOPEN_METHODDEF \
Serhiy Storchaka4a934d42018-11-27 11:27:36 +0200121 {"open", (PyCFunction)(void(*)(void))dbmopen, METH_FASTCALL, dbmopen__doc__},
Serhiy Storchaka1009bf12015-04-03 23:53:51 +0300122
123static PyObject *
Serhiy Storchaka6f600ff2018-02-26 16:02:22 +0200124dbmopen_impl(PyObject *module, PyObject *filename, const char *flags,
Larry Hastings89964c42015-04-14 18:07:59 -0400125 int mode);
Serhiy Storchaka1009bf12015-04-03 23:53:51 +0300126
127static PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200128dbmopen(PyObject *module, PyObject *const *args, Py_ssize_t nargs)
Serhiy Storchaka1009bf12015-04-03 23:53:51 +0300129{
130 PyObject *return_value = NULL;
Serhiy Storchaka6f600ff2018-02-26 16:02:22 +0200131 PyObject *filename;
Serhiy Storchaka1009bf12015-04-03 23:53:51 +0300132 const char *flags = "r";
133 int mode = 438;
134
Serhiy Storchaka4fa95912019-01-11 16:01:14 +0200135 if (!_PyArg_CheckPositional("open", nargs, 1, 3)) {
Victor Stinner259f0e42017-01-17 01:35:17 +0100136 goto exit;
137 }
Serhiy Storchaka4fa95912019-01-11 16:01:14 +0200138 if (!PyUnicode_Check(args[0])) {
139 _PyArg_BadArgument("open", 1, "str", args[0]);
140 goto exit;
141 }
142 if (PyUnicode_READY(args[0]) == -1) {
143 goto exit;
144 }
145 filename = args[0];
146 if (nargs < 2) {
147 goto skip_optional;
148 }
149 if (!PyUnicode_Check(args[1])) {
150 _PyArg_BadArgument("open", 2, "str", args[1]);
151 goto exit;
152 }
153 Py_ssize_t flags_length;
154 flags = PyUnicode_AsUTF8AndSize(args[1], &flags_length);
155 if (flags == NULL) {
156 goto exit;
157 }
158 if (strlen(flags) != (size_t)flags_length) {
159 PyErr_SetString(PyExc_ValueError, "embedded null character");
160 goto exit;
161 }
162 if (nargs < 3) {
163 goto skip_optional;
164 }
165 if (PyFloat_Check(args[2])) {
166 PyErr_SetString(PyExc_TypeError,
167 "integer argument expected, got float" );
168 goto exit;
169 }
170 mode = _PyLong_AsInt(args[2]);
171 if (mode == -1 && PyErr_Occurred()) {
172 goto exit;
173 }
174skip_optional:
Serhiy Storchaka1009bf12015-04-03 23:53:51 +0300175 return_value = dbmopen_impl(module, filename, flags, mode);
176
177exit:
178 return return_value;
179}
Serhiy Storchaka4fa95912019-01-11 16:01:14 +0200180/*[clinic end generated code: output=7f5d30ef5d820b8a input=a9049054013a1b77]*/