如何在python中使用枚举?

时间:2013-11-19 14:06:21

标签: python enums swig

我有一个枚举声明如下:

typedef enum mail_ {
 Out = 0,
 Int = 1,
 Spam = 2
} mail;

功能:

mail status;
int fill_mail_data(int i, &status);

在上面的函数中,status被填满并将发送。

当我通过swig尝试这个时,我面临以下问题:

  1. 没有显示mail的详细信息。当我尝试打印mail.__doc__help(mail)时,它会抛出错误,说没有这样的属性,尽管我可以使用这些值(SpamIn ,和Out)。
  2. 如上所示,Swig不知道main是什么,因此它不接受mail的任何函数参数。

1 个答案:

答案 0 :(得分:5)

对于SWIG,enum只是一个整数。要将它用作示例中的输出参数,您还可以将参数声明为输出参数,如下所示:

%module x

// Declare "mail* status" as an output parameter.
// It will be returned along with the return value of a function
// as a tuple if necessary, and will not be required as a function
// parameter.
%include <typemaps.i>
%apply int *OUTPUT {mail* status};

%inline %{

typedef enum mail_ {
    Out  = 0,
    Int  = 1,
    Spam = 2
} mail;

int fill_mail_data(int i, mail* status)
{
    *status = Spam;
    return i+1;
}

%}

使用:

>>> import x
>>> dir(x)    # Note no "mail" object, just Int, Out, Spam which are ints.
['Int', 'Out', 'Spam', '__builtins__', '__cached__', '__doc__', '__file__', '__initializing__', '__loader__', '__name__', '__package__', '_newclass', '_object', '_swig_getattr', '_swig_property', '_swig_repr', '_swig_setattr', '_swig_setattr_nondynamic', '_x', 'fill_mail_data']
>>> x.fill_mail_data(5)
[6, 2]
>>> ret,mail = x.fill_mail_data(5)
>>> mail == x.Spam
True