75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
"""Update the IP addresses of your Cloudflare DNS records."""
|
|
from datetime import timedelta
|
|
import logging
|
|
|
|
from pycfdns import CloudflareUpdater
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import CONF_API_KEY, CONF_EMAIL, CONF_ZONE
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.event import track_time_interval
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONF_RECORDS = "records"
|
|
|
|
DOMAIN = "cloudflare"
|
|
|
|
INTERVAL = timedelta(minutes=60)
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: vol.Schema(
|
|
{
|
|
vol.Required(CONF_EMAIL): cv.string,
|
|
vol.Required(CONF_API_KEY): cv.string,
|
|
vol.Required(CONF_ZONE): cv.string,
|
|
vol.Required(CONF_RECORDS): vol.All(cv.ensure_list, [cv.string]),
|
|
}
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
|
|
|
|
def setup(hass, config):
|
|
"""Set up the Cloudflare component."""
|
|
|
|
cfupdate = CloudflareUpdater()
|
|
email = config[DOMAIN][CONF_EMAIL]
|
|
key = config[DOMAIN][CONF_API_KEY]
|
|
zone = config[DOMAIN][CONF_ZONE]
|
|
records = config[DOMAIN][CONF_RECORDS]
|
|
|
|
def update_records_interval(now):
|
|
"""Set up recurring update."""
|
|
_update_cloudflare(cfupdate, email, key, zone, records)
|
|
|
|
def update_records_service(now):
|
|
"""Set up service for manual trigger."""
|
|
_update_cloudflare(cfupdate, email, key, zone, records)
|
|
|
|
track_time_interval(hass, update_records_interval, INTERVAL)
|
|
hass.services.register(DOMAIN, "update_records", update_records_service)
|
|
return True
|
|
|
|
|
|
def _update_cloudflare(cfupdate, email, key, zone, records):
|
|
"""Update DNS records for a given zone."""
|
|
_LOGGER.debug("Starting update for zone %s", zone)
|
|
|
|
headers = cfupdate.set_header(email, key)
|
|
_LOGGER.debug("Header data defined as: %s", headers)
|
|
|
|
zoneid = cfupdate.get_zoneID(headers, zone)
|
|
_LOGGER.debug("Zone ID is set to: %s", zoneid)
|
|
|
|
update_records = cfupdate.get_recordInfo(headers, zoneid, zone, records)
|
|
_LOGGER.debug("Records: %s", update_records)
|
|
|
|
result = cfupdate.update_records(headers, zoneid, update_records)
|
|
_LOGGER.debug("Update for zone %s is complete", zone)
|
|
|
|
if result is not True:
|
|
_LOGGER.warning(result)
|