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

justjanne · Aug 13, 2023 · a307d69 · a307d69
1 parent a8a1891
commit a307d69
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/segment-virtualenv.go b/segment-virtualenv.go
@@ -16,12 +16,15 @@ func segmentVirtualEnv(p *powerline) []pwl.Segment {
 		env = env[1 : len(env)-2]
 	}
 	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 == "" {