Python C++ api - 在函数重载中返回不同的类型答案

【问题标题】：Python C++ api - return different types in function overloadingPython C++ api - 在函数重载中返回不同的类型
【发布时间】：2019-05-17 15:55:01
【问题描述】：

我有一个 C++ 类，它包含两个重载方法 sayHi()，每个方法都有不同的返回类型：

class Box {
   public:

      Box();

      void sayHi(char *name);

      int sayHi(int number);

};

当我尝试包装这个函数时，我遵循了ShadowRanger 的建议并 “编写一个单独的包装函数，对提供的参数进行类型检查并分派到正确的‘真实’函数”。

这里是：

static PyObject *pyBox_sayHi_char(PyBox *self, char *Name)
{
    self->bx->sayHi(Name);

    Py_RETURN_NONE;
}

static int *pyBox_sayHi_int(PyBox *self,  int number)
{
    int answer;

    answer = self->bx->sayHi(number);

    return answer;
}

static PyObject *Hi_overload_switch(PyBox *self,  PyObject *args)
{
    PyObject *x;

    if (!PyArg_ParseTuple(args, "O", &x))
        return NULL;

    if (PyUnicode_Check(x))
    {
        const char* s = PyBytes_AsString(PyUnicode_AsUTF8String(x)); // convert PyObject to char*

        Py_DECREF(x);

        return pyBox_sayHi_char(self, s);
    }
    if (PyLong_Check(x))
    {   
        return pyBox_sayHi_int( self, PyLong_AsLong(x) );
    }
    Py_RETURN_NOTIMPLEMENTED;
}

以及以下方法表：

static PyMethodDef pyBox_methods[] = {
    {"Hi", (PyCFunction)Hi_overload_switch, METH_VARARGS, "Hi"},
    {NULL, NULL, 0, NULL}
};

但是，我收到了错误：

error: cannot convert ‘int*’ to ‘PyObject*’ {aka ‘_object*’} in return
         return pyBox_sayHi_int( self, PyLong_AsLong(x) );

AFAIK，我不想将返回类型转换为 PyObjects，因为重载的方法可能需要返回自定义类型。因此，我的示例中的 int vs void 仅用于说明目的 - 我可能会写 PyBanana vs PyOrange数据类型。

我的问题是，如何让函数 Hi_overload_switch 返回不同的类型？

谢谢！

【问题讨论】：

标签： python c++ overloading python-c-api

【解决方案1】：

您需要返回一个 Python int。在 C/C++ 源代码中，这是 PyLongObject（它是 PyObject 的子类型，因此您可以将 PyLongObject* 转换为 PyObject* 并返回）。要将 C++ int 转换为 PyObject* Python int，请使用 PyObject* PyLong_FromLong(long)：

static PyObject *pyBox_sayHi_int(PyBox *self,  int number)
{
    int answer;

    answer = self->bx->sayHi(number);

    return PyLong_FromLong(answer);
}

【讨论】：