432d9a8f19
Co-authored-by: Marvin Wichmann <marvin@fam-wichmann.de> Co-authored-by: J. Nick Koston <nick@koston.org> Co-authored-by: Jan Stienstra <65826735+j-stienstra@users.noreply.github.com> Co-authored-by: Franck Nijhof <frenck@frenck.nl> Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io> Co-authored-by: Franck Nijhof <git@frenck.dev> Co-authored-by: Paulus Schoutsen <balloob@gmail.com>
52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
"""WiZ Platform integration."""
|
|
from dataclasses import dataclass
|
|
import logging
|
|
|
|
from pywizlight import wizlight
|
|
from pywizlight.exceptions import WizLightConnectionError, WizLightTimeOutError
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_HOST
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
|
|
from .const import DOMAIN
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
PLATFORMS = ["light"]
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up the wiz integration from a config entry."""
|
|
ip_address = entry.data.get(CONF_HOST)
|
|
_LOGGER.debug("Get bulb with IP: %s", ip_address)
|
|
try:
|
|
bulb = wizlight(ip_address)
|
|
scenes = await bulb.getSupportedScenes()
|
|
await bulb.getMac()
|
|
except (
|
|
WizLightTimeOutError,
|
|
WizLightConnectionError,
|
|
ConnectionRefusedError,
|
|
) as err:
|
|
raise ConfigEntryNotReady from err
|
|
|
|
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = WizData(bulb=bulb, scenes=scenes)
|
|
hass.config_entries.async_setup_platforms(entry, PLATFORMS)
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
|
|
hass.data[DOMAIN].pop(entry.entry_id)
|
|
return unload_ok
|
|
|
|
|
|
@dataclass
|
|
class WizData:
|
|
"""Data for the wiz integration."""
|
|
|
|
bulb: wizlight
|
|
scenes: list
|