nikdoof/homeassistant-prometheus-query
1
0
Fork 0
mirror of https://github.com/nikdoof/homeassistant-prometheus-query.git synced 2025-12-18 04:59:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

device_class added

Browse Source
This commit is contained in:
FriendlyNGeeks
2021-11-11 10:44:40 -05:00
committed by GitHub
parent d3510719c3
commit 962ab5e3fa
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
custom_components/prometheus_query/sensor.py
View File

@@ -11,6 +11,7 @@ from homeassistant.util import Throttle
from homeassistant.const import (
CONF_NAME,
CONF_UNIT_OF_MEASUREMENT,
CONF_DEVICE_CLASS,
STATE_UNKNOWN,
)
from homeassistant.components.sensor import (
@@ -23,6 +24,7 @@ from prometheus_client import Summary
CONF_PROMETHEUS_URL = 'prometheus_url'
CONF_PROMETHEUS_QUERY = 'prometheus_query'
CONF_STATE_CLASS = 'state_class'
CONF_DEVICE_CLASS = 'device_class'
CONF_UNIQUE_ID = 'unique_id'
SCAN_INTERVAL = timedelta(seconds=600)
@@ -35,6 +37,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_NAME): cv.string,
vol.Optional(CONF_UNIT_OF_MEASUREMENT): cv.string,
vol.Optional(CONF_STATE_CLASS): STATE_CLASSES_SCHEMA,
vol.Optional(CONF_DEVICE_CLASS): DEVICE_CLASSES_SCHEMA,
vol.Optional(CONF_UNIQUE_ID): cv.string,
})
@@ -46,6 +49,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
'name': str(config.get(CONF_NAME)),
'unit': str(config.get(CONF_UNIT_OF_MEASUREMENT)),
'state_class': str(config.get(CONF_STATE_CLASS)),
'device_class': str(config.get(CONF_DEVICE_CLASS)),
'unique_id': str(config.get(CONF_UNIQUE_ID)),
}
@@ -62,6 +66,7 @@ class PrometheusQuery(SensorEntity):
self._state = None
self._attr_native_unit_of_measurement = prom_data["unit"]
self._attr_state_class = prom_data["state_class"]
self._attr_device_class = prom_data["device_class"]
self._attr_unique_id = f"${prom_data['url']}$${prom_data['query']}"
if prom_data["unique_id"] is not None:
self._attr_unique_id = prom_data["unique_id"]