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

Move ensure_domain_data

Browse Source
This commit is contained in:
Steven Looman 2018-09-02 16:40:39 +02:00
parent 8bec4a55d1
commit a4a175440b
3 changed files with 14 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
homeassistant/components/igd/__init__.py
View File

@ -21,7 +21,6 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.typing import ConfigType, HomeAssistantType from homeassistant.helpers.typing import ConfigType, HomeAssistantType
from homeassistant.util import get_local_ip from homeassistant.util import get_local_ip
from homeassistant.components.discovery import DOMAIN as DISCOVERY_DOMAIN from homeassistant.components.discovery import DOMAIN as DISCOVERY_DOMAIN
import homeassistant.components.igd.config_flow # noqa: 401
from .const import ( from .const import (
CONF_ENABLE_PORT_MAPPING, CONF_ENABLE_SENSORS, CONF_ENABLE_PORT_MAPPING, CONF_ENABLE_SENSORS,
@ -29,7 +28,7 @@ from .const import (
) )
from .const import DOMAIN from .const import DOMAIN
from .const import LOGGER as _LOGGER from .const import LOGGER as _LOGGER
from .const import ensure_domain_data from .config_flow import ensure_domain_data
REQUIREMENTS = ['async-upnp-client==0.12.4'] REQUIREMENTS = ['async-upnp-client==0.12.4']

14
homeassistant/components/igd/config_flow.py
View File

@ -9,7 +9,19 @@ from .const import (
CONF_SSDP_DESCRIPTION, CONF_UDN CONF_SSDP_DESCRIPTION, CONF_UDN
) )
from .const import DOMAIN from .const import DOMAIN
from .const import ensure_domain_data
def ensure_domain_data(hass):
"""Ensure hass.data is filled properly."""
hass.data[DOMAIN] = hass.data.get(DOMAIN, {})
hass.data[DOMAIN]['devices'] = hass.data[DOMAIN].get('devices', {})
hass.data[DOMAIN]['sensors'] = hass.data[DOMAIN].get('sensors', {})
hass.data[DOMAIN]['discovered'] = hass.data[DOMAIN].get('discovered', {})
hass.data[DOMAIN]['auto_config'] = hass.data[DOMAIN].get('auto_config', {
'active': False,
'port_forward': False,
'sensors': False,
})
@config_entries.HANDLERS.register(DOMAIN) @config_entries.HANDLERS.register(DOMAIN)

13
homeassistant/components/igd/const.py
View File

@ -8,16 +8,3 @@ CONF_ENABLE_PORT_MAPPING = 'port_forward'
CONF_ENABLE_SENSORS = 'sensors' CONF_ENABLE_SENSORS = 'sensors'
CONF_UDN = 'udn' CONF_UDN = 'udn'
CONF_SSDP_DESCRIPTION = 'ssdp_description' CONF_SSDP_DESCRIPTION = 'ssdp_description'
def ensure_domain_data(hass):
"""Ensure hass.data is filled properly."""
hass.data[DOMAIN] = hass.data.get(DOMAIN, {})
hass.data[DOMAIN]['devices'] = hass.data[DOMAIN].get('devices', {})
hass.data[DOMAIN]['sensors'] = hass.data[DOMAIN].get('sensors', {})
hass.data[DOMAIN]['discovered'] = hass.data[DOMAIN].get('discovered', {})
hass.data[DOMAIN]['auto_config'] = hass.data[DOMAIN].get('auto_config', {
'active': False,
'port_forward': False,
'sensors': False,
})