66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
"""Support the UPB PIM."""
|
|
|
|
import upb_lib
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import ATTR_COMMAND, CONF_FILE_PATH, CONF_HOST, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .const import (
|
|
ATTR_ADDRESS,
|
|
ATTR_BRIGHTNESS_PCT,
|
|
ATTR_RATE,
|
|
DOMAIN,
|
|
EVENT_UPB_SCENE_CHANGED,
|
|
)
|
|
|
|
PLATFORMS = [Platform.LIGHT, Platform.SCENE]
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
|
|
"""Set up a new config_entry for UPB PIM."""
|
|
|
|
url = config_entry.data[CONF_HOST]
|
|
file = config_entry.data[CONF_FILE_PATH]
|
|
|
|
upb = upb_lib.UpbPim({"url": url, "UPStartExportFile": file})
|
|
await upb.async_connect()
|
|
hass.data.setdefault(DOMAIN, {})
|
|
hass.data[DOMAIN][config_entry.entry_id] = {"upb": upb}
|
|
|
|
await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
|
|
|
|
def _element_changed(element, changeset):
|
|
if (change := changeset.get("last_change")) is None:
|
|
return
|
|
if change.get("command") is None:
|
|
return
|
|
|
|
hass.bus.async_fire(
|
|
EVENT_UPB_SCENE_CHANGED,
|
|
{
|
|
ATTR_COMMAND: change["command"],
|
|
ATTR_ADDRESS: element.addr.index,
|
|
ATTR_BRIGHTNESS_PCT: change.get("level", -1),
|
|
ATTR_RATE: change.get("rate", -1),
|
|
},
|
|
)
|
|
|
|
for link in upb.links:
|
|
element = upb.links[link]
|
|
element.add_callback(_element_changed)
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
|
|
"""Unload the config_entry."""
|
|
unload_ok = await hass.config_entries.async_unload_platforms(
|
|
config_entry, PLATFORMS
|
|
)
|
|
if unload_ok:
|
|
upb = hass.data[DOMAIN][config_entry.entry_id]["upb"]
|
|
upb.disconnect()
|
|
hass.data[DOMAIN].pop(config_entry.entry_id)
|
|
return unload_ok
|