fix construct enums for hass

jblance · Jan 15, 2025 · c7d4970 · c7d4970
1 parent 823d1f1
commit c7d4970
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/powermon/outputformats/hass.py b/powermon/outputformats/hass.py
@@ -2,6 +2,8 @@
 import json as js
 import logging
 from datetime import datetime
+import construct as cs
+from enum import Enum
 
 from powermon.commands.command import Command
 from powermon.commands.reading import Reading
@@ -132,6 +134,9 @@ def format(self, command: Command, result: Result, device_info) -> list:
             config_msgs.append(msg)
 
             # VALUE SETTING
+            # convert construct EnumIntegerStrings to a str
+            if isinstance(value, cs.EnumIntegerString):
+                value = str(value)
             msg = {"topic": state_topic, "payload": value}
             value_msgs.append(msg)
 

diff --git a/tests/config/ble_neey.yaml b/tests/config/ble_neey.yaml
@@ -14,8 +14,9 @@ commands:
   outputs:
   - type: screen
     format: 
-      type: table
-      draw_lines: True
+      type: hass
+      # draw_lines: True
+      filter: buzzer
 
 mqttbroker:
   name: localhost