limit options to enum device class

jblance · Jan 14, 2025 · b126e63 · b126e63
1 parent f52e5b6
commit b126e63
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/powermon/outputformats/hass.py b/powermon/outputformats/hass.py
@@ -118,7 +118,7 @@ def format(self, command: Command, result: Result, device_info) -> list:
                 payload["state_class"] = state_class
 
             # Add options
-            if response.definition.options is not None:
+            if response.definition.options is not None and device_class == "enum":
                 payload["options"] = list(response.definition.options.values())
 
             payloads = js.dumps(payload)

diff --git a/powermon/protocols/pi30max.py b/powermon/protocols/pi30max.py
@@ -281,7 +281,7 @@
                 "reading_type": ReadingType.MESSAGE, "icon": "mdi:identifier",
                 "response_type": ResponseType.BYTES},
             {"description": "Work mode",
-                "reading_type": ReadingType.MESSAGE,
+                "reading_type": ReadingType.MESSAGE, "device_class": "enum",
                 "response_type": ResponseType.OPTION,
                 "options": {
                     "P": "Power On Mode",