如何使用SWIG根据C函数返回类型返回特定的异常?

时间:2016-06-29 02:10:51

标签: python c swig

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

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

do_something将返回0表示成功,或-1并将errno设置为失败。 get_something会在成功或NULL上返回有效指针,并将errno设置为失败。

我正在使用SWIG 2.0来生成python绑定。对于do_something绑定,如果失败则返回基于errno的异常,如果成功则返回Python None对象。对于get_something绑定,如果失败则返回基于errno的异常,如果成功则返回不透明对象。

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

目前我正在使用SWIG 2.0。

我可以使用%exception并为每个符号定义不同的异常,但我希望它基于返回类型。我将有很多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约束力是完美的。

如果我使用do_something,我的%typemap绑定仍然需要这样的东西让它返回Python None对象:

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

我正在寻找的魔法是存在的还是我认为这一切都错了?有更好的方法吗?

1 个答案:

答案 0 :(得分:0)

为了帮助回答这个问题,我接受了你的功能并创建了以下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,因为您想要的行为取决于返回类型,而不是函数的名称调用。你也关心回报价值的事实进一步支持了这一点。您可以在某种程度上避免使用$typemap重复引用您内部的其他类型图。

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

%module test

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

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

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

  // Return passed through unless NULL
  $typemap(out,$1_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')
>>> 

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

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

  $typemap(out,$1_ltype);
%}

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

如果这太痛苦了,你当然可以手工编写类型图:

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

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

如果您真的希望无处不在 struct object*,那么这似乎是最简单的选择。

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