b-ryan.powerline-shell/powerline_shell/segments/ruby_version.py
Blue-Dog-Archolite 6ff4b50376 Ruby Segment changes:
- use "GEM_HOME" in keys
- removal of python2 specific code
- verification of python2 / python3 functionality
- use of format in append to replace string join
2018-02-22 15:39:11 -07:00

23 lines
678 B
Python

import os
import subprocess
from ..utils import BasicSegment
class Segment(BasicSegment):
def add_to_powerline(self):
powerline = self.powerline
try:
p1 = subprocess.Popen(['ruby', '-v'], stdout=subprocess.PIPE)
p2 = subprocess.Popen(['sed', "s/ (.*//"], stdin=p1.stdout, stdout=subprocess.PIPE)
ruby_and_gemset = p2.communicate()[0].decode('utf-8').rstrip()
gem_set = os.environ.get('GEM_HOME', '@').split('@')
if len(gem_set) > 1:
ruby_and_gemset += "@{}".format(gem_set.pop())
powerline.append(ruby_and_gemset, 15, 1)
except OSError:
return