highlevel.go

package gopython

//#cgo pkg-config: python-3.6
//#include "go-python.h"
import "C"

import (
	"fmt"
)

var PyStr = PyString_FromString
var GoStr = PyString_AsString
var GoInt = PyInt_AsLong

func InsertPackagePath(path string) error {
	interr := PyRun_SimpleString("import sys")
	if interr == 0 {
		interr = PyRun_SimpleString(fmt.Sprintf("sys.path.append('%s')", path))
		if interr == 0 {
			return nil
		}
	}

	return fmt.Errorf("python: could not invoke PyRun_SimpleString to append extra path %s", path)
}

// CallFunc 调用 Python 中的方法
// 指定模块名称，方法名称，排列参数
func CallFunc(modulename string, funcname string, args ...interface{}) (*PyObject, error) {
	module, err := getModule(modulename)
	if err == nil {
		if len(args) > 0 {
			funcArgs := PyTuple_New(len(args))
			for i := 0; i < len(args); i++ {
				switch args[i].(type) {
				case int:
					PyTuple_SetItem(funcArgs, i, PyInt_FromLong(args[i].(int)))
				case string:
					PyTuple_SetItem(funcArgs, i, PyStr(args[i].(string)))
				}
			}

			var attr = module.GetAttrString(funcname)
			if attr != nil {
				res := attr.CallFunction(funcArgs)
				return res, nil
			}
		} else {
			var attr = module.GetAttrString(funcname)
			if attr != nil {
				res := module.GetAttrString(funcname).CallFunction()
				return res, nil
			}
		}

		return nil, fmt.Errorf("python: could not get func named %s after importing module named %s", funcname, modulename)
	}

	return nil, err
}

func getModule(modulename string) (*PyObject, error) {
	if obj := PyImport_ImportModule(modulename); obj != nil {
		return obj, nil
	}

	return nil, fmt.Errorf("python: could not import The specified module named %s", modulename)
}