在Python C API中，如果没有设置第二个参数怎么能默认为第一个的值？

问题描述

我有一个 x, y 参数。 使用 python C API，如果 y 为 None，我如何将 y 默认为 x 的值。

python 等效项是：

def scale(x: float, y:Optional[float]=None):
    if y is None:
       y = x

    c_scale_api(x, y)

您可以使用| 到 state 解析 arg 元组时，一些 arguments 是可选的。 您必须确保|之后的任何 arguments 在使用它们之前总是有一个值。 也就是说，如果参数不存在， PyArg_ParseTuple将保持变量不变。 通常，这意味着您应该在调用PyArg_ParseTuple之前分配默认值。 但是，由于您的要求，我们可以检查我们实际获得了多少个参数，然后如果只给出 1 个参数，则将其分配给y 。

此代码将使第二个参数可选，但不完全等同于 python function。 也就是说None不是第二个参数的有效值。 所以scale(1, 2)和scale(1)都很好，但是scale(1, None)会失败。

static PyObject*
scale(PyObject* self, PyObject* args) // type: METH_VARARGS
{
    double x;
    double y;
    if (!PyArg_ParseTuple(args, "d|d", &x, &y)) {
        return NULL;
    }
    if (PyTuple_GET_SIZE(args) == 1) {
        y = x;
    }
    return PyFloat_FromDouble(x + y);
}

对于完全等效的 C function，请尝试以下操作。 现在像scale(x=1, y=None)这样的东西也会起作用。

static PyObject* 
scale(PyObject* self, PyObject* args, PyObject* kwargs) // type: METH_VARARGS | METH_KEYWORDS
{
    static char* keywords[] = { "x", "y", NULL };

    double x;
    double y;
    PyObject* y_obj;

    if (
        !PyArg_ParseTupleAndKeywords(
            args, kwargs, "d|O:scale", keywords, &x, &y_obj
        )
     ) {
        return NULL;
    }
    if (y_obj == Py_None) {
        y = x;
    } else if (PyFloat_Check(y_obj)) {
        y = PyFloat_AS_DOUBLE(y_obj);
    } else if (PyLong_Check(y_obj)) {
        y = PyLong_AS_LONG(y_obj);
    } else {
        PyErr_SetString(PyExc_TypeError, "y must be float | int | None");
        return NULL;
    }

    return PyFloat_FromDouble(x + y);
}