Golang调用Python
Python是时髦的机器学习御用开发语言，Golang是大红大紫的新时代后端开发语言。Python很适合让搞算法的写写模型，而Golang很适合提供API服务，而我最近碰到一个场景，在golang提供Api的服务时需要爬取别的页面的数据，而python是爬取数据最好的语言，于是我就想到了Go服务调用python模块处理数据。
原理
Python提供了丰富的C-API。而C和Go又可以通过cgo无缝集成。所以，直接通过Golang调用libpython，就可以实现Go调Python的功能了。确实没啥神奇，只要会用C调Python，马上就知道怎么用了。但问题是，如果有的选择，这个年代还有多少人愿意去裸写C和C++呢？诚心默念Golang大法好。虽然直接用cgo调用libpython也不是不可以，但是有native-binding用起来肯定要爽的多，Github上有一个现成的Binding库go-python。
准备工作
Linux安装golang安装python
我自己利用docker构建了一个镜像。安装了 golang,python2.7 docker pull shaynemiller/go-python
代码
将golang调用代码写成对象
package cpython
import (
"fmt"
python "github.com/sbinet/go-python"
)
func init() {
err := python.Initialize()//加载python
if err != nil {
panic(err.Error())
}
}
var PyStr = python.PyString_FromString//将golang字符串转python字符串
var GoStr = python.PyString_AS_STRING//python字符串转golang字符串
type PythonTools struct {
Moudle string //模块路径
//Path string
Param string //调用函数传入参数
Func string//包名（取名取错了）
Name string //调用函数名
}
func NewPythonTools(moudle, name, param, funcname string) *PythonTools {
return &PythonTools{
Moudle: moudle,
Param: param,
Func: funcname,
Name: name,
}
}
func (pt *PythonTools) Do() string {
//获取系统中python的包
pt.InsertBeforeSysPath()
//获取模块
module := pt.ImportModule()
//获取脚本中的变量（假如python中有a这个变量）
a := module.GetAttrString("a")
fmt.Printf("[VARS] a = %#v\n", python.PyInt_AsLong(a))
//获取函数
funcname := module.GetAttrString(pt.Func)
//转换参数 列表参数使用Tuple来构建。
param := python.PyTuple_New(1)
python.PyTuple_SetItem(param, 0, PyStr(pt.Param))
//调用函数
res := funcname.Call(param, python.Py_None)
//将调用结果转换成golang字符串并返回
return GoStr(res)
}
func (pt *PythonTools) InsertBeforeSysPath() string {
sysModule := python.PyImport_ImportModule("sys")
path := sysModule.GetAttrString("path")
//python.PyList_Insert(path, 0, PyStr("/usr/local/lib/python2.7/site-packages"))传入的是系统中python包的路径
python.PyList_Insert(path, 0, PyStr("/usr/local/lib/python2.7/site-packages"))
return GoStr(path.Repr())
}
// ImportModule will import python module from given directory
func (pt *PythonTools) ImportModule() *python.PyObject {
sysModule := python.PyImport_ImportModule("sys") // import sys
path := sysModule.GetAttrString("path") // path = sys.path
python.PyList_Insert(path, 0, PyStr(pt.Moudle)) // path.insert(0, dir)
return python.PyImport_ImportModule(pt.Name) // return __import__(name)
}
python脚本
#! -*- coding:utf8 -*-
__author__ = 'zyx'
a = 1
def test(param):
print(param)
调用方法
pyTools := cpython.NewPythonTools("/usr/local/src/script/", "test", "test param", "test")
res := pyTools.Do()
println(res)
注
使用GetAttrString可以根据属性名获取对象的属性，相当于python中的.操作。调用Python函数可以采用Object.Call方法，，列表参数使用Tuple来构建。返回值用PyString_AS_STRING从Python字符串转换为C或Go的字符串。 更多用法可以参考Python-C API文档。 此文参考 墨航
参考文章
发表评论