Merge pull request #207 from CoMPaTech/dnsfix

Initial connection DNS fix

Author: CoMPaTech

CHANGELOG.md

@@ -2,6 +2,14 @@
 
 ## Changelog
 
+### JUL 2025 [0.4.2]
+
+- Attempt to fix getting initial code (due to DNS issues) #139
+
+### SEP 2024 [0.4.1]
+
+- Onboarding devices fix
+
 ### AUG 2024 [0.4.0]
 
 - Add multibike support

custom_components/stromer/manifest.json

@@ -8,5 +8,5 @@
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/CoMPaTech/stromer/issues",
   "requirements": [],
-  "version": "0.4.0"
+  "version": "0.4.2"
 }

custom_components/stromer/stromer.py

@@ -2,12 +2,14 @@
 
 __version__ = "0.2.0"
 
+import asyncio
 import json
 import logging
 import re
 from typing import Any
 from urllib.parse import urlencode
 
+import aiodns
 import aiohttp
 
 LOGGER = logging.getLogger(__name__)
@@ -109,12 +111,33 @@ async def stromer_update(self) -> None:
         LOGGER.error("Stromer error: api call failed 10 times, cowardly failing")
         raise ApiError
 
+    async def stromer_api_debouncer(self, url: str, timeout: int = 10, retries: int = 10, delay: int = 10) -> aiohttp.ClientResponse:
+        """Debounce API-request to leverage DNS issues."""
+        for attempt in range(retries):
+            try:
+                log = f"Attempt {attempt + 1}/{retries} to interface with Stromer on {url}"
+                LOGGER.debug(log)
+                res = await self._websession.get(url, timeout=timeout)
+                res.raise_for_status()
+                return res
+            except (aiodns.error.DNSError, aiohttp.ClientError, TimeoutError) as e:
+                log = f"Error getting Stromer API (attempt {attempt + 1}/{retries}): {e}"
+                LOGGER.warning(log)
+                if attempt < retries - 1:
+                    log = f"Retrying in {delay} seconds..."
+                    LOGGER.warning(log)
+                    await asyncio.sleep(delay)
+                else:
+                    log = f"Failed to get to Stromer API after {retries} attempts."
+                    LOGGER.error(log)
+                    raise  # Re-raise the last exception if all retries fail
+
     async def stromer_get_code(self) -> None:
         """Retrieve authorization code from API."""
         url = f"{self.base_url}/mobile/v4/login/"
         if self._api_version == "v3":
             url = f"{self.base_url}/users/login/"
-        res = await self._websession.get(url)
+        res = await self.stromer_api_debouncer(url)
         try:
             cookie = res.headers.get("Set-Cookie")
             pattern = "=(.*?);"

pyproject.toml

@@ -51,7 +51,6 @@ lint.select = [
     "S317",  # suspicious-xml-sax-usage
     "S318",  # suspicious-xml-mini-dom-usage
     "S319",  # suspicious-xml-pull-dom-usage
-    "S320",  # suspicious-xmle-tree-usage
     "S601",  # paramiko-call
     "S602",  # subprocess-popen-with-shell-equals-true
     "S604",  # call-with-shell-equals-true