Support parsing mqtt messages via jsonpath

This commit is contained in:
Oliver van Porten
2015-11-20 22:03:17 +01:00
parent d25f58e650
commit b4cf0e874a
4 changed files with 35 additions and 5 deletions

View File

@@ -7,6 +7,8 @@ For more details about this platform, please refer to the documentation at
https://home-assistant.io/components/sensor.mqtt/
"""
import logging
import json
import jsonpath_rw
from homeassistant.helpers.entity import Entity
import homeassistant.components.mqtt as mqtt
@@ -31,23 +33,30 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
config.get('name', DEFAULT_NAME),
config.get('state_topic'),
config.get('qos', DEFAULT_QOS),
config.get('unit_of_measurement'))])
config.get('unit_of_measurement'),
config.get('state_format'))])
# pylint: disable=too-many-arguments, too-many-instance-attributes
class MqttSensor(Entity):
""" Represents a sensor that can be updated using MQTT. """
def __init__(self, hass, name, state_topic, qos, unit_of_measurement):
def __init__(self, hass, name, state_topic, qos, unit_of_measurement,state_format):
self._state = "-"
self._hass = hass
self._name = name
self._state_topic = state_topic
self._qos = qos
self._unit_of_measurement = unit_of_measurement
self._state_format = state_format
if self._state_format.startswith('json:'):
self._parser = mqtt.JsonFmtParser(self._state_format[5:])
else:
self._parser = lambda x: x
def message_received(topic, payload, qos):
""" A new MQTT message has been received. """
self._state = payload
self._state = self._parser(payload)
self.update_ha_state()
mqtt.subscribe(hass, self._state_topic, message_received, self._qos)