mudhorn/oh-my-posh
1
0
Fork 0
mirror of https://github.com/JanDeDobbeleer/oh-my-posh.git synced 2025-02-02 05:41:10 -08:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor(python): add log statements

Browse source
This commit is contained in:
Jan De Dobbeleer 2024-12-12 13:19:23 +01:00 committed by Jan De Dobbeleer
parent bda70f0612
commit b0ecd21ce7
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/segments/python.go
View file

@ -7,6 +7,7 @@ import (
"slices"
"strings"
"github.com/jandedobbeleer/oh-my-posh/src/log"
"github.com/jandedobbeleer/oh-my-posh/src/properties"
"github.com/jandedobbeleer/oh-my-posh/src/runtime/path"
)
@ -68,6 +69,7 @@ func (p *Python) loadContext() {
p.Venv = prompt
return
}
venvVars := []string{
"VIRTUAL_ENV",
"CONDA_ENV_PATH",
@ -88,9 +90,11 @@ func (p *Python) loadContext() {
}
name := path.Base(venv)
log.Debugf("virtual env name: %s", name)
if folderNameFallback && slices.Contains(defaultVenvNames, name) {
venv = strings.TrimSuffix(venv, name)
name = path.Base(venv)
log.Debugf("virtual env name (fallback): %s", name)
}
if p.canUseVenvName(name) {
@ -108,12 +112,15 @@ func (p *Python) canUseVenvName(name string) bool {
if p.language.props.GetBool(properties.DisplayDefault, true) {
return true
}
invalidNames := [2]string{"system", "base"}
for _, a := range invalidNames {
if a == name {
log.Debugf("virtual env name %s is invalid", name)
return false
}
}
return true
}
@ -193,6 +200,7 @@ func (p *Python) pyvenvCfgPrompt() string {
if len(lineSplit) != 2 {
continue
}
key := strings.TrimSpace(lineSplit[0])
if key == "prompt" {
value := strings.TrimSpace(lineSplit[1])