feat(virtualenv): look up VIRTUAL_ENV_PROMPT first

Standard venv activation typically sets it to `(name) `. Parse that first to avoid having to parse pyvenv.cfg. Co-authored-by: Janne Mareike Koschinski <[email protected]>
justjanne · Jun 17, 2024 · 3c6ebb4 · 3c6ebb4
1 parent df3475d
commit 3c6ebb4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/segment-virtualenv.go b/segment-virtualenv.go
@@ -3,14 +3,19 @@ package main
 import (
 	"os"
 	"path"
+	"strings"
 
 	"gopkg.in/ini.v1"
 
 	pwl "github.com/justjanne/powerline-go/powerline"
 )
 
 func segmentVirtualEnv(p *powerline) []pwl.Segment {
-	var env string
+	env := os.Getenv("VIRTUAL_ENV_PROMPT")
+	if strings.HasPrefix(env, "(") && strings.HasSuffix(env, ") ") {
+		env = strings.TrimPrefix(env, "(")
+		env = strings.TrimSuffix(env, ") ")
+	}
 	if env == "" {
 		env, _ = os.LookupEnv("VIRTUAL_ENV")
 		if env != "" {