grand refactor

1 files changed, 19 insertions, 0 deletions

diff --git a/src/venv.rs b/src/venv.rs
new file mode 100644
index 0000000..7906abc
--- /dev/null
+++ b/src/venv.rs
@@ -0,0 +1,19 @@
+use std::env;
+use colored::*;
+use std::path::Path;
+
+pub fn get_name() -> colored::ColoredString {
+    match env::var("VIRTUAL_ENV") {
+        Ok(venv_path) => {
+            let venv_name = Path::new(&venv_path[..]).file_name();
+            if let Some(name) = venv_name {
+                if let Some(valid_name) = name.to_str() {
+                    return format!("({})", valid_name).bright_black();
+                }
+            }
+        }
+        Err(_) => {}
+    }
+    return "".white()
+}
+