Merge pull request #260 from dogo/feature/add-rbenv-segment

[New Segment] rbenv
b-ryan · Jun 21, 2017 · 7830815 · 7830815
2 parents f77348b + a253a72
commit 7830815
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/config.py.dist b/config.py.dist
@@ -13,6 +13,9 @@ SEGMENTS = [
 # Show current virtual environment (see http://www.virtualenv.org/)
  'virtual_env',
 
+# Show current ruby environment (see http://rbenv.org/)
+ 'rbenv', 
+
 # Show the current user's username as in ordinary prompts
  'username',
 

diff --git a/segments/rbenv.py b/segments/rbenv.py
@@ -0,0 +1,13 @@
+import subprocess
+
+
+def add_rbenv_segment(powerline):
+ try:
+ p1 = subprocess.Popen(["rbenv", "local"], stdout=subprocess.PIPE)
+ version = p1.communicate()[0].decode("utf-8").rstrip()
+ if len(version) <= 0:
+ return
+
+ powerline.append(' %s ' % version, Color.VIRTUAL_ENV_FG, Color.VIRTUAL_ENV_BG)
+ except OSError:
+ return