python UrlService class: add find_binary method which uses PATH env var

ilyam8 · ilyam8 · commit df0fef4435b6 · 2017-01-31T12:23:13.000+09:00
diff --git a/python.d/python_modules/base.py b/python.d/python_modules/base.py
@@ -32,6 +32,11 @@
 import threading
 import msg
 
+try:
+    PATH = os.getenv('PATH').split(':')
+except NameError:
+    PATH = '/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin'.split(':')
+
 
 # class BaseService(threading.Thread):
 class SimpleService(threading.Thread):
@@ -431,6 +436,18 @@ def update(self, interval):
 
         return updated
 
+    def find_binary(self, binary):
+        try:
+            if isinstance(binary, str):
+                binary = os.path.basename(binary)
+                return next(('/'.join([p, binary]) for p in PATH
+                            if os.path.isfile('/'.join([p, binary]))
+                            and os.access('/'.join([p, binary]), os.X_OK)))
+            else:
+                return None
+        except StopIteration:
+            return None
+
 
 class UrlService(SimpleService):
     # TODO add support for https connections
