Use correct device class from pytuya

2020-09-10 10:42:44 +02:00
parent 11763d90c5
commit ae64158da5
3 changed files with 10 additions and 4 deletions
--- a/custom_components/localtuya/init.py
+++ b/custom_components/localtuya/init.py
@@ -36,7 +36,7 @@ BASE_PLATFORM_SCHEMA = {
 }


-def prepare_setup_entities(config_entry, platform):
+def prepare_setup_entities(config_entry, platform, device_class):
    """Prepare ro setup entities for a platform."""
    entities_to_setup = [
        entity
@@ -46,7 +46,7 @@ def prepare_setup_entities(config_entry, platform):
    if not entities_to_setup:
        return None, None

-    device = pytuya.BulbDevice(
+    device = device_class(
        config_entry.data[CONF_DEVICE_ID],
        config_entry.data[CONF_HOST],
        config_entry.data[CONF_LOCAL_KEY],