Question

基本上，我在Python中包装一个C API（使用CFFI（很棒））。

我在C API中有一个枚举：

typedef enum
{
    enum1 = value1;
    enum2 = value2;
    ...
} enumSomething;

在Python中，我收到了该枚举中变量的名称，字符串为："enum1"。我想得到value1。

使用CFFI，您可以包装枚举并轻松访问其成员（here is an example）。我的问题是我有代表变量名称的字符串，而不是变量本身。

因此，cwrapper.c."enum1"不起作用（cwrapper是包装模块的名称，c包装库时ffi.dlopen(libname)返回的名称）。

我可以创建一个字典，将变量名称作为字符串映射到实变量，例如：

enum = { "enum1" : cwrapper.c.enum1, "enum2" : cwrapper.c.enum2, ... }

但是在这一点上，我最好将枚举重写为Python字典。

我知道我也可以使用正则表达式来读取C文件，但我认为还有更好的方法来处理它，我没有想到。有什么想法吗？

Answer 1

cwrapper.c."enum1"在Python中通常不是有效的语法。您需要使用getattr：

>>> getattr(cwrapper.c, "enum1")
42

然而，更简洁的方法是使用typeof获取实际类型;然后得到它的relements，这是一本字典！

>>> import cffi
>>> ffi = cffi.FFI()
>>> ffi.cdef('typedef enum {RANDOM, IMMEDIATE, SEARCH} strategy;')
>>> typeof_strategy = ffi.typeof('strategy')
>>> typeof_strategy.relements
{'RANDOM': 0, 'SEARCH': 2, 'IMMEDIATE': 1}
>>> typeof_strategy.relements['SEARCH']
2

Answer 2

在Antti的答案的基础上，可以构建一个合适的Python Enum：

Strategy = enum.Enum('Strategy', ffi.typeof('strategy').relements)

list(Strategy)
# [<Strategy.RANDOM: 0>, <Strategy.IMMEDIATE: 1>, <Strategy.SEARCH: 2>]

使用Python中的名称（Python字符串）获取枚举中的C变量值

2 个答案: