mirror of
https://github.com/stashapp/stash.git
synced 2025-12-16 20:07:05 +03:00
Log more when resolving Python (#4185)
* Log more when resolving Python Users often have problems configuring their Python installations * Convert if-else ladder to switch statement * Consolidate Python resolution Adds additional logging to plugin tasks to align with the logging that scrapers output.
This commit is contained in:
@@ -7,6 +7,7 @@ import (
|
||||
"fmt"
|
||||
"io"
|
||||
"os/exec"
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
stashExec "github.com/stashapp/stash/pkg/exec"
|
||||
@@ -45,21 +46,17 @@ func (t *rawPluginTask) Start() error {
|
||||
var cmd *exec.Cmd
|
||||
if python.IsPythonCommand(command[0]) {
|
||||
pythonPath := t.serverConfig.GetPythonPath()
|
||||
var p *python.Python
|
||||
if pythonPath != "" {
|
||||
p = python.New(pythonPath)
|
||||
} else {
|
||||
p, _ = python.Resolve()
|
||||
}
|
||||
p, err := python.Resolve(pythonPath)
|
||||
|
||||
if p != nil {
|
||||
if err != nil {
|
||||
logger.Warnf("%s", err)
|
||||
} else {
|
||||
cmd = p.Command(context.TODO(), command[1:])
|
||||
}
|
||||
|
||||
// if could not find python, just use the command args as-is
|
||||
}
|
||||
|
||||
if cmd == nil {
|
||||
// if could not find python, just use the command args as-is
|
||||
cmd = stashExec.Command(command[0], command[1:]...)
|
||||
}
|
||||
|
||||
@@ -99,6 +96,8 @@ func (t *rawPluginTask) Start() error {
|
||||
go t.handlePluginStderr(t.plugin.Name, stderr)
|
||||
t.cmd = cmd
|
||||
|
||||
logger.Debugf("Plugin %s started: %s", t.plugin.Name, strings.Join(cmd.Args, " "))
|
||||
|
||||
// send the stdout to the plugin output
|
||||
go func() {
|
||||
defer t.waitGroup.Done()
|
||||
@@ -113,6 +112,7 @@ func (t *rawPluginTask) Start() error {
|
||||
errStr := err.Error()
|
||||
output.Error = &errStr
|
||||
}
|
||||
logger.Debugf("Plugin %s finished", t.plugin.Name)
|
||||
|
||||
t.result = &output
|
||||
}()
|
||||
|
||||
@@ -2,9 +2,12 @@ package python
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os/exec"
|
||||
|
||||
stashExec "github.com/stashapp/stash/pkg/exec"
|
||||
"github.com/stashapp/stash/pkg/fsutil"
|
||||
"github.com/stashapp/stash/pkg/logger"
|
||||
)
|
||||
|
||||
type Python string
|
||||
@@ -22,19 +25,32 @@ func New(path string) *Python {
|
||||
// Resolve tries to find the python executable in the system.
|
||||
// It first checks for python3, then python.
|
||||
// Returns nil and an exec.ErrNotFound error if not found.
|
||||
func Resolve() (*Python, error) {
|
||||
_, err := exec.LookPath("python3")
|
||||
func Resolve(configuredPythonPath string) (*Python, error) {
|
||||
if configuredPythonPath != "" {
|
||||
isFile, err := fsutil.FileExists(configuredPythonPath)
|
||||
switch {
|
||||
case err == nil && isFile:
|
||||
logger.Tracef("using configured python path: %s", configuredPythonPath)
|
||||
return New(configuredPythonPath), nil
|
||||
case err == nil && !isFile:
|
||||
logger.Warnf("configured python path is not a file: %s", configuredPythonPath)
|
||||
case err != nil:
|
||||
logger.Warnf("unable to use configured python path: %s", err)
|
||||
}
|
||||
}
|
||||
|
||||
python3, err := exec.LookPath("python3")
|
||||
|
||||
if err != nil {
|
||||
_, err = exec.LookPath("python")
|
||||
python, err := exec.LookPath("python")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
return nil, fmt.Errorf("python executable not in PATH: %s", err)
|
||||
}
|
||||
ret := Python("python")
|
||||
ret := Python(python)
|
||||
return &ret, nil
|
||||
}
|
||||
|
||||
ret := Python("python3")
|
||||
ret := Python(python3)
|
||||
return &ret, nil
|
||||
}
|
||||
|
||||
|
||||
@@ -38,21 +38,17 @@ func (s *scriptScraper) runScraperScript(ctx context.Context, inString string, o
|
||||
var cmd *exec.Cmd
|
||||
if python.IsPythonCommand(command[0]) {
|
||||
pythonPath := s.globalConfig.GetPythonPath()
|
||||
var p *python.Python
|
||||
if pythonPath != "" {
|
||||
p = python.New(pythonPath)
|
||||
p, err := python.Resolve(pythonPath)
|
||||
|
||||
if err != nil {
|
||||
logger.Warnf("%s", err)
|
||||
} else {
|
||||
p, _ = python.Resolve()
|
||||
cmd = p.Command(context.TODO(), command[1:])
|
||||
}
|
||||
|
||||
if p != nil {
|
||||
cmd = p.Command(ctx, command[1:])
|
||||
}
|
||||
|
||||
// if could not find python, just use the command args as-is
|
||||
}
|
||||
|
||||
if cmd == nil {
|
||||
// if could not find python, just use the command args as-is
|
||||
cmd = stashExec.Command(command[0], command[1:]...)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user