feat(virtualenv): fall back to VIRTUAL_ENV dir if prompt is not set

justjanne · Jun 17, 2024 · ef5e672 · ef5e672
1 parent 3c6ebb4
commit ef5e672
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/segment-virtualenv.go b/segment-virtualenv.go
@@ -17,12 +17,15 @@ func segmentVirtualEnv(p *powerline) []pwl.Segment {
 		env = strings.TrimSuffix(env, ") ")
 	}
 	if env == "" {
-		env, _ = os.LookupEnv("VIRTUAL_ENV")
-		if env != "" {
-			cfg, err := ini.Load(path.Join(env, "pyvenv.cfg"))
+		venv, _ := os.LookupEnv("VIRTUAL_ENV")
+		if venv != "" {
+			cfg, err := ini.Load(path.Join(venv, "pyvenv.cfg"))
 			if err == nil {
 				env = cfg.Section("").Key("prompt").String()
 			}
+			if env == "" {
+				env = venv
+			}
 		}
 	}
 	if env == "" {