bpo-40676: Use Argument Clinic for csv (where possible) (GH-20200)

1 files changed, 134 insertions, 0 deletions

diff --git a/Modules/clinic/_csv.c.h b/Modules/clinic/_csv.c.h
new file mode 100644
index 0000000..41c0cdd
--- /dev/null
+++ b/Modules/clinic/_csv.c.h
@@ -0,0 +1,134 @@
+/*[clinic input]
+preserve
+[clinic start generated code]*/
+
+PyDoc_STRVAR(_csv_list_dialects__doc__,
+"list_dialects($module, /)\n"
+"--\n"
+"\n"
+"Return a list of all known dialect names.\n"
+"\n"
+"    names = csv.list_dialects()");
+
+#define _CSV_LIST_DIALECTS_METHODDEF    \
+    {"list_dialects", (PyCFunction)_csv_list_dialects, METH_NOARGS, _csv_list_dialects__doc__},
+
+static PyObject *
+_csv_list_dialects_impl(PyObject *module);
+
+static PyObject *
+_csv_list_dialects(PyObject *module, PyObject *Py_UNUSED(ignored))
+{
+    return _csv_list_dialects_impl(module);
+}
+
+PyDoc_STRVAR(_csv_unregister_dialect__doc__,
+"unregister_dialect($module, /, name)\n"
+"--\n"
+"\n"
+"Delete the name/dialect mapping associated with a string name.\n"
+"\n"
+"    csv.unregister_dialect(name)");
+
+#define _CSV_UNREGISTER_DIALECT_METHODDEF    \
+    {"unregister_dialect", (PyCFunction)(void(*)(void))_csv_unregister_dialect, METH_FASTCALL|METH_KEYWORDS, _csv_unregister_dialect__doc__},
+
+static PyObject *
+_csv_unregister_dialect_impl(PyObject *module, PyObject *name);
+
+static PyObject *
+_csv_unregister_dialect(PyObject *module, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)
+{
+    PyObject *return_value = NULL;
+    static const char * const _keywords[] = {"name", NULL};
+    static _PyArg_Parser _parser = {NULL, _keywords, "unregister_dialect", 0};
+    PyObject *argsbuf[1];
+    PyObject *name;
+
+    args = _PyArg_UnpackKeywords(args, nargs, NULL, kwnames, &_parser, 1, 1, 0, argsbuf);
+    if (!args) {
+        goto exit;
+    }
+    name = args[0];
+    return_value = _csv_unregister_dialect_impl(module, name);
+
+exit:
+    return return_value;
+}
+
+PyDoc_STRVAR(_csv_get_dialect__doc__,
+"get_dialect($module, /, name)\n"
+"--\n"
+"\n"
+"Return the dialect instance associated with name.\n"
+"\n"
+"    dialect = csv.get_dialect(name)");
+
+#define _CSV_GET_DIALECT_METHODDEF    \
+    {"get_dialect", (PyCFunction)(void(*)(void))_csv_get_dialect, METH_FASTCALL|METH_KEYWORDS, _csv_get_dialect__doc__},
+
+static PyObject *
+_csv_get_dialect_impl(PyObject *module, PyObject *name);
+
+static PyObject *
+_csv_get_dialect(PyObject *module, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)
+{
+    PyObject *return_value = NULL;
+    static const char * const _keywords[] = {"name", NULL};
+    static _PyArg_Parser _parser = {NULL, _keywords, "get_dialect", 0};
+    PyObject *argsbuf[1];
+    PyObject *name;
+
+    args = _PyArg_UnpackKeywords(args, nargs, NULL, kwnames, &_parser, 1, 1, 0, argsbuf);
+    if (!args) {
+        goto exit;
+    }
+    name = args[0];
+    return_value = _csv_get_dialect_impl(module, name);
+
+exit:
+    return return_value;
+}
+
+PyDoc_STRVAR(_csv_field_size_limit__doc__,
+"field_size_limit($module, /, new_limit=<unrepresentable>)\n"
+"--\n"
+"\n"
+"Sets an upper limit on parsed fields.\n"
+"\n"
+"    csv.field_size_limit([limit])\n"
+"\n"
+"Returns old limit. If limit is not given, no new limit is set and\n"
+"the old limit is returned");
+
+#define _CSV_FIELD_SIZE_LIMIT_METHODDEF    \
+    {"field_size_limit", (PyCFunction)(void(*)(void))_csv_field_size_limit, METH_FASTCALL|METH_KEYWORDS, _csv_field_size_limit__doc__},
+
+static PyObject *
+_csv_field_size_limit_impl(PyObject *module, PyObject *new_limit);
+
+static PyObject *
+_csv_field_size_limit(PyObject *module, PyObject *const *args, Py_ssize_t nargs, PyObject *kwnames)
+{
+    PyObject *return_value = NULL;
+    static const char * const _keywords[] = {"new_limit", NULL};
+    static _PyArg_Parser _parser = {NULL, _keywords, "field_size_limit", 0};
+    PyObject *argsbuf[1];
+    Py_ssize_t noptargs = nargs + (kwnames ? PyTuple_GET_SIZE(kwnames) : 0) - 0;
+    PyObject *new_limit = NULL;
+
+    args = _PyArg_UnpackKeywords(args, nargs, NULL, kwnames, &_parser, 0, 1, 0, argsbuf);
+    if (!args) {
+        goto exit;
+    }
+    if (!noptargs) {
+        goto skip_optional_pos;
+    }
+    new_limit = args[0];
+skip_optional_pos:
+    return_value = _csv_field_size_limit_impl(module, new_limit);
+
+exit:
+    return return_value;
+}
+/*[clinic end generated code: output=d51f3c42e9b88802 input=a9049054013a1b77]*/