fix: 设置pip超时时间

reverse-direction-god · reverse-direction-god · commit 952e78281d64 · 2024-12-30T23:49:56.000+08:00
diff --git a/module/python/venv.go b/module/python/venv.go
@@ -136,6 +136,9 @@ func pipreqs(dir string, projectPath, savePath string, logger *zap.SugaredLogger
 	}
 	return nil
 }
+func setPipTimeout() error {
+	return os.Setenv("PIP_DEFAULT_TIMEOUT", "120")
+}
 func installpipreqs(dir string, logger *zap.SugaredLogger) error {
 	var out bytes.Buffer
 	var stderr bytes.Buffer
@@ -298,6 +301,9 @@ func Run(ctx context.Context, dir string, logger *zap.SugaredLogger, nvMp map[st
 	if err := updatePip(venvPath, logger); err != nil {
 		return nil, err
 	}
+	if err := setPipTimeout(); err != nil {
+		return nil, err
+	}
 	if err := installpipreqs(venvPath, logger); err != nil {
 		return nil, err
 	}

Original file line number	Diff line number	Diff line change
`@@ -136,6 +136,9 @@ func pipreqs(dir string, projectPath, savePath string, logger *zap.SugaredLogger`
`136`	`136`	`}`
`137`	`137`	`return nil`
`138`	`138`	`}`
	`139`	`+func setPipTimeout() error {`
	`140`	`+ return os.Setenv("PIP_DEFAULT_TIMEOUT", "120")`
	`141`	`+}`
`139`	`142`	`func installpipreqs(dir string, logger *zap.SugaredLogger) error {`
`140`	`143`	`var out bytes.Buffer`
`141`	`144`	`var stderr bytes.Buffer`
`@@ -298,6 +301,9 @@ func Run(ctx context.Context, dir string, logger *zap.SugaredLogger, nvMp map[st`
`298`	`301`	`if err := updatePip(venvPath, logger); err != nil {`
`299`	`302`	`return nil, err`
`300`	`303`	`}`
	`304`	`+ if err := setPipTimeout(); err != nil {`
	`305`	`+ return nil, err`
	`306`	`+ }`
`301`	`307`	`if err := installpipreqs(venvPath, logger); err != nil {`
`302`	`308`	`return nil, err`
`303`	`309`	`}`