如何使用 SWIG 来 return 基于 C 函数 return 类型的特定异常?

How do I use SWIG to return specific exceptions based on C function return type?

以下是我的 C 中的函数示例 API:

int do_something(struct object *with_this)
struct object *get_something(struct object *from_this)

do_something 将 return 0 表示成功或 -1 并设置 errno 表示失败。 get_something 将 return 成功时的有效指针或 NULL 并设置 errno 失败。

我正在使用 SWIG 2.0 生成 python 绑定。对于 do_something 绑定,如果失败,我希望它 return 基于 errno 的异常,如果失败,则 return Python None 对象它成功了。对于 get_something 绑定,我希望它 return 一个基于 errno 的异常(如果失败)和不透明对象(如果成功)。

问题是如何让 SWIG 为我完成所有这些魔术?

目前我正在使用 SWIG 2.0。

我可以使用 %exception 并为每个符号定义不同的例外,但我希望它基于 return 类型。我会有很多 API 函数,我不想一一列举。所以我可以为每个符号执行此操作:

%exception do_something {
    $action
    if (result < 0) {
        return PyErr_SetFromErrno(PyExc_RuntimeError);
    }
}

%exception get_something {
    $action
    if (result == NULL) {
        return PyErr_SetFromErrno(PyExc_RuntimeError);
    }
}

如果我能做这样的事情会好很多(就像你可以用 %typemap 做的那样):

%exception int {
    $action
    if (result < 0) {
        return PyErr_SetFromErrno(PyExc_RuntimeError);
    }
}

%exception struct object * {
    $action
    if (result == NULL) {
        return PyErr_SetFromErrno(PyExc_RuntimeError);
    }
}

我可以做这样的事情吗?如果我有这样的东西,那么就会处理异常。我相信我的 get_something 绑定会很完美。

如果我使用 %typemap,我的 do_something 绑定仍然需要类似这样的东西才能将其发送到 return Python None 对象:

%typemap(out) int {
    $result = Py_BuildValue("");
}

我正在寻找的魔法是否存在,还是我做错了?有更好的方法吗?

为了帮助回答这个问题,我采用了您的函数并创建了以下 test.h 文件,其中包含足够的真实代码来实际说明问题:

struct object { int bar; };

static int do_something(struct object *with_this) { 
  errno = EACCES;
  return -1; 
}

static struct object *get_something(struct object *from_this) {
  errno = EAGAIN;
  return NULL;
}

我相当确定 %typemap(out) 是编写这段代码的地方,而不是 %exception,因为你想要的行为是由 return 类型决定的,而不是调用函数的名称。您还关心 return 的值这一事实进一步支持了这一点。您可以在某种程度上避免重复使用 $typemap 来引用您内部的其他类型映射。

所以我写了一个小的 SWIG 界面来说明这一点:

%module test

%{
#include <errno.h>
#include "test.h"
%}

%typemap(out) int do_something %{
  if ( < 0) {
    PyErr_SetFromErrno(PyExc_RuntimeError);
    SWIG_fail;    
  }
  $result = Py_None; // Return only controls exception
  Py_INCREF($result);
%}

%typemap(out) struct object *get_something %{
  if (!) {
    PyErr_SetFromErrno(PyExc_RuntimeError);
    SWIG_fail; 
  }

  // Return passed through unless NULL
  $typemap(out,_ltype);
%}

%include "test.h"

测试时有效:

Python 2.7.6 (default, Jun 22 2015, 17:58:13) 
[GCC 4.8.2] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import test
>>> test.do_something(None)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
RuntimeError: (13, 'Permission denied')
>>> test.get_something(None)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
RuntimeError: (11, 'Resource temporarily unavailable')
>>> 

这只是因为我在匹配类型映射时明确命名为 do_something,所以没有命名函数的回退是可以的。如果您只是尝试删除该限制,您将收到有关递归类型映射的错误。您可以使用 %apply 来解决这个问题,例如

%typemap(out) struct object * MY_NULL_OR_ERRNO %{
  if (!) {
    PyErr_SetFromErrno(PyExc_RuntimeError);
    SWIG_fail;
  }

  $typemap(out,_ltype);
%}

%apply struct object * MY_NULL_OR_ERRNO { struct object *get_something, struct object *some_other_function };

如果这太痛苦了,你当然可以手写类型映射:

%typemap(out) struct object * %{
  if (!) {
    PyErr_SetFromErrno(PyExc_RuntimeError);
    SWIG_fail;
  }

  $result = SWIG_NewPointerObj(SWIG_as_voidptr(), _descriptor, $owner);
%}

这似乎是最简单的选择,如果你真的想要 无处不在 struct object* 得到 returned。

还有其他可能的解决方案使用 %pythoncode%pythonappend,但在我看来,它们都不是对上述方案的真正改进。