jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Postpone broadlink platform switch until config entry is ready (#76371)

Browse Source
This commit is contained in:
Joakim Plate 2022-08-07 15:42:47 +02:00 committed by GitHub
parent cd1227d8b9
commit 4aeaeeda0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
homeassistant/components/broadlink/switch.py
View File

@ -26,11 +26,13 @@ from homeassistant.const import (
Platform, Platform,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import BroadlinkDevice
from .const import DOMAIN from .const import DOMAIN
from .entity import BroadlinkEntity from .entity import BroadlinkEntity
from .helpers import data_packet, import_device, mac_address from .helpers import data_packet, import_device, mac_address
@ -80,8 +82,18 @@ async def async_setup_platform(
host = config.get(CONF_HOST) host = config.get(CONF_HOST)
if switches := config.get(CONF_SWITCHES): if switches := config.get(CONF_SWITCHES):
platform_data = hass.data[DOMAIN].platforms.setdefault(Platform.SWITCH, {}) platform_data = hass.data[DOMAIN].platforms.get(Platform.SWITCH, {})
platform_data.setdefault(mac_addr, []).extend(switches) async_add_entities_config_entry: AddEntitiesCallback
device: BroadlinkDevice
async_add_entities_config_entry, device = platform_data.get(
mac_addr, (None, None)
)
if not async_add_entities_config_entry:
raise PlatformNotReady
async_add_entities_config_entry(
BroadlinkRMSwitch(device, config) for config in switches
)
else: else:
_LOGGER.warning( _LOGGER.warning(
@ -104,12 +116,8 @@ async def async_setup_entry(
switches: list[BroadlinkSwitch] = [] switches: list[BroadlinkSwitch] = []
if device.api.type in {"RM4MINI", "RM4PRO", "RMMINI", "RMMINIB", "RMPRO"}: if device.api.type in {"RM4MINI", "RM4PRO", "RMMINI", "RMMINIB", "RMPRO"}:
platform_data = hass.data[DOMAIN].platforms.get(Platform.SWITCH, {}) platform_data = hass.data[DOMAIN].platforms.setdefault(Platform.SWITCH, {})
user_defined_switches = platform_data.get(device.api.mac, {}) platform_data[device.api.mac] = async_add_entities, device
switches.extend(
BroadlinkRMSwitch(device, config) for config in user_defined_switches
)
elif device.api.type == "SP1": elif device.api.type == "SP1":
switches.append(BroadlinkSP1Switch(device)) switches.append(BroadlinkSP1Switch(device))