如何从 C 扩展返回 Python 枚举？

我正在为 Python 编写一个封装 C 库的 C 扩展。 C 库有一些枚举类型，我已经编写了相应的IntEnums 用于这些，例如：

from enum import IntEnum
# _enum_consts is a C extension creating Python
# constants for the library's enum constants
from . import _enum_consts

class Align(IntEnum):
    DEFAULT = _enum_consts.MD_ALIGN_DEFAULT
    LEFT = _enum_consts.MD_ALIGN_LEFT
    CENTER = _enum_consts.MD_ALIGN_CENTER
    RIGHT = _enum_consts.MD_ALIGN_RIGHT

现在，在我包装库主要功能的 C 扩展中，一些库函数返回枚举类型，我想将它们转换为 PythonIntEnum价值观。 （我意识到我可以简单地返回int直接进行比较，他们会成功地与IntEnums，但这有点违背了目的。）

问题是，我如何创建IntEnum要返回的值？我能找到的只是将 C 枚举常量转换为 Python 常量的示例PyModule_AddIntConstant()（这正是_enum_consts扩展模块可以）。

首先，模块IntEnum需要导入到扩展模块中（如果您还没有这样做）。在模块初始化函数中：

PyMODINIT_FUNC PyInit_someextension(void)
{
    // ...

    PyObject *m;
    m = PyModule_Create(&someextension_module);
    if (m == NULL) {
        return NULL;
    }

    // ...

    // Import the enums module, where "somepackage.enums"
    // is the full name of the enums module
    PyObject *enums = PyImport_ImportModule("somepackage.enums");
    if (enums == NULL) {
        Py_DECREF(m);
        return NULL;
    }
    Py_DECREF(enums);

    return m;
}

然后，在你想要实例化的地方IntEnumvalue，首先再次获取它所在的模块：

PyObject *enums = PyImport_AddModule("somepackage.enums");
if (enums == NULL) {
    // Return NULL, -1, etc. to signal an exception
}

之后，立即获取您的IntEnum type (Align在本例中）来自模块：

PyObject *align_enum = PyObject_GetAttrString(enums, "Align");
if (align_enum == NULL) {
    // Return NULL, -1, etc. to signal an exception
}

然后，您可以创建您需要的实例。align这将是从中获取的值IntEnum。我们通过调用类型对象来做到这一点，就像在常规 Python 中一样（请注意PyObject_CallFunction实际上并不限于调用函数——它可以调用任何可调用对象）：

PyObject *align_instance = PyObject_CallFunction(align_enum, "(i)", align);
if (align_instance == NULL) {
    Py_DECREF(align_enum);
    // Return NULL, -1, etc. to signal an exception
}

Now, align_instance包含IntEnum要返回的实例。务必Py_DECREF(align_enum)返回之前。