hOn/custom_components/hon/__init__.py

55 lines
1.6 KiB
Python
Raw Normal View History

2023-02-19 02:58:21 +01:00
import logging
2023-06-25 17:33:30 +02:00
from pathlib import Path
2023-02-19 02:58:21 +01:00
import voluptuous as vol # type: ignore[import]
2023-02-19 02:58:21 +01:00
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
from homeassistant.helpers import config_validation as cv, aiohttp_client
from homeassistant.helpers.typing import HomeAssistantType
2023-06-25 17:33:30 +02:00
from pyhon import Hon
2023-03-03 18:23:30 +01:00
from .const import DOMAIN, PLATFORMS
2023-02-19 02:58:21 +01:00
_LOGGER = logging.getLogger(__name__)
HON_SCHEMA = vol.Schema(
{
vol.Required(CONF_EMAIL): cv.string,
vol.Required(CONF_PASSWORD): cv.string,
}
)
CONFIG_SCHEMA = vol.Schema(
{DOMAIN: vol.Schema(vol.All(cv.ensure_list, [HON_SCHEMA]))},
extra=vol.ALLOW_EXTRA,
)
2023-07-24 21:37:48 +02:00
async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool:
2023-02-19 02:58:21 +01:00
session = aiohttp_client.async_get_clientsession(hass)
if (config_dir := hass.config.config_dir) is None:
raise ValueError("Missing Config Dir")
2023-04-10 19:51:16 +02:00
hon = await Hon(
2023-06-25 17:33:30 +02:00
entry.data["email"],
entry.data["password"],
session=session,
test_data_path=Path(config_dir),
2023-04-10 19:51:16 +02:00
).create()
2023-02-19 02:58:21 +01:00
hass.data.setdefault(DOMAIN, {})
hass.data[DOMAIN][entry.unique_id] = hon
hass.data[DOMAIN]["coordinators"] = {}
for platform in PLATFORMS:
hass.async_create_task(
hass.config_entries.async_forward_entry_setup(entry, platform)
)
2023-07-24 21:37:48 +02:00
return True
2023-03-05 00:54:57 +01:00
async def async_unload_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool:
2023-03-05 00:54:57 +01:00
unload = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
if unload:
if not hass.data[DOMAIN]:
hass.data.pop(DOMAIN, None)
return unload