python利用什么写模块_使用C/C++编写Python模块扩展

起步

前期准备

系统：Ubuntu 15.10

GCC：5.2.1

Python：2.7.10

开始

#include

#include //有的是#include

//判断是否是质数

static PyObject *pr_isprime(PyObject *self, PyObject *args) {

int n, num;

//解析参数

if (!PyArg_ParseTuple(args, "i", &num)) {

return NULL;

}

if (num < 1) {

return Py_BuildValue("i", 0); //C类型转成python对象

}

n = num - 1;

while (n > 1) {

if (num % n == 0)

return Py_BuildValue("i", 0);

n--;

}

return Py_BuildValue("i", 1);

}

static PyMethodDef PrMethods[] = {

//方法名，导出函数，参数传递方式，方法描述。

{"isPrime", pr_isprime, METH_VARARGS, "check if an input number is prime or not."},

{NULL, NULL, 0, NULL}

};

void initpr(void) {

(void) Py_InitModule("pr", PrMethods);

}

导出函数：C模块对外暴露的接口函数为pr_isprime，带有self和args两个参数，args包含了python解释器要传给c函数的所有参数，通常使用PyArg_ParseTuple()来获得这些参数值。

初始化函数：一遍python解释器能够对模块进行正确的初始化，初始化要以init开头，如initp。

方法列表：提供给外部的python程序使用函数名称映射表PrMethods，它是一个PyMethodDef结构体，成员依次是方法名，导出函数，参数传递方式，方法描述。

struct PyMethodDef {

char* ml_name; #方法名

PyCFunction ml_meth; #导出函数

int ml_flags; #参数传递方式

char* ml_doc; #方法描述

}

setup.py脚本

#!/usr/bin/env python

# coding=utf-8

from distutils.core import setup, Extension

module = Extension('pr', sources = ['py.c'])

setup(name = 'Pr test', version = '1.0', ext_modules = [module])

安装模块

测试