Merge pull request #3 from fingon/process-source-up

Added a simple process state monitor

homeassistant/bootstrap.py

@@ -156,6 +156,13 @@ def from_config_file(config_path, enable_logging=True):
 
         add_status("WeMo", wemo.setup(hass, hosts))
 
+    # Process tracking
+    if has_section("process"):
+        process = load_module('process')
+
+        processes = dict(config.items('process'))
+        add_status("process", process.setup(hass, processes))
+
     # Light control
     if has_section("light.hue"):
         light = load_module('light')

homeassistant/components/process.py

@@ -0,0 +1,51 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# -*- Python -*-
+#
+# $Id: process.py $
+#
+# Author: Markus Stenberg <fingon@iki.fi>
+#
+# Copyright (c) 2014 Markus Stenberg
+#
+# Created:       Wed Apr 23 23:33:26 2014 mstenber
+# Last modified: Thu Apr 24 17:13:04 2014 mstenber
+# Edit time:     19 min
+#
+"""
+
+Process watcher.
+
+The arguments are <subentityname>=<substring to find in process list>
+
+"""
+
+from homeassistant.components import (STATE_ON, STATE_OFF)
+import os
+
+DOMAIN = 'process'
+ENTITY_ID_FORMAT = DOMAIN + '.{}'
+PS_STRING = 'ps awx'
+
+
+def setup(hass, processes):
+    """ Track local processes. """
+
+    # pylint: disable=unused-argument
+    def _update_process_state(time):
+        """ Check ps for currently running processes. """
+        with os.popen(PS_STRING, 'r') as psfile:
+            lines = list(iter(psfile))
+            for pname, pstring in processes.items():
+                found = False
+                for line in lines:
+                    if pstring in line:
+                        found = True
+                        break
+                entity_id = ENTITY_ID_FORMAT.format(pname)
+                state = found and STATE_ON or STATE_OFF
+                hass.states.set(entity_id, state)
+
+    _update_process_state(None)
+    hass.track_time_change(_update_process_state, second=[0, 30])
+    return True