Add missing variables (#185)

* Add missing logger to switch.py * Add logger to number.py * Apply black formatting * Add missing variables
vincentwolsink · Dec 11, 2024 · afeb9a8 · afeb9a8
1 parent 08ca496
commit afeb9a8
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/custom_components/aguaiot/aguaiot.py b/custom_components/aguaiot/aguaiot.py
@@ -157,7 +157,7 @@ async def login(self):
                     response.status_code,
                     response.text,
                 )
-        except httpx.TransportError:
+        except httpx.TransportError as e:
             raise ConnectionError(f"Connection error to {url}: {e}")
 
         if response.status_code != 200:
@@ -203,7 +203,7 @@ async def do_refresh_token(self):
                     response.status_code,
                     response.text,
                 )
-        except httpx.TransportError:
+        except httpx.TransportError as e:
             raise ConnectionError(f"Connection error to {url}: {e}")
 
         if response.status_code != 201:
@@ -298,7 +298,7 @@ async def handle_webcall(self, method, url, payload):
                 response.status_code,
                 response.text,
             )
-        except httpx.TransportError:
+        except httpx.TransportError as e:
             raise ConnectionError(f"Connection error to {url}: {e}")
 
         if response.status_code == 401:

diff --git a/custom_components/aguaiot/number.py b/custom_components/aguaiot/number.py
@@ -1,3 +1,4 @@
+import logging
 from homeassistant.helpers.update_coordinator import (
     CoordinatorEntity,
     DataUpdateCoordinator,
@@ -7,6 +8,8 @@
 from .const import NUMBERS, DOMAIN
 from .aguaiot import AguaIOTError
 
+_LOGGER = logging.getLogger(__name__)
+
 
 async def async_setup_entry(hass, entry, async_add_entities):
     coordinator: DataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][

diff --git a/custom_components/aguaiot/switch.py b/custom_components/aguaiot/switch.py
@@ -1,3 +1,4 @@
+import logging
 from homeassistant.helpers.update_coordinator import (
     CoordinatorEntity,
     DataUpdateCoordinator,
@@ -7,6 +8,8 @@
 from .const import SWITCHES, DOMAIN
 from .aguaiot import AguaIOTError
 
+_LOGGER = logging.getLogger(__name__)
+
 
 async def async_setup_entry(hass, entry, async_add_entities):
     coordinator: DataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][