evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use platform enum (1) [A-D] (#60908)

Browse source 
* Use platform enum (1) [A-D]

* Fix imports

* Fix tests

* Use Platform even in tests
This commit is contained in:
Marc Mueller 2021-12-03 17:51:30 +01:00 • committed by GitHub
parent ef458b237c
commit f57d42a9e8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
65 changed files with 272 additions and 183 deletions
Download patch file Download diff file Expand all files Collapse all files

7
homeassistant/components/broadlink/switch.py
View file

@ -22,11 +22,12 @@ from homeassistant.const import (
CONF_TIMEOUT,
CONF_TYPE,
STATE_ON,
Platform,
)
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.restore_state import RestoreEntity
from .const import DOMAIN, SWITCH_DOMAIN
from .const import DOMAIN
from .entity import BroadlinkEntity
from .helpers import data_packet, import_device, mac_address
@ -90,7 +91,7 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
)
if switches:
platform_data = hass.data[DOMAIN].platforms.setdefault(SWITCH_DOMAIN, {})
platform_data = hass.data[DOMAIN].platforms.setdefault(Platform.SWITCH, {})
platform_data.setdefault(mac_addr, []).extend(switches)
else:
@ -110,7 +111,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
switches = []
if device.api.type in {"RM4MINI", "RM4PRO", "RMMINI", "RMMINIB", "RMPRO"}:
platform_data = hass.data[DOMAIN].platforms.get(SWITCH_DOMAIN, {})
platform_data = hass.data[DOMAIN].platforms.get(Platform.SWITCH, {})
user_defined_switches = platform_data.get(device.api.mac, {})
switches.extend(
BroadlinkRMSwitch(device, config) for config in user_defined_switches