mirror of
https://github.com/home-assistant/core.git
synced 2025-11-05 00:49:37 +00:00
Support parsing mqtt messages via jsonpath
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user