91 lines
2.4 KiB
Python
91 lines
2.4 KiB
Python
"""Support for AVM Fritz!Box smarthome devices."""
|
|
import logging
|
|
|
|
from pyfritzhome import Fritzhome, LoginError
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import (
|
|
CONF_DEVICES,
|
|
CONF_HOST,
|
|
CONF_PASSWORD,
|
|
CONF_USERNAME,
|
|
EVENT_HOMEASSISTANT_STOP,
|
|
)
|
|
from homeassistant.helpers import discovery
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
SUPPORTED_DOMAINS = ["binary_sensor", "climate", "switch", "sensor"]
|
|
|
|
DOMAIN = "fritzbox"
|
|
|
|
ATTR_STATE_BATTERY_LOW = "battery_low"
|
|
ATTR_STATE_DEVICE_LOCKED = "device_locked"
|
|
ATTR_STATE_HOLIDAY_MODE = "holiday_mode"
|
|
ATTR_STATE_LOCKED = "locked"
|
|
ATTR_STATE_SUMMER_MODE = "summer_mode"
|
|
ATTR_STATE_WINDOW_OPEN = "window_open"
|
|
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: vol.Schema(
|
|
{
|
|
vol.Required(CONF_DEVICES): vol.All(
|
|
cv.ensure_list,
|
|
[
|
|
vol.Schema(
|
|
{
|
|
vol.Required(CONF_HOST): cv.string,
|
|
vol.Required(CONF_PASSWORD): cv.string,
|
|
vol.Required(CONF_USERNAME): cv.string,
|
|
}
|
|
)
|
|
],
|
|
)
|
|
}
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
|
|
|
|
def setup(hass, config):
|
|
"""Set up the fritzbox component."""
|
|
|
|
fritz_list = []
|
|
|
|
configured_devices = config[DOMAIN].get(CONF_DEVICES)
|
|
for device in configured_devices:
|
|
host = device.get(CONF_HOST)
|
|
username = device.get(CONF_USERNAME)
|
|
password = device.get(CONF_PASSWORD)
|
|
fritzbox = Fritzhome(host=host, user=username, password=password)
|
|
try:
|
|
fritzbox.login()
|
|
_LOGGER.info("Connected to device %s", device)
|
|
except LoginError:
|
|
_LOGGER.warning("Login to Fritz!Box %s as %s failed", host, username)
|
|
continue
|
|
|
|
fritz_list.append(fritzbox)
|
|
|
|
if not fritz_list:
|
|
_LOGGER.info("No fritzboxes configured")
|
|
return False
|
|
|
|
hass.data[DOMAIN] = fritz_list
|
|
|
|
def logout_fritzboxes(event):
|
|
"""Close all connections to the fritzboxes."""
|
|
for fritz in fritz_list:
|
|
fritz.logout()
|
|
|
|
hass.bus.listen_once(EVENT_HOMEASSISTANT_STOP, logout_fritzboxes)
|
|
|
|
for domain in SUPPORTED_DOMAINS:
|
|
discovery.load_platform(hass, domain, DOMAIN, {}, config)
|
|
|
|
return True
|