add alert enabled attribute

rsnodgrass · Jan 20, 2021 · c5702f2 · c5702f2
1 parent b4eabc6
commit c5702f2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/custom_components/sensorpush/__init__.py b/custom_components/sensorpush/__init__.py
@@ -21,8 +21,8 @@
 from homeassistant.const import CONF_NAME, CONF_USERNAME, CONF_PASSWORD, CONF_SCAN_INTERVAL
 
 from .const import (ATTR_BATTERY_VOLTAGE, ATTR_DEVICE_ID, ATTR_OBSERVED_TIME, ATTR_AGE,
-                    ATTR_ATTRIBUTION, ATTRIBUTION, MEASURES,
-                    ATTR_ALERT_MIN, ATTR_ALERT_MAX, CONF_UNIT_SYSTEM, CONF_MAXIMUM_AGE,
+                    ATTR_ATTRIBUTION, ATTRIBUTION, MEASURES, CONF_UNIT_SYSTEM, CONF_MAXIMUM_AGE,
+                    ATTR_ALERT_MIN, ATTR_ALERT_MAX, ATTR_ALERT_ENABLED,
                     SENSORPUSH_DOMAIN, UNIT_SYSTEM_IMPERIAL, UNIT_SYSTEM_METRIC, UNIT_SYSTEMS)
 
 LOG = logging.getLogger(__name__)
@@ -172,7 +172,8 @@ def _update_callback(self):
         if alerts.get("min"):
             self._attrs.update({
                 ATTR_ALERT_MIN: alerts.get("min"),
-                ATTR_ALERT_MAX: alerts.get("max")
+                ATTR_ALERT_MAX: alerts.get("max"),
+                ATTR_ALERT_ENABLED: alerts.get("enabled")
             })
 
 #        LOG.info(f"{self._state} ... {self._attrs} ... {sensor_data} ... {self._sensor_info}")

diff --git a/custom_components/sensorpush/const.py b/custom_components/sensorpush/const.py
@@ -11,6 +11,7 @@
 ATTR_OBSERVED_TIME   = 'observed_time'
 ATTR_ALERT_MIN       = 'alert_min'
 ATTR_ALERT_MAX       = 'alert_max'
+ATTR_ALERT_ENABLED   = 'alert_enabled'
 
 CONF_UNIT_SYSTEM = 'unit_system'
 CONF_MAXIMUM_AGE = 'maximum_age' # maximum age (in minutes) of observations before they expire