"""Integrate with DuckDNS.""" from datetime import timedelta import logging import voluptuous as vol from homeassistant.const import CONF_ACCESS_TOKEN, CONF_DOMAIN import homeassistant.helpers.config_validation as cv from homeassistant.helpers.event import async_track_time_interval from homeassistant.helpers.aiohttp_client import async_get_clientsession _LOGGER = logging.getLogger(__name__) ATTR_TXT = 'txt' DOMAIN = 'duckdns' INTERVAL = timedelta(minutes=5) SERVICE_SET_TXT = 'set_txt' UPDATE_URL = 'https://www.duckdns.org/update' CONFIG_SCHEMA = vol.Schema({ DOMAIN: vol.Schema({ vol.Required(CONF_DOMAIN): cv.string, vol.Required(CONF_ACCESS_TOKEN): cv.string, }) }, extra=vol.ALLOW_EXTRA) SERVICE_TXT_SCHEMA = vol.Schema({ vol.Required(ATTR_TXT): vol.Any(None, cv.string) }) async def async_setup(hass, config): """Initialize the DuckDNS component.""" domain = config[DOMAIN][CONF_DOMAIN] token = config[DOMAIN][CONF_ACCESS_TOKEN] session = async_get_clientsession(hass) result = await _update_duckdns(session, domain, token) if not result: return False async def update_domain_interval(now): """Update the DuckDNS entry.""" await _update_duckdns(session, domain, token) async def update_domain_service(call): """Update the DuckDNS entry.""" await _update_duckdns( session, domain, token, txt=call.data[ATTR_TXT]) async_track_time_interval(hass, update_domain_interval, INTERVAL) hass.services.async_register( DOMAIN, SERVICE_SET_TXT, update_domain_service, schema=SERVICE_TXT_SCHEMA) return result _SENTINEL = object() async def _update_duckdns(session, domain, token, *, txt=_SENTINEL, clear=False): """Update DuckDNS.""" params = { 'domains': domain, 'token': token, } if txt is not _SENTINEL: if txt is None: # Pass in empty txt value to indicate it's clearing txt record params['txt'] = '' clear = True else: params['txt'] = txt if clear: params['clear'] = 'true' resp = await session.get(UPDATE_URL, params=params) body = await resp.text() if body != 'OK': _LOGGER.warning("Updating DuckDNS domain failed: %s", domain) return False return True