Add rbenv script

alexmuller · alexmuller · commit 79cdfdf7f8a1 · 2017-01-17T15:22:27.000Z
To help figure out what rbenv versions aren't being used.
diff --git a/fabfile.py b/fabfile.py
@@ -36,6 +36,7 @@
 import performanceplatform
 import postgresql
 import rabbitmq
+import rbenv
 import rkhunter
 import search
 import statsd
diff --git a/rbenv.py b/rbenv.py
@@ -0,0 +1,20 @@
+from fabric.api import run, sudo, task, hide
+from fabric.contrib.files import exists
+
+
+@task
+def version_in_use(version):
+    """Tests whether an rbenv version is in use on a machine."""
+
+    rbenv_path = '/usr/lib/rbenv/versions/{0}/'.format(version)
+
+    with hide('running'):
+        with hide('output'):
+            if exists(rbenv_path):
+                pids = sudo("lsof +D {0} | tail -n +2 | tr -s ' ' | cut -f 2 -d ' ' | sort -n | uniq".format(rbenv_path)).split('\r\n')
+            else:
+                pids = []
+        for pid in pids:
+            if pid == '':
+                continue
+            run('ps --no-headers {0}'.format(pid))
