在python的C api中解析unsigned int(uint32_t)

时间:2015-07-10 10:19:18

标签: python c python-2.x cpython

如果我写一个函数接受一个无符号整数(0 - 0xFFFFFFFF),我可以使用:

uint32_t myInt;
if(!PyArg_ParseTuple(args, "I", &myInt))
    return NULL;

然后从python,我可以传递intlong

但是如果我传递了一个整数列表呢?

uint32_t* myInts;
PyObject* pyMyInts;
PyArg_ParseTuple(args, "O", &pyMyInts);

if (PyList_Check(intsObj)) {
    size_t n = PyList_Size(v);
    myInts = calloc(n, sizeof(*myInts));

    for(size_t i = 0; i < n; i++) {
        PyObject* item = PyList_GetItem(pyMyInts, i);

        // What function do I want here?
        if(!GetAUInt(item, &myInts[i]))
            return NULL;
    }
}

// cleanup calloc'd array on exit, etc

具体来说,我的问题是处理:

  • 包含intlong s
  • 混合的列表
  • 在分配到uint32时检测到溢出

1 个答案:

答案 0 :(得分:1)

您可以创建一个元组并使用与单个参数相同的方法。在C方面,元组对象并不是真正不可变的,因此不会有太多麻烦。

同样Documentation也适合你。它接受int和long对象,否则会引发错误。但是如果sizeof(long)大于4,您可能需要自己检查上限溢出。

static int 
GetAUInt(PyObject *pylong, uint32_t *myint) {
    static unsigned long MAX = 0xffffffff;
    unsigned long l = PyLong_AsUnsignedLong(pylong);

    if (l == -1 && PyErr_Occurred() || l > MAX) {
        PyErr_SetString(PyExc_OverflowError, "can't convert to uint32_t");
        return false;
    }

    *myint = (uint32_t) l;
    return true;
}