Python ctypes _Pointer类型注解运行报错问题求助

问题背景 #

尝试编写带类型注解的函数遍历C数组/指针，IDE能正确识别类型，但运行时触发TypeError: type '_ctypes._Pointer' is not subscriptable错误。代码示例如下：

from ctypes import *
from ctypes import _Pointer
from typing import TypeVar, Callable
T = TypeVar('T')

def foreach(array: _Pointer[T], cond: Callable[[T], bool]):
  i = 0
  v: T = array[i]
  while cond(v):
    yield v
    i += 1
    v = array[i]

# 示例
c_array = ...                               # _Pointer[int]
for i in foreach(c_array, lambda i: i < 10):
  print(i)                                  # 期望IDE识别i为int而非Any

错误原因是_ctypes._Pointer在运行时不是泛型类型，不支持下标语法_Pointer[T]，但类型检查工具（如VS Code的Pyright）能解析该语法。

解决方案 #

方法1：使用字符串类型注解 #

将类型注解用字符串包裹，运行时Python不会解析字符串内的下标操作，而类型检查工具会正确解析其中的泛型语法：

from ctypes import *
from typing import TypeVar, Callable

T = TypeVar('T')

# 用字符串包裹泛型注解，避免运行时下标求值
def foreach(array: '_Pointer[T]', cond: Callable[[T], bool]):
    i = 0
    v: T = array[i]
    while cond(v):
        yield v
        i += 1
        v = array[i]

# 实际使用示例
c_array = cast((c_int * 5)(1, 2, 3, 4, 10), POINTER(c_int))
for num in foreach(c_array, lambda x: x < 10):
    print(num)  # IDE会正确识别num为int类型

方法2：利用TYPE_CHECKING区分运行时与类型检查 #

TYPE_CHECKING是typing模块中的常量，仅在类型检查阶段为True，运行时为False。通过条件导入和类型别名，避免运行时处理泛型下标：

from ctypes import *
from typing import TypeVar, Callable, TYPE_CHECKING

T = TypeVar('T')

# 仅在类型检查时导入_Pointer并定义泛型别名
if TYPE_CHECKING:
    from ctypes import _Pointer
    PointerType = _Pointer[T]
else:
    # 运行时直接使用_Pointer类型
    PointerType = _Pointer

def foreach(array: PointerType, cond: Callable[[T], bool]):
    i = 0
    v: T = array[i]
    while cond(v):
        yield v
        i += 1
        v = array[i]

方法3：用Protocol定义指针协议 #

定义一个支持下标访问的协议，让类型检查器识别ctypes指针的行为，同时避免运行时类型错误：

from ctypes import *
from typing import Protocol, TypeVar, Callable

T = TypeVar('T')

# 定义符合ctypes指针行为的协议
class CPointer(Protocol[T]):
    def __getitem__(self, index: int) -> T: ...

def foreach(array: CPointer[T], cond: Callable[[T], bool]):
    i = 0
    v: T = array[i]
    while cond(v):
        yield v
        i += 1
        v = array[i]

这种方式更灵活，任何实现了__getitem__返回对应类型的对象都可以传入，ctypes指针天然满足该协议。

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