fix: Fix invalid event type being sent

RogerSelwyn · Jan 21, 2025 · 4b21a8b · 4b21a8b
1 parent d87e8c6
commit 4b21a8b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/custom_components/mqtt_discoverystream_alt/classes/event.py b/custom_components/mqtt_discoverystream_alt/classes/event.py
@@ -27,7 +27,11 @@ def build_config(self, config, entity_info: EntityInfo):
     async def async_publish_state(self, new_state, mybase):
         """Publish the state for a text."""
 
-        payload = {ATTR_EVENT_TYPE: new_state.attributes[ATTR_EVENT_TYPE]}
-        await self._async_mqtt_publish(ATTR_STATE, payload, mybase, encoded=True)
+        if (
+            new_state.attributes[ATTR_EVENT_TYPE]
+            in new_state.attributes[ATTR_EVENT_TYPES]
+        ):
+            payload = {ATTR_EVENT_TYPE: new_state.attributes[ATTR_EVENT_TYPE]}
+            await self._async_mqtt_publish(ATTR_STATE, payload, mybase, encoded=True)
 
         await super().async_publish_state(new_state, mybase)