Merge pull request #38 from junalmeida/device_class

Add `device_class` to water meter sensor.
junalmeida · Nov 4, 2022 · 0e3f850 · 0e3f850
2 parents 2745fa6 + 5c97de5
commit 0e3f850
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/meterparser/CHANGELOG.md b/meterparser/CHANGELOG.md
@@ -1,4 +1,6 @@
-# Changelog
+### [1.0.1.4]
+
+- Add `device_class` to water meter sensor.
 
 ### [1.0.1.3]
 

diff --git a/meterparser/config.yaml b/meterparser/config.yaml
@@ -1,5 +1,5 @@
 name: Meter Parser
-version: 1.0.1.3
+version: 1.0.1.4
 slug: meter-parser
 description: Read meter needles and numbers from a camera snapshot.
 url: https://github.com/junalmeida/homeassistant-addons/tree/main/meterparser

diff --git a/meterparser/src/app/mqtt.py b/meterparser/src/app/mqtt.py
@@ -124,8 +124,7 @@ def mqtt_sensor_discovery(self, entity_id: str, name: str, device_class: str, un
             "unique_id": "%s_%s" % (self.device_id, entity_id),
             "device": self.device
         }
-        if device_class != "water":
-            payload_sensor["device_class"] = device_class
+        payload_sensor["device_class"] = device_class
 
 
         topic_camera = "%s/camera/%s/%s/config" % (discovery_prefix, self.device_id, entity_id)