如何为ctypes.POINTER(ctypes.c_int)添加类型提示?mypy报错求解
正确为ctypes.POINTER(ctypes.c_int)添加类型提示的方法
问题核心在于ctypes.POINTER是一个工厂函数,而非直接可用的类型,所以mypy会报错它不能作为类型使用。以下是两种可行的解决方案:
方案1:直接使用ctypes内部的泛型指针类型
修改函数的返回类型注解,用ctypes._Pointer[ctypes.c_int]替代原有的ctypes.POINTER(ctypes.c_int)。_Pointer[T]是mypy识别的、用来表示POINTER(T)创建的指针实例的实际类型:
import ctypes import numpy as np # type: ignore # (no stubs for numpy) def np_to_c(arr: np.ndarray) -> tuple[ctypes._Pointer[ctypes.c_int], ctypes.c_int]: return arr.ctypes.data_as(ctypes.POINTER(ctypes.c_int)), ctypes.c_int(len(arr))
方案2:定义类型别名提升代码可读性
如果觉得_Pointer的下划线不够直观,可以先定义一个类型别名,再在注解中使用:
import ctypes from typing import TypeAlias import numpy as np # type: ignore # (no stubs for numpy) # 定义类型别名,明确表示指向c_int的指针 CIntPointer: TypeAlias = ctypes._Pointer[ctypes.c_int] def np_to_c(arr: np.ndarray) -> tuple[CIntPointer, ctypes.c_int]: return arr.ctypes.data_as(ctypes.POINTER(ctypes.c_int)), ctypes.c_int(len(arr))
这两种方案都能通过mypy的类型检查,同时准确表达返回值的类型。
内容的提问来源于stack exchange,提问作者FluidMechanics Potential Flows
相关产品推荐
相关产品推荐

