如何在Python脚本中向C库传递typedef enum类型参数？

C库定义 #

defs.h 中的枚举定义 #

typedef enum _CustomParam
{
    OptionOne,  /// 默认值 0
    OptionTwo,  /// 默认值 1（注：原代码注释写Default 0有误，枚举值默认递增，OptionTwo实际为1）
} CustomParam;

lib.h 中的函数声明 #

extern "C"
{
    /// @param[in] 要获取值的参数
    /// @return 参数的值
    /// @retval -1 获取失败
    LIB_EXPORT double getParam(CustomParam param);

    /// @param[in] param 要设置的参数
    /// @param[in] val 要设置的参数值
    /// @retval 0 设置成功
    /// @retval -1 设置失败
    LIB_EXPORT int setParam(CustomParam param, double value);
}

现有Python调用代码 #

def.py #

from ctypes import *

class CustomParam(c_int):
    OptionOne = 0
    OptionTwo = 1

main.py #

from .lib import Lib
library = Lib("lib/lib.dll")
library.init()

lib.py #

from def.h import *  # 此处路径错误，应导入Python模块而非C头文件
def __init__(self, libpath):
    super().__init__()
    self.lib = CDLL(libpath)

def init(self):
    self.lib.getParam.argtypes = [CustomParam]
    self.lib.getParam.restype = c_double
    result = self.lib.getParam(CustomParam.OptionOne)
    print("Parameter: ", result)

问题现象 #

调用getParam或setParam时返回值始终为-1，所有接收CustomParam类型参数的函数均调用失败，但接收通用类型（如c_int）的函数可正常工作。尝试传入c_int(0)或直接传整数0，问题依旧。

解决方案 #

1. 修正枚举类型定义 #

C语言的typedef enum本质是整数类型的别名，无需在Python中继承c_int定义类，直接用简单类封装常量即可：

# def.py
class CustomParam:
    OptionOne = 0
    OptionTwo = 1

2. 修正函数参数类型声明 #

在ctypes中，需将argtypes指定为C枚举对应的底层类型（通常是c_int），而非自定义的Python类：

# lib.py
from ctypes import CDLL, c_double, c_int
from .def import CustomParam

class Lib:
    def __init__(self, libpath):
        self.lib = CDLL(libpath)

    def init(self):
        # 配置getParam的参数和返回值类型
        self.lib.getParam.argtypes = [c_int]
        self.lib.getParam.restype = c_double
        # 传入枚举对应的整数值
        result = self.lib.getParam(CustomParam.OptionOne)
        print("Parameter: ", result)

        # 同理配置setParam
        self.lib.setParam.argtypes = [c_int, c_double]
        self.lib.setParam.restype = c_int
        set_result = self.lib.setParam(CustomParam.OptionTwo, 10.5)
        print("Set result: ", set_result)

3. 排查其他潜在问题 #

• 确认DLL路径正确：lib/lib.dll为相对路径，需保证脚本运行时能定位到该文件
• 检查C库导出：确保LIB_EXPORT宏正确定义，getParam和setParam已被成功导出
• 验证C库初始化：部分C库需要先调用特定初始化函数才能正常访问参数，确认是否遗漏相关步骤
• 检查枚举值合法性：确认C库实现中是否对传入的枚举值有范围校验，避免传入未定义的数值

核心原理 #

C枚举在编译后会被处理为整数类型（通常是int），因此Python调用时只需传递对应的整数值，并将argtypes设置为c_int即可，ctypes会自动完成整数到Cint的类型转换，无需自定义继承c_int的类。

内容的提问来源于stack exchange，提问作者ChrizZlyBear