From 8d80d2ef062bdd9e8a7ae9d5ada7d465ae55c235 Mon Sep 17 00:00:00 2001 From: cbolgiano Date: Tue, 9 Nov 2021 15:55:53 -0500 Subject: [PATCH] feat(python): Show value of PYENV_VERSION when present (#3144) * 2872: Show value of PYENV_VERSION when present * 2872: Run cargo fmt --- src/modules/python.rs | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/src/modules/python.rs b/src/modules/python.rs index 7ece718c9..4576db949 100644 --- a/src/modules/python.rs +++ b/src/modules/python.rs @@ -76,12 +76,19 @@ pub fn module<'a>(context: &'a Context) -> Option> { } fn get_pyenv_version(context: &Context) -> Option { - let version_name = context - .exec_cmd("pyenv", &["version-name"])? - .stdout - .trim() - .to_string(); - Some(version_name) + let mut version_name = context.get_env("PYENV_VERSION"); + + if version_name.is_none() { + version_name = Some( + context + .exec_cmd("pyenv", &["version-name"])? + .stdout + .trim() + .to_string(), + ) + } + + version_name } fn get_python_version(context: &Context, config: &PythonConfig) -> Option {