swig封装c++成Python，兼容所有python3

摘要

因为swig在封装成Python3可调用的so的时候，一般情况下都需要给每一个python3版本封装一个so；为了解决这个问题，就有了这篇文章。本文附带介绍一下如何做到Python2和Python3都能顺利run起来。

背景

之前使用swig封装成了Python2可调用的so，也封装了Python3.6可调用的so，发现Python3.7无法调用Python3.6封装的so，感觉非常奇怪，为了解决这个问题，找啊找，找了很久也没解决。今天突然有了一个小发现，顺利把这个问题解决了，过程还是很曲折的。

环境

linux，Python2.7，Python3.8，没有的话可以用miniconda3安装虚拟环境

代码

add.cc

#include <Python.h>
#include "add.h"
int add(int a,int b){
    return a+b;
}

add.h

#pragma once
int add(int a, int b);

分别为python2和Python3封装

demo2.i

%module demo2
%{
#include "add.h"
%}
int add(int a, int b);

demo_py2_setup.py

#!/bin/env python
#-*- coding:utf-8 -*-

from distutils.core import setup, Extension

demo_module = Extension('_demo2',
    sources= ['demo2_wrap.cxx', 'add.cc'],
    include_dirs = ['./'],
    extra_compile_args = ['-std=c++0x', '-O3']
    )

setup(name='demo2',
    version='0.1',
    author="SWIG Docs",
    description="""py for demo2""",
    ext_modules=[demo_module],
    py_modules=["demo2"],
    )

demo3.i

%module demo3
%{
#include "add.h"
%}
int add(int a, int b);

demo_py3_setup.py 

#!/bin/env python
#-*- coding:utf-8 -*-

from distutils.core import setup, Extension

demo_module = Extension('_demo3',
    sources= ['demo3_wrap.cxx', 'add.cc'],
    include_dirs = ['./'],
    extra_compile_args = ['-std=c++0x', '-O3']
    )

setup(name='demo3',
    version='0.1',
    author="SWIG Docs",
    description="""py for demo3""",
    ext_modules=[demo_module],
    py_modules=["demo3"],
    )

run.sh

/home/wangyiwen03/local/bin/swig -c++ -python demo2.i # 产出demo2_wrap.cxx
python2 demo_py2_setup.py build_ext --inplace # 产出_demo2.so

/home/wangyiwen03/local/bin/swig -c++ -python demo3.i # 产出demo3_wrap.cxx
python3 demo_py3_setup.py build_ext --inplace 
# 产出_demo3.cpython-38-x86_64-linux-gnu.so.so 和具体的Python3的版本有关

测试demo_test.py

import sys
if sys.version_info < (3, 0, 0):
    import _demo2 as demo
else:
    import _demo3 as demo

print(demo.add(10, 1))

执行过程

sh run.sh #会成功生成两个so(_demo2.so, _demo3.cpython-38-x86_64-linux-gnu.so)
mv _demo3.cpython-38-x86_64-linux-gnu.so _demo3.so
# 这一步的目的就是为了兼容所有的python3，不然的话要为每一个Python3封装一个so，很闹心
python2 demo_test.py # 输出 11
python3 demo_test.py # 输出 11

如果有更好的方案请留言评论，不胜感激

欢迎打赏，你的鼓励非常重要，谢谢