package python

import (
	"os"
	"path/filepath"
	"strings"
)

const PYTHONPATH = "PYTHONPATH"

func SetPythonPath(path ...string) {
	path_ := strings.Join(path, string(filepath.ListSeparator))
	os.Setenv(PYTHONPATH, path_)
}

func AppendPythonPath(path ...string) {
	path_ := filepath.SplitList(os.Getenv(PYTHONPATH))
	path_ = append(path_, path...)
	path__ := strings.Join(path_, string(filepath.ListSeparator))
	os.Setenv(PYTHONPATH, path__)
}

func PrependPythonPath(path ...string) {
	path_ := filepath.SplitList(os.Getenv(PYTHONPATH))
	path_ = append(path, path_...)
	path__ := strings.Join(path_, string(filepath.ListSeparator))
	os.Setenv(PYTHONPATH, path__)
}