diff --git a/docs/docs/segment-python.md b/docs/docs/segment-python.md index 13bd1f81..7c4eaa4d 100644 --- a/docs/docs/segment-python.md +++ b/docs/docs/segment-python.md @@ -44,6 +44,7 @@ properties below. Defaults to `{{ .Full }}` ## Template Properties +- `.Venv`: `string` - the virtual environment name (if present) - `.Full`: `string` - the full version - `.Major`: `string` - is the major version - `.Minor`: `string` - is the minor version diff --git a/docs/docs/segment-session.md b/docs/docs/segment-session.md index f0067415..fcb0b92b 100644 --- a/docs/docs/segment-session.md +++ b/docs/docs/segment-session.md @@ -13,7 +13,7 @@ Show the current user and host name. ```json { "type": "session", - "style": "diamond", + "style": "diamond", "foreground": "#ffffff", "background": "#c386f1", "leading_diamond": "\uE0B6", diff --git a/src/segment_python.go b/src/segment_python.go index c36010ed..203c2d02 100644 --- a/src/segment_python.go +++ b/src/segment_python.go @@ -4,7 +4,8 @@ import "fmt" type python struct { language - venvName string + + Venv string } const ( @@ -13,14 +14,14 @@ const ( ) func (p *python) string() string { - if p.venvName == "" { + if p.Venv == "" { return p.language.string() } version := p.language.string() if version == "" { - return p.venvName + return p.Venv } - return fmt.Sprintf("%s %s", p.venvName, version) + return fmt.Sprintf("%s %s", p.Venv, version) } func (p *python) init(props properties, env environmentInfo) { @@ -67,14 +68,14 @@ func (p *python) loadContext() { venv = p.language.env.getenv(venvVar) name := base(venv, p.language.env) if p.canUseVenvName(name) { - p.venvName = name + p.Venv = name break } } } func (p *python) inContext() bool { - return p.venvName != "" + return p.Venv != "" } func (p *python) canUseVenvName(name string) bool {