摘要
因为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
如果有更好的方案请留言评论,不胜感激
欢迎打赏,你的鼓励非常重要,谢谢