Merge pull request #312 from amkarthik/master

Real Virtual env name while using "pipenv"
b-ryan · Feb 26, 2018 · ae33f7e · ae33f7e
2 parents d0dec6c + 21cfded
commit ae33f7e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/powerline_shell/segments/virtual_env.py b/powerline_shell/segments/virtual_env.py
@@ -7,6 +7,9 @@ def add_to_powerline(self):
         env = os.getenv('VIRTUAL_ENV') \
             or os.getenv('CONDA_ENV_PATH') \
             or os.getenv('CONDA_DEFAULT_ENV')
+        if os.getenv('VIRTUAL_ENV') \
+            and os.path.basename(env) == '.venv':
+            env = os.path.basename(os.path.dirname(env))
         if not env:
             return
         env_name = os.path.basename(env)