Migrate tuya event platform to use DPCodeWrapper (home-assistant#156127)

Author: epenet

homeassistant/components/tuya/event.py

@@ -14,9 +14,9 @@
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 
 from . import TuyaConfigEntry
-from .const import TUYA_DISCOVERY_NEW, DeviceCategory, DPCode, DPType
+from .const import TUYA_DISCOVERY_NEW, DeviceCategory, DPCode
 from .entity import TuyaEntity
-from .models import find_dpcode
+from .models import DPCodeEnumWrapper
 
 # All descriptions can be found here. Mostly the Enum data types in the
 # default status set of each category (that don't have a set instruction)
@@ -96,10 +96,17 @@ def async_discover_device(device_ids: list[str]) -> None:
         for device_id in device_ids:
             device = manager.device_map[device_id]
             if descriptions := EVENTS.get(device.category):
-                for description in descriptions:
-                    dpcode = description.key
-                    if dpcode in device.status:
-                        entities.append(TuyaEventEntity(device, manager, description))
+                entities.extend(
+                    TuyaEventEntity(
+                        device, manager, description, dpcode_wrapper=dpcode_wrapper
+                    )
+                    for description in descriptions
+                    if (
+                        dpcode_wrapper := DPCodeEnumWrapper.find_dpcode(
+                            device, description.key, prefer_function=True
+                        )
+                    )
+                )
 
         async_add_entities(entities)
 
@@ -120,14 +127,14 @@ def __init__(
         device: CustomerDevice,
         device_manager: Manager,
         description: EventEntityDescription,
+        dpcode_wrapper: DPCodeEnumWrapper,
     ) -> None:
         """Init Tuya event entity."""
         super().__init__(device, device_manager)
         self.entity_description = description
         self._attr_unique_id = f"{super().unique_id}{description.key}"
-
-        if dpcode := find_dpcode(self.device, description.key, dptype=DPType.ENUM):
-            self._attr_event_types: list[str] = dpcode.range
+        self._dpcode_wrapper = dpcode_wrapper
+        self._attr_event_types = dpcode_wrapper.enum_type_information.range
 
     async def _handle_state_update(
         self,
@@ -136,10 +143,10 @@ async def _handle_state_update(
     ) -> None:
         if (
             updated_status_properties is None
-            or self.entity_description.key not in updated_status_properties
+            or self._dpcode_wrapper.dpcode not in updated_status_properties
+            or (value := self._dpcode_wrapper.read_device_status(self.device)) is None
         ):
             return
 
-        value = self.device.status.get(self.entity_description.key)
         self._trigger_event(value)
         self.async_write_ha_state()