Updated readme and added better update error handling

dcmeglio · Jan 7, 2023 · f835a8f · f835a8f
1 parent 389fae9
commit f835a8f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/README → README.md b/README → README.md
@@ -1,6 +1,7 @@
 # homeassistant-petsafe
 
 [![hacs_badge](https://img.shields.io/badge/HACS-Default-41BDF5.svg?style=for-the-badge)](https://github.com/hacs/integration)
+<a href="https://www.buymeacoffee.com/dcmeglio" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/default-orange.png" alt="Buy Me A Coffee" height="28" width="119"></a>
 
 PetSafe Integration for Home Assistant
 

diff --git a/custom_components/petsafe/__init__.py b/custom_components/petsafe/__init__.py
@@ -18,7 +18,7 @@
 )
 from homeassistant.core import HomeAssistant, ServiceCall
 from homeassistant.exceptions import ConfigEntryAuthFailed
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
 import petsafe
 
@@ -162,7 +162,11 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
 
 
 class PetSafeData:
-    def __init__(self, feeders, litterboxes):
+    def __init__(
+        self,
+        feeders: list[petsafe.devices.DeviceSmartFeed],
+        litterboxes: list[petsafe.devices.DeviceScoopfree],
+    ):
         self.feeders = feeders
         self.litterboxes = litterboxes
 
@@ -226,4 +230,6 @@ async def _async_update_data(self) -> PetSafeData:
             if ex.response.status_code in (401, 403):
                 raise ConfigEntryAuthFailed() from ex
             else:
-                raise
+                raise UpdateFailed() from ex
+        except Exception as ex:
+            raise UpdateFailed() from ex