blob: f43bc14ce1b623fc7c06c5b3314bb0478e2855c3 [file] [log] [blame]
Victor Stinnerb05cbac2017-01-17 03:46:13 +01001/*[clinic input]
2preserve
3[clinic start generated code]*/
4
5PyDoc_STRVAR(OrderedDict_fromkeys__doc__,
6"fromkeys($type, /, iterable, value=None)\n"
7"--\n"
8"\n"
Serhiy Storchaka78d9e582017-01-25 00:30:04 +02009"Create a new ordered dictionary with keys from iterable and values set to value.");
Victor Stinnerb05cbac2017-01-17 03:46:13 +010010
11#define ORDEREDDICT_FROMKEYS_METHODDEF \
Serhiy Storchaka4a934d42018-11-27 11:27:36 +020012 {"fromkeys", (PyCFunction)(void(*)(void))OrderedDict_fromkeys, METH_FASTCALL|METH_KEYWORDS|METH_CLASS, OrderedDict_fromkeys__doc__},
Victor Stinnerb05cbac2017-01-17 03:46:13 +010013
14static PyObject *
15OrderedDict_fromkeys_impl(PyTypeObject *type, PyObject *seq, PyObject *value);
16
17static PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +020018OrderedDict_fromkeys(PyTypeObject *type, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)
Victor Stinnerb05cbac2017-01-17 03:46:13 +010019{
20 PyObject *return_value = NULL;
21 static const char * const _keywords[] = {"iterable", "value", NULL};
Serhiy Storchaka31913912019-03-14 10:32:22 +020022 static _PyArg_Parser _parser = {NULL, _keywords, "fromkeys", 0};
23 PyObject *argsbuf[2];
24 Py_ssize_t noptargs = nargs + (kwnames ? PyTuple_GET_SIZE(kwnames) : 0) - 1;
Victor Stinnerb05cbac2017-01-17 03:46:13 +010025 PyObject *seq;
26 PyObject *value = Py_None;
27
Serhiy Storchaka31913912019-03-14 10:32:22 +020028 args = _PyArg_UnpackKeywords(args, nargs, NULL, kwnames, &_parser, 1, 2, 0, argsbuf);
29 if (!args) {
Victor Stinnerb05cbac2017-01-17 03:46:13 +010030 goto exit;
31 }
Serhiy Storchaka31913912019-03-14 10:32:22 +020032 seq = args[0];
33 if (!noptargs) {
34 goto skip_optional_pos;
35 }
36 value = args[1];
37skip_optional_pos:
Victor Stinnerb05cbac2017-01-17 03:46:13 +010038 return_value = OrderedDict_fromkeys_impl(type, seq, value);
39
40exit:
41 return return_value;
42}
43
44PyDoc_STRVAR(OrderedDict_setdefault__doc__,
45"setdefault($self, /, key, default=None)\n"
46"--\n"
47"\n"
Serhiy Storchaka78d9e582017-01-25 00:30:04 +020048"Insert key with a value of default if key is not in the dictionary.\n"
49"\n"
50"Return the value for key if key is in the dictionary, else default.");
Victor Stinnerb05cbac2017-01-17 03:46:13 +010051
52#define ORDEREDDICT_SETDEFAULT_METHODDEF \
Serhiy Storchaka4a934d42018-11-27 11:27:36 +020053 {"setdefault", (PyCFunction)(void(*)(void))OrderedDict_setdefault, METH_FASTCALL|METH_KEYWORDS, OrderedDict_setdefault__doc__},
Victor Stinnerb05cbac2017-01-17 03:46:13 +010054
55static PyObject *
56OrderedDict_setdefault_impl(PyODictObject *self, PyObject *key,
Serhiy Storchakaa70eaf22017-01-19 19:38:13 +020057 PyObject *default_value);
Victor Stinnerb05cbac2017-01-17 03:46:13 +010058
59static PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +020060OrderedDict_setdefault(PyODictObject *self, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)
Victor Stinnerb05cbac2017-01-17 03:46:13 +010061{
62 PyObject *return_value = NULL;
63 static const char * const _keywords[] = {"key", "default", NULL};
Serhiy Storchaka31913912019-03-14 10:32:22 +020064 static _PyArg_Parser _parser = {NULL, _keywords, "setdefault", 0};
65 PyObject *argsbuf[2];
66 Py_ssize_t noptargs = nargs + (kwnames ? PyTuple_GET_SIZE(kwnames) : 0) - 1;
Victor Stinnerb05cbac2017-01-17 03:46:13 +010067 PyObject *key;
Serhiy Storchakaa70eaf22017-01-19 19:38:13 +020068 PyObject *default_value = Py_None;
Victor Stinnerb05cbac2017-01-17 03:46:13 +010069
Serhiy Storchaka31913912019-03-14 10:32:22 +020070 args = _PyArg_UnpackKeywords(args, nargs, NULL, kwnames, &_parser, 1, 2, 0, argsbuf);
71 if (!args) {
Victor Stinnerb05cbac2017-01-17 03:46:13 +010072 goto exit;
73 }
Serhiy Storchaka31913912019-03-14 10:32:22 +020074 key = args[0];
75 if (!noptargs) {
76 goto skip_optional_pos;
77 }
78 default_value = args[1];
79skip_optional_pos:
Serhiy Storchakaa70eaf22017-01-19 19:38:13 +020080 return_value = OrderedDict_setdefault_impl(self, key, default_value);
Victor Stinnerb05cbac2017-01-17 03:46:13 +010081
82exit:
83 return return_value;
84}
85
86PyDoc_STRVAR(OrderedDict_popitem__doc__,
87"popitem($self, /, last=True)\n"
88"--\n"
89"\n"
Serhiy Storchaka78d9e582017-01-25 00:30:04 +020090"Remove and return a (key, value) pair from the dictionary.\n"
Victor Stinnerb05cbac2017-01-17 03:46:13 +010091"\n"
92"Pairs are returned in LIFO order if last is true or FIFO order if false.");
93
94#define ORDEREDDICT_POPITEM_METHODDEF \
Serhiy Storchaka4a934d42018-11-27 11:27:36 +020095 {"popitem", (PyCFunction)(void(*)(void))OrderedDict_popitem, METH_FASTCALL|METH_KEYWORDS, OrderedDict_popitem__doc__},
Victor Stinnerb05cbac2017-01-17 03:46:13 +010096
97static PyObject *
98OrderedDict_popitem_impl(PyODictObject *self, int last);
99
100static PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200101OrderedDict_popitem(PyODictObject *self, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100102{
103 PyObject *return_value = NULL;
104 static const char * const _keywords[] = {"last", NULL};
Serhiy Storchaka31913912019-03-14 10:32:22 +0200105 static _PyArg_Parser _parser = {NULL, _keywords, "popitem", 0};
106 PyObject *argsbuf[1];
107 Py_ssize_t noptargs = nargs + (kwnames ? PyTuple_GET_SIZE(kwnames) : 0) - 0;
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100108 int last = 1;
109
Serhiy Storchaka31913912019-03-14 10:32:22 +0200110 args = _PyArg_UnpackKeywords(args, nargs, NULL, kwnames, &_parser, 0, 1, 0, argsbuf);
111 if (!args) {
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100112 goto exit;
113 }
Serhiy Storchaka31913912019-03-14 10:32:22 +0200114 if (!noptargs) {
115 goto skip_optional_pos;
116 }
117 last = PyObject_IsTrue(args[0]);
118 if (last < 0) {
119 goto exit;
120 }
121skip_optional_pos:
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100122 return_value = OrderedDict_popitem_impl(self, last);
123
124exit:
125 return return_value;
126}
127
128PyDoc_STRVAR(OrderedDict_move_to_end__doc__,
129"move_to_end($self, /, key, last=True)\n"
130"--\n"
131"\n"
Serhiy Storchaka78d9e582017-01-25 00:30:04 +0200132"Move an existing element to the end (or beginning if last is false).\n"
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100133"\n"
Serhiy Storchaka78d9e582017-01-25 00:30:04 +0200134"Raise KeyError if the element does not exist.");
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100135
136#define ORDEREDDICT_MOVE_TO_END_METHODDEF \
Serhiy Storchaka4a934d42018-11-27 11:27:36 +0200137 {"move_to_end", (PyCFunction)(void(*)(void))OrderedDict_move_to_end, METH_FASTCALL|METH_KEYWORDS, OrderedDict_move_to_end__doc__},
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100138
139static PyObject *
140OrderedDict_move_to_end_impl(PyODictObject *self, PyObject *key, int last);
141
142static PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200143OrderedDict_move_to_end(PyODictObject *self, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100144{
145 PyObject *return_value = NULL;
146 static const char * const _keywords[] = {"key", "last", NULL};
Serhiy Storchaka31913912019-03-14 10:32:22 +0200147 static _PyArg_Parser _parser = {NULL, _keywords, "move_to_end", 0};
148 PyObject *argsbuf[2];
149 Py_ssize_t noptargs = nargs + (kwnames ? PyTuple_GET_SIZE(kwnames) : 0) - 1;
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100150 PyObject *key;
151 int last = 1;
152
Serhiy Storchaka31913912019-03-14 10:32:22 +0200153 args = _PyArg_UnpackKeywords(args, nargs, NULL, kwnames, &_parser, 1, 2, 0, argsbuf);
154 if (!args) {
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100155 goto exit;
156 }
Serhiy Storchaka31913912019-03-14 10:32:22 +0200157 key = args[0];
158 if (!noptargs) {
159 goto skip_optional_pos;
160 }
161 last = PyObject_IsTrue(args[1]);
162 if (last < 0) {
163 goto exit;
164 }
165skip_optional_pos:
Victor Stinnerb05cbac2017-01-17 03:46:13 +0100166 return_value = OrderedDict_move_to_end_impl(self, key, last);
167
168exit:
169 return return_value;
170}
Serhiy Storchaka31913912019-03-14 10:32:22 +0200171/*[clinic end generated code: output=8eb1296df9142908 input=a9049054013a1b77]*/