39 lines
715 B
Go
39 lines
715 B
Go
|
package cmd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"os/exec"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
var PythonNotFound = errors.New("python not found")
|
||
|
|
|
||
|
|
func Python(options BasicOptions, args ...string) (output string, err error) {
|
||
|
|
|
||
|
|
if _, err := Which("python3", BasicOptions{
|
||
|
|
Env: options.Env,
|
||
|
|
Sources: options.Sources,
|
||
|
|
Cwd: options.Cwd,
|
||
|
|
}); err != nil {
|
||
|
|
if errors.Is(err, ErrNotFound) {
|
||
|
|
return "", PythonNotFound
|
||
|
|
} else {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
command := exec.Command("python3", args...)
|
||
|
|
|
||
|
|
outputBytes, err := command.Output()
|
||
|
|
if err != nil {
|
||
|
|
var exitErr *exec.ExitError
|
||
|
|
if errors.As(err, &exitErr) {
|
||
|
|
return "", fmt.Errorf("command error: %w", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return strings.Trim(string(outputBytes), "\n"), nil
|
||
|
|
|
||
|
|
}
|