fix: adjust HASS state classes for rain entities (#1018)

bachya · Jun 21, 2024 · e11dbb4 · e11dbb4
1 parent 58b0d24
commit e11dbb4
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 50 deletions.
diff --git a/ecowitt2mqtt/helpers/publisher/mqtt/hass.py b/ecowitt2mqtt/helpers/publisher/mqtt/hass.py
@@ -428,19 +428,19 @@ class HassDiscoveryInfo:
 }
 
 STATE_CLASS_OVERRIDES = {
-    DATA_POINT_DAILY_RAIN: StateClass.TOTAL_INCREASING,
-    DATA_POINT_DRAIN_PIEZO: StateClass.TOTAL_INCREASING,
-    DATA_POINT_ERAIN_PIEZO: StateClass.TOTAL_INCREASING,
-    DATA_POINT_EVENT_RAIN: StateClass.TOTAL_INCREASING,
-    DATA_POINT_HOURLY_RAIN: StateClass.MEASUREMENT,
-    DATA_POINT_HRAIN_PIEZO: StateClass.MEASUREMENT,
-    DATA_POINT_MONTHLY_RAIN: StateClass.TOTAL_INCREASING,
-    DATA_POINT_MRAIN_PIEZO: StateClass.TOTAL_INCREASING,
-    DATA_POINT_TOTAL_RAIN: StateClass.TOTAL_INCREASING,
-    DATA_POINT_WEEKLY_RAIN: StateClass.TOTAL_INCREASING,
-    DATA_POINT_WRAIN_PIEZO: StateClass.TOTAL_INCREASING,
-    DATA_POINT_YEARLY_RAIN: StateClass.TOTAL_INCREASING,
-    DATA_POINT_YRAIN_PIEZO: StateClass.TOTAL_INCREASING,
+    DATA_POINT_DAILY_RAIN: StateClass.TOTAL,
+    DATA_POINT_DRAIN_PIEZO: StateClass.TOTAL,
+    DATA_POINT_ERAIN_PIEZO: StateClass.TOTAL,
+    DATA_POINT_EVENT_RAIN: StateClass.TOTAL,
+    DATA_POINT_HOURLY_RAIN: StateClass.TOTAL,
+    DATA_POINT_HRAIN_PIEZO: StateClass.TOTAL,
+    DATA_POINT_MONTHLY_RAIN: StateClass.TOTAL,
+    DATA_POINT_MRAIN_PIEZO: StateClass.TOTAL,
+    DATA_POINT_TOTAL_RAIN: StateClass.TOTAL,
+    DATA_POINT_WEEKLY_RAIN: StateClass.TOTAL,
+    DATA_POINT_WRAIN_PIEZO: StateClass.TOTAL,
+    DATA_POINT_YEARLY_RAIN: StateClass.TOTAL,
+    DATA_POINT_YRAIN_PIEZO: StateClass.TOTAL,
 }
 
 

diff --git a/ecowitt2mqtt/runtime.py b/ecowitt2mqtt/runtime.py
@@ -146,7 +146,7 @@ async def create_loop() -> None:
                         )
                         await asyncio.sleep(delay)
             except Exception as err:  # pylint: disable=broad-except
-                LOGGER.exception("Exception caused a shutdown: %s", err)
+                LOGGER.exception("%s exception caused a shutdown: %s", type(err), err)
                 LOGGER.debug("".join(traceback.format_tb(err.__traceback__)))
                 self.stop()