Fix mysensors device tracker (#47536)

pull/47542/head
Martin Hjelmare 2021-03-06 23:41:43 +01:00 committed by GitHub
parent b01a6367cc
commit f542b360d5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 16 deletions

View File

@ -8,6 +8,7 @@ from mysensors import BaseAsyncGateway
import voluptuous as vol
from homeassistant import config_entries
from homeassistant.components.device_tracker import DOMAIN as DEVICE_TRACKER_DOMAIN
from homeassistant.components.mqtt import valid_publish_topic, valid_subscribe_topic
from homeassistant.components.notify import DOMAIN as NOTIFY_DOMAIN
from homeassistant.config_entries import ConfigEntry
@ -195,11 +196,14 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool
hass.data[DOMAIN][MYSENSORS_GATEWAYS][entry.entry_id] = gateway
# Connect notify discovery as that integration doesn't support entry forwarding.
# Allow loading device tracker platform via discovery
# until refactor to config entry is done.
load_notify_platform = partial(
for platform in (DEVICE_TRACKER_DOMAIN, NOTIFY_DOMAIN):
load_discovery_platform = partial(
async_load_platform,
hass,
NOTIFY_DOMAIN,
platform,
DOMAIN,
hass_config=hass.data[DOMAIN][DATA_HASS_CONFIG],
)
@ -209,8 +213,8 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool
entry.entry_id,
async_dispatcher_connect(
hass,
MYSENSORS_DISCOVERY.format(entry.entry_id, NOTIFY_DOMAIN),
load_notify_platform,
MYSENSORS_DISCOVERY.format(entry.entry_id, platform),
load_discovery_platform,
),
)