Add base entity class LocalTuyaEntity

All platforms will inherit from this class in the end, but this commit
starts with switch only.

custom_components/localtuya/__init__.py

@@ -1,4 +1,5 @@
 """The LocalTuya integration integration."""
+import logging
 import voluptuous as vol
 
 from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
@@ -14,10 +15,12 @@ from homeassistant.const import (
     CONF_ENTITIES,
 )
 import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.entity import Entity
 
 from . import pytuya
 from .const import CONF_LOCAL_KEY, CONF_PROTOCOL_VERSION, DOMAIN
 
+_LOGGER = logging.getLogger(__name__)
 
 DEFAULT_ID = "1"
 DEFAULT_PROTOCOL_VERSION = 3.3
@@ -76,7 +79,7 @@ async def async_setup(hass: HomeAssistant, config: dict):
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
     """Set up LocalTuya integration from a config entry."""
     for platform in set(entity[CONF_PLATFORM] for entity in entry.data[CONF_ENTITIES]):
-#        print("ASE*** [{}] [{}]".format(entry.data["entities"][0][CONF_PLATFORM], platform))
+        #        print("ASE*** [{}] [{}]".format(entry.data["entities"][0][CONF_PLATFORM], platform))
         hass.async_create_task(
             hass.config_entries.async_forward_entry_setup(entry, platform)
         )
@@ -87,3 +90,62 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
     """Unload a config entry."""
     # Nothing is stored and no persistent connections exist, so nothing to do
     return True
+
+
+class LocalTuyaEntity(Entity):
+    """Representation of a Tuya entity."""
+
+    def __init__(
+        self,
+        device,
+        friendly_name,
+        dps_id,
+    ):
+        """Initialize the Tuya entity."""
+        self._device = device
+        self._name = friendly_name
+        self._available = False
+        self._dps_id = dps_id
+        self._status = None
+
+    @property
+    def name(self):
+        """Get name of Tuya entity."""
+        return self._name
+
+    @property
+    def unique_id(self):
+        """Return unique device identifier."""
+        return f"local_{self._device.unique_id}_{self._dps_id}"
+
+    @property
+    def available(self):
+        """Return if device is available or not."""
+        return self._available
+
+    def dps(self, dps_index):
+        """Return cached value for DPS index."""
+        value = self._status["dps"].get(dps_index)
+        if value is None:
+            _LOGGER.warning(
+                "Entity %s is requesting unknown DPS index %s",
+                self.entity_id,
+                dps_index,
+            )
+        return value
+
+    def update(self):
+        """Update state of Tuya entity."""
+        try:
+            self._status = self._device.status()
+            self.status_updated()
+        except Exception:
+            self._available = False
+        else:
+            self._available = True
+
+    def status_updated(self):
+        """Device status was updated.
+
+        Override in subclasses and update entity specific state.
+        """