Formalise constant

rgc99 · Oct 24, 2023 · e410b96 · e410b96
1 parent 10dbf6d
commit e410b96
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/custom_components/irrigation_unlimited/binary_sensor.py b/custom_components/irrigation_unlimited/binary_sensor.py
@@ -47,6 +47,7 @@
     RES_MANUAL,
     RES_NOT_RUNNING,
     RES_NONE,
+    ATTR_VOLUME,
 )
 
 
@@ -257,5 +258,5 @@ def extra_state_attributes(self):
             attr[ATTR_CONFIGURATION] = self._zone.configuration
         if self._zone.show_timeline:
             attr[ATTR_TIMELINE] = self._zone.timeline()
-        attr["volume"] = self._zone.volume.total
+        attr[ATTR_VOLUME] = self._zone.volume.total
         return attr
diff --git a/custom_components/irrigation_unlimited/const.py b/custom_components/irrigation_unlimited/const.py
@@ -193,6 +193,7 @@
 ATTR_NEXT_TICK = "next_tick"
 ATTR_TICK_LOG = "tick_log"
 ATTR_SUSPENDED = "suspended"
+ATTR_VOLUME = "volume"
 
 # Resources
 RES_MANUAL = "Manual"