如何使模板函数指针适配首参数类型不同的Python扩展函数？

要让模板函数支持首参数类型不同的函数指针，核心是让模板自动适配任意的self类型，而非固定为PyObject*。以下是具体实现步骤：

1. 调整模板函数，支持任意Self类型 #

通过类型萃取从函数指针中提取首参数的类型，再对传入的self指针做安全强制转换（Python所有对象都是PyObject子类，内存布局兼容，转换合法）：

#include <type_traits>

// 类型萃取工具：从函数指针中提取第一个参数的类型
template <typename Fn>
struct function_self_arg;

// 特化处理符合目标签名的函数指针
template <typename Self>
struct function_self_arg<PyObject*(*)(Self*, PyObject*[], Py_ssize_t)>
{
    using type = Self;
};

// 简化类型别名
template <typename Fn>
using function_self_arg_t = typename function_self_arg<Fn>::type;

// 模板函数：接受任意符合签名的函数指针
template <auto fn>
PyObject* python_fastcall(PyObject* self, PyObject* args)
{
    typedef struct
    {
        PyObject_VAR_HEAD
        PyObject *ob_item[1];
    } PyTupleObject;

    // 提取函数指针的self参数类型
    using Self = function_self_arg_t<decltype(fn)>;
    // 转换self指针并调用目标函数
    return fn(reinterpret_cast<Self*>(self), 
              reinterpret_cast<PyTupleObject*>(args)->ob_item, 
              PyTuple_Size(args));
}

2. 保留原有宏定义 #

宏无需修改，可直接使用：

using PyCFunction = PyObject* (*)(PyObject*, PyObject*[]);
#define PYTHON_FASTCALL(f) (PyCFunction)python_fastcall<f>

3. 验证调用 #

现在可以正常调用所有目标函数：

PYTHON_FASTCALL(foo)
PYTHON_FASTCALL(bar)
PYTHON_FASTCALL(meh)

原写法问题分析 #

你之前尝试的enable_if写法存在两个核心问题：

• 误将模板参数fn当作类型处理，而非函数指针实例
• 硬编码允许的函数类型，扩展性差且语法错误（std::is_same<T, Ts || ...)写法不符合C++语法规范）

新方案通过auto模板参数直接接收函数指针实例，再用类型萃取动态获取self类型，既灵活又符合Python C扩展的内存模型。

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