mirror of
https://github.com/home-assistant/core.git
synced 2025-06-15 18:47:05 +00:00
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
#!/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
|