williamp/localtuya-modded
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use correct device class from pytuya

Browse Source
This commit is contained in:
Pierre Ståhl
2020-09-10 10:42:44 +02:00
parent 11763d90c5
commit ae64158da5
3 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
custom_components/localtuya/__init__.py
View File

@@ -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.""" """Prepare ro setup entities for a platform."""
entities_to_setup = [ entities_to_setup = [
entity entity
@@ -46,7 +46,7 @@ def prepare_setup_entities(config_entry, platform):
if not entities_to_setup: if not entities_to_setup:
return None, None return None, None
device = pytuya.BulbDevice( device = device_class(
config_entry.data[CONF_DEVICE_ID], config_entry.data[CONF_DEVICE_ID],
config_entry.data[CONF_HOST], config_entry.data[CONF_HOST],
config_entry.data[CONF_LOCAL_KEY], config_entry.data[CONF_LOCAL_KEY],

5
custom_components/localtuya/light.py
View File

@@ -34,6 +34,7 @@ from homeassistant.components.light import (
from homeassistant.util import color as colorutil from homeassistant.util import color as colorutil
from . import BASE_PLATFORM_SCHEMA, import_from_yaml, prepare_setup_entities from . import BASE_PLATFORM_SCHEMA, import_from_yaml, prepare_setup_entities
from .pytuya import BulbDevice
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@@ -51,7 +52,9 @@ def flow_schema(dps):
async def async_setup_entry(hass, config_entry, async_add_entities): async def async_setup_entry(hass, config_entry, async_add_entities):
"""Setup a Tuya switch based on a config entry.""" """Setup a Tuya switch based on a config entry."""
device, entities_to_setup = prepare_setup_entities(config_entry, "light") device, entities_to_setup = prepare_setup_entities(
config_entry, "light", BulbDevice
)
if not entities_to_setup: if not entities_to_setup:
return return

5
custom_components/localtuya/switch.py
View File

@@ -48,6 +48,7 @@ from .const import (
CONF_CURRENT_CONSUMPTION, CONF_CURRENT_CONSUMPTION,
CONF_VOLTAGE, CONF_VOLTAGE,
) )
from .pytuya import OutletDevice
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@@ -86,7 +87,9 @@ def flow_schema(dps):
async def async_setup_entry(hass, config_entry, async_add_entities): async def async_setup_entry(hass, config_entry, async_add_entities):
"""Setup a Tuya switch based on a config entry.""" """Setup a Tuya switch based on a config entry."""
device, entities_to_setup = prepare_setup_entities(config_entry, "switch") device, entities_to_setup = prepare_setup_entities(
config_entry, "switch", OutletDevice
)
if not entities_to_setup: if not entities_to_setup:
return return