feat(python): Show value of PYENV_VERSION when present (#3144)

* 2872: Show value of PYENV_VERSION when present

* 2872: Run cargo fmt

1 files 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<Module<'a>> {
 }
 
 fn get_pyenv_version(context: &Context) -> Option<String> {
-    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<String> {