Convert pytuya to asyncio

custom_components/localtuya/__init__.py

@@ -52,12 +52,10 @@ localtuya:
 """
 import asyncio
 import logging
-from datetime import timedelta, datetime
 
 from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.const import (
-    CONF_DEVICE_ID,
     CONF_PLATFORM,
     CONF_ENTITIES,
     SERVICE_RELOAD,
@@ -73,9 +71,6 @@ from .common import TuyaDevice
 _LOGGER = logging.getLogger(__name__)
 
 UNSUB_LISTENER = "unsub_listener"
-UNSUB_TRACK = "unsub_track"
-
-POLL_INTERVAL = 30
 
 CONFIG_SCHEMA = config_schema()
 
@@ -136,24 +131,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
 
     device = TuyaDevice(hass, entry.data)
 
-    async def update_state(now):
-        """Read device status and update platforms."""
-        status = None
-        try:
-            status = await hass.async_add_executor_job(device.status)
-        except Exception:
-            _LOGGER.debug("update failed")
-
-        signal = f"localtuya_{entry.data[CONF_DEVICE_ID]}"
-        async_dispatcher_send(hass, signal, status)
-
-    unsub_track = async_track_time_interval(
-        hass, update_state, timedelta(seconds=POLL_INTERVAL)
-    )
-
     hass.data[DOMAIN][entry.entry_id] = {
         UNSUB_LISTENER: unsub_listener,
-        UNSUB_TRACK: unsub_track,
         TUYA_DEVICE: device,
     }
 
@@ -166,8 +145,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
                 for entity in entry.data[CONF_ENTITIES]
             ]
         )
-
-        await update_state(datetime.now())
+        device.connect()
 
     hass.async_create_task(setup_entities())
 
@@ -188,7 +166,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
     )
 
     hass.data[DOMAIN][entry.entry_id][UNSUB_LISTENER]()
-    hass.data[DOMAIN][entry.entry_id][UNSUB_TRACK]()
+    hass.data[DOMAIN][entry.entry_id][TUYA_DEVICE].close()
     if unload_ok:
         hass.data[DOMAIN].pop(entry.entry_id)