From 4f2d313a4a4b89070e1bfaabfc0bc386615a3c29 Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Tue, 19 Oct 2021 10:56:37 +0200 Subject: [PATCH] Remove device category filtering from Tuya init (#58019) --- homeassistant/components/tuya/__init__.py | 25 +++++++--------- homeassistant/components/tuya/const.py | 35 ----------------------- 2 files changed, 11 insertions(+), 49 deletions(-) diff --git a/homeassistant/components/tuya/__init__.py b/homeassistant/components/tuya/__init__.py index b5ff3ec19ad..d1e661fbe42 100644 --- a/homeassistant/components/tuya/__init__.py +++ b/homeassistant/components/tuya/__init__.py @@ -33,7 +33,6 @@ from .const import ( PLATFORMS, TUYA_DISCOVERY_NEW, TUYA_HA_SIGNAL_UPDATE_ENTITY, - TUYA_SUPPORTED_PRODUCT_CATEGORIES, ) _LOGGER = logging.getLogger(__name__) @@ -118,8 +117,7 @@ async def _init_tuya_sdk(hass: HomeAssistant, entry: ConfigEntry) -> bool: # Register known device IDs for device in device_manager.device_map.values(): - if device.category in TUYA_SUPPORTED_PRODUCT_CATEGORIES: - device_ids.add(device.id) + device_ids.add(device.id) hass.config_entries.async_setup_platforms(entry, PLATFORMS) return True @@ -178,20 +176,19 @@ class DeviceListener(TuyaDeviceListener): def add_device(self, device: TuyaDevice) -> None: """Add device added listener.""" - if device.category in TUYA_SUPPORTED_PRODUCT_CATEGORIES: - # Ensure the device isn't present stale - self.hass.add_job(self.async_remove_device, device.id) + # Ensure the device isn't present stale + self.hass.add_job(self.async_remove_device, device.id) - self.device_ids.add(device.id) - dispatcher_send(self.hass, TUYA_DISCOVERY_NEW, [device.id]) + self.device_ids.add(device.id) + dispatcher_send(self.hass, TUYA_DISCOVERY_NEW, [device.id]) - device_manager = self.device_manager - device_manager.mq.stop() - tuya_mq = TuyaOpenMQ(device_manager.api) - tuya_mq.start() + device_manager = self.device_manager + device_manager.mq.stop() + tuya_mq = TuyaOpenMQ(device_manager.api) + tuya_mq.start() - device_manager.mq = tuya_mq - tuya_mq.add_message_listener(device_manager.on_message) + device_manager.mq = tuya_mq + tuya_mq.add_message_listener(device_manager.on_message) def remove_device(self, device_id: str) -> None: """Add device removed listener.""" diff --git a/homeassistant/components/tuya/const.py b/homeassistant/components/tuya/const.py index 9c0c1ce2105..bc1bbd49d98 100644 --- a/homeassistant/components/tuya/const.py +++ b/homeassistant/components/tuya/const.py @@ -84,41 +84,6 @@ TUYA_RESPONSE_MSG = "msg" TUYA_RESPONSE_SUCCESS = "success" TUYA_RESPONSE_PLATFORM_URL = "platform_url" -TUYA_SUPPORTED_PRODUCT_CATEGORIES = ( - "bh", # Smart Kettle - "cwysj", # Pet Water Feeder - "cz", # Socket - "dc", # Light string - "dd", # Light strip - "dj", # Light - "dlq", # Breaker - "fs", # Fan - "fsd", # Ceiling Fan Light - "fwd", # Ambient Light - "fwl", # Ambient light - "gyd", # Motion Sensor Light - "jsq", # Humidifier's light - "kfj", # Coffee maker - "kg", # Switch - "kj", # Air Purifier - "kt", # Air conditioner - "ldcg", # Luminance Sensor - "mcs", # Door Window Sensor - "pc", # Power Strip - "pir", # PIR Detector - "qn", # Heater - "sd", # Robot vacuum - "sgbj", # Siren Alarm - "sos", # SOS Button - "sp", # Smart Camera - "tgq", # Dimmer - "tyndj", # Solar Light - "wk", # Thermostat - "xdd", # Ceiling Light - "xxj", # Diffuser - "zd", # Vibration Sensor -) - TUYA_SMART_APP = "tuyaSmart" SMARTLIFE_APP = "smartlife"