Bump pycfdns from 2.0.1 to 3.0.0 (#103426)

homeassistant/components/cloudflare/__init__.py

@@ -1,17 +1,12 @@
 """Update the IP addresses of your Cloudflare DNS records."""
 from __future__ import annotations
 
+import asyncio
 from datetime import timedelta
 import logging
 
 from aiohttp import ClientSession
-from pycfdns import CloudflareUpdater
-from pycfdns.exceptions import (
-    CloudflareAuthenticationException,
-    CloudflareConnectionException,
-    CloudflareException,
-    CloudflareZoneException,
-)
+import pycfdns
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_API_TOKEN, CONF_ZONE
@@ -37,32 +32,43 @@
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up Cloudflare from a config entry."""
     session = async_get_clientsession(hass)
-    cfupdate = CloudflareUpdater(
-        session,
-        entry.data[CONF_API_TOKEN],
-        entry.data[CONF_ZONE],
-        entry.data[CONF_RECORDS],
+    client = pycfdns.Client(
+        api_token=entry.data[CONF_API_TOKEN],
+        client_session=session,
     )
 
     try:
-        zone_id = await cfupdate.get_zone_id()
-    except CloudflareAuthenticationException as error:
+        dns_zones = await client.list_zones()
+        dns_zone = next(
+            zone for zone in dns_zones if zone["name"] == entry.data[CONF_ZONE]
+        )
+    except pycfdns.AuthenticationException as error:
         raise ConfigEntryAuthFailed from error
-    except (CloudflareConnectionException, CloudflareZoneException) as error:
+    except pycfdns.ComunicationException as error:
         raise ConfigEntryNotReady from error
 
     async def update_records(now):
         """Set up recurring update."""
         try:
-            await _async_update_cloudflare(session, cfupdate, zone_id)
-        except CloudflareException as error:
+            await _async_update_cloudflare(
+                session, client, dns_zone, entry.data[CONF_RECORDS]
+            )
+        except (
+            pycfdns.AuthenticationException,
+            pycfdns.ComunicationException,
+        ) as error:
             _LOGGER.error("Error updating zone %s: %s", entry.data[CONF_ZONE], error)
 
     async def update_records_service(call: ServiceCall) -> None:
         """Set up service for manual trigger."""
         try:
-            await _async_update_cloudflare(session, cfupdate, zone_id)
-        except CloudflareException as error:
+            await _async_update_cloudflare(
+                session, client, dns_zone, entry.data[CONF_RECORDS]
+            )
+        except (
+            pycfdns.AuthenticationException,
+            pycfdns.ComunicationException,
+        ) as error:
             _LOGGER.error("Error updating zone %s: %s", entry.data[CONF_ZONE], error)
 
     update_interval = timedelta(minutes=DEFAULT_UPDATE_INTERVAL)
@@ -87,18 +93,42 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
 
 async def _async_update_cloudflare(
     session: ClientSession,
-    cfupdate: CloudflareUpdater,
-    zone_id: str,
+    client: pycfdns.Client,
+    dns_zone: pycfdns.ZoneModel,
+    target_records: list[str],
 ) -> None:
-    _LOGGER.debug("Starting update for zone %s", cfupdate.zone)
+    _LOGGER.debug("Starting update for zone %s", dns_zone["name"])
 
-    records = await cfupdate.get_record_info(zone_id)
+    records = await client.list_dns_records(zone_id=dns_zone["id"], type="A")
     _LOGGER.debug("Records: %s", records)
 
     location_info = await async_detect_location_info(session)
 
     if not location_info or not is_ipv4_address(location_info.ip):
         raise HomeAssistantError("Could not get external IPv4 address")
 
-    await cfupdate.update_records(zone_id, records, location_info.ip)
-    _LOGGER.debug("Update for zone %s is complete", cfupdate.zone)
+    filtered_records = [
+        record
+        for record in records
+        if record["name"] in target_records and record["content"] != location_info.ip
+    ]
+
+    if len(filtered_records) == 0:
+        _LOGGER.debug("All target records are up to date")
+        return
+
+    await asyncio.gather(
+        *[
+            client.update_dns_record(
+                zone_id=dns_zone["id"],
+                record_id=record["id"],
+                record_content=location_info.ip,
+                record_name=record["name"],
+                record_type=record["type"],
+                record_proxied=record["proxied"],
+            )
+            for record in filtered_records
+        ]
+    )
+
+    _LOGGER.debug("Update for zone %s is complete", dns_zone["name"])

homeassistant/components/cloudflare/config_flow.py

@@ -5,12 +5,7 @@
 import logging
 from typing import Any
 
-from pycfdns import CloudflareUpdater
-from pycfdns.exceptions import (
-    CloudflareAuthenticationException,
-    CloudflareConnectionException,
-    CloudflareZoneException,
-)
+import pycfdns
 import voluptuous as vol
 
 from homeassistant.components import persistent_notification
@@ -23,6 +18,7 @@
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 
 from .const import CONF_RECORDS, DOMAIN
+from .helpers import get_zone_id
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -33,54 +29,45 @@
 )
 
 
-def _zone_schema(zones: list[str] | None = None) -> vol.Schema:
+def _zone_schema(zones: list[pycfdns.ZoneModel] | None = None) -> vol.Schema:
     """Zone selection schema."""
     zones_list = []
 
     if zones is not None:
-        zones_list = zones
+        zones_list = [zones["name"] for zones in zones]
 
     return vol.Schema({vol.Required(CONF_ZONE): vol.In(zones_list)})
 
 
-def _records_schema(records: list[str] | None = None) -> vol.Schema:
+def _records_schema(records: list[pycfdns.RecordModel] | None = None) -> vol.Schema:
     """Zone records selection schema."""
     records_dict = {}
 
     if records:
-        records_dict = {name: name for name in records}
+        records_dict = {name["name"]: name["name"] for name in records}
 
     return vol.Schema({vol.Required(CONF_RECORDS): cv.multi_select(records_dict)})
 
 
 async def _validate_input(
-    hass: HomeAssistant, data: dict[str, Any]
-) -> dict[str, list[str] | None]:
+    hass: HomeAssistant,
+    data: dict[str, Any],
+) -> dict[str, Any]:
     """Validate the user input allows us to connect.
 
     Data has the keys from DATA_SCHEMA with values provided by the user.
     """
     zone = data.get(CONF_ZONE)
-    records: list[str] | None = None
+    records: list[pycfdns.RecordModel] = []
 
-    cfupdate = CloudflareUpdater(
-        async_get_clientsession(hass),
-        data[CONF_API_TOKEN],
-        zone,
-        [],
+    client = pycfdns.Client(
+        api_token=data[CONF_API_TOKEN],
+        client_session=async_get_clientsession(hass),
     )
 
-    try:
-        zones: list[str] | None = await cfupdate.get_zones()
-        if zone:
-            zone_id = await cfupdate.get_zone_id()
-            records = await cfupdate.get_zone_records(zone_id, "A")
-    except CloudflareConnectionException as error:
-        raise CannotConnect from error
-    except CloudflareAuthenticationException as error:
-        raise InvalidAuth from error
-    except CloudflareZoneException as error:
-        raise InvalidZone from error
+    zones = await client.list_zones()
+    if zone and (zone_id := get_zone_id(zone, zones)) is not None:
+        records = await client.list_dns_records(zone_id=zone_id, type="A")
 
     return {"zones": zones, "records": records}
 
@@ -95,8 +82,8 @@ class CloudflareConfigFlow(ConfigFlow, domain=DOMAIN):
     def __init__(self) -> None:
         """Initialize the Cloudflare config flow."""
         self.cloudflare_config: dict[str, Any] = {}
-        self.zones: list[str] | None = None
-        self.records: list[str] | None = None
+        self.zones: list[pycfdns.ZoneModel] | None = None
+        self.records: list[pycfdns.RecordModel] | None = None
 
     async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> FlowResult:
         """Handle initiation of re-authentication with Cloudflare."""
@@ -195,18 +182,16 @@ async def async_step_records(
 
     async def _async_validate_or_error(
         self, config: dict[str, Any]
-    ) -> tuple[dict[str, list[str] | None], dict[str, str]]:
+    ) -> tuple[dict[str, list[Any]], dict[str, str]]:
         errors: dict[str, str] = {}
         info = {}
 
         try:
             info = await _validate_input(self.hass, config)
-        except CannotConnect:
+        except pycfdns.ComunicationException:
             errors["base"] = "cannot_connect"
-        except InvalidAuth:
+        except pycfdns.AuthenticationException:
             errors["base"] = "invalid_auth"
-        except InvalidZone:
-            errors["base"] = "invalid_zone"
         except Exception:  # pylint: disable=broad-except
             _LOGGER.exception("Unexpected exception")
             errors["base"] = "unknown"
@@ -220,7 +205,3 @@ class CannotConnect(HomeAssistantError):
 
 class InvalidAuth(HomeAssistantError):
     """Error to indicate there is invalid auth."""
-
-
-class InvalidZone(HomeAssistantError):
-    """Error to indicate we cannot validate zone exists in account."""

homeassistant/components/cloudflare/helpers.py

@@ -0,0 +1,10 @@
+"""Helpers for the CloudFlare integration."""
+import pycfdns
+
+
+def get_zone_id(target_zone_name: str, zones: list[pycfdns.ZoneModel]) -> str | None:
+    """Get the zone ID for the target zone name."""
+    for zone in zones:
+        if zone["name"] == target_zone_name:
+            return zone["id"]
+    return None

homeassistant/components/cloudflare/manifest.json

@@ -6,5 +6,5 @@
   "documentation": "https://www.home-assistant.io/integrations/cloudflare",
   "iot_class": "cloud_push",
   "loggers": ["pycfdns"],
-  "requirements": ["pycfdns==2.0.1"]
+  "requirements": ["pycfdns==3.0.0"]
 }

homeassistant/components/cloudflare/strings.json

@@ -30,8 +30,7 @@
     },
     "error": {
       "cannot_connect": "[%key:common::config_flow::error::cannot_connect%]",
-      "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]",
-      "invalid_zone": "Invalid zone"
+      "invalid_auth": "[%key:common::config_flow::error::invalid_auth%]"
     },
     "abort": {
       "reauth_successful": "[%key:common::config_flow::abort::reauth_successful%]",

requirements_all.txt

@@ -1630,7 +1630,7 @@ pybravia==0.3.3
 pycarwings2==2.14
 
 # homeassistant.components.cloudflare
-pycfdns==2.0.1
+pycfdns==3.0.0
 
 # homeassistant.components.channels
 pychannels==1.2.3

requirements_test_all.txt

@@ -1242,7 +1242,7 @@ pybotvac==0.0.24
 pybravia==0.3.3
 
 # homeassistant.components.cloudflare
-pycfdns==2.0.1
+pycfdns==3.0.0
 
 # homeassistant.components.comfoconnect
 pycomfoconnect==0.5.1

tests/components/cloudflare/__init__.py

@@ -3,7 +3,7 @@
 
 from unittest.mock import AsyncMock, patch
 
-from pycfdns import CFRecord
+import pycfdns
 
 from homeassistant.components.cloudflare.const import CONF_RECORDS, DOMAIN
 from homeassistant.const import CONF_API_TOKEN, CONF_ZONE
@@ -26,9 +26,8 @@
 
 USER_INPUT_RECORDS = {CONF_RECORDS: ["ha.mock.com", "homeassistant.mock.com"]}
 
-MOCK_ZONE = "mock.com"
-MOCK_ZONE_ID = "mock-zone-id"
-MOCK_ZONE_RECORDS = [
+MOCK_ZONE: pycfdns.ZoneModel = {"name": "mock.com", "id": "mock-zone-id"}
+MOCK_ZONE_RECORDS: list[pycfdns.RecordModel] = [
     {
         "id": "zone-record-id",
         "type": "A",
@@ -77,21 +76,12 @@ async def init_integration(
     return entry
 
 
-def _get_mock_cfupdate(
-    zone: str = MOCK_ZONE,
-    zone_id: str = MOCK_ZONE_ID,
-    records: list = MOCK_ZONE_RECORDS,
-):
-    client = AsyncMock()
+def _get_mock_client(zone: str = MOCK_ZONE, records: list = MOCK_ZONE_RECORDS):
+    client: pycfdns.Client = AsyncMock()
 
-    zone_records = [record["name"] for record in records]
-    cf_records = [CFRecord(record) for record in records]
-
-    client.get_zones = AsyncMock(return_value=[zone])
-    client.get_zone_records = AsyncMock(return_value=zone_records)
-    client.get_record_info = AsyncMock(return_value=cf_records)
-    client.get_zone_id = AsyncMock(return_value=zone_id)
-    client.update_records = AsyncMock(return_value=None)
+    client.list_zones = AsyncMock(return_value=[zone])
+    client.list_dns_records = AsyncMock(return_value=records)
+    client.update_dns_record = AsyncMock(return_value=None)
 
     return client
 

tests/components/cloudflare/conftest.py

@@ -3,15 +3,15 @@
 
 import pytest
 
-from . import _get_mock_cfupdate
+from . import _get_mock_client
 
 
 @pytest.fixture
 def cfupdate(hass):
     """Mock the CloudflareUpdater for easier testing."""
-    mock_cfupdate = _get_mock_cfupdate()
+    mock_cfupdate = _get_mock_client()
     with patch(
-        "homeassistant.components.cloudflare.CloudflareUpdater",
+        "homeassistant.components.cloudflare.pycfdns.Client",
         return_value=mock_cfupdate,
     ) as mock_api:
         yield mock_api
@@ -20,9 +20,9 @@ def cfupdate(hass):
 @pytest.fixture
 def cfupdate_flow(hass):
     """Mock the CloudflareUpdater for easier config flow testing."""
-    mock_cfupdate = _get_mock_cfupdate()
+    mock_cfupdate = _get_mock_client()
     with patch(
-        "homeassistant.components.cloudflare.config_flow.CloudflareUpdater",
+        "homeassistant.components.cloudflare.pycfdns.Client",
         return_value=mock_cfupdate,
     ) as mock_api:
         yield mock_api