Merge pull request #88 from RobertD502/fix_device_query

Fix device query to use renamed function
RobertD502 · Nov 4, 2024 · f124257 · f124257
2 parents e3a3975 + 13b215b
commit f124257
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/custom_components/petkit/manifest.json b/custom_components/petkit/manifest.json
@@ -14,5 +14,5 @@
     "petkitaio==0.1.11.2",
     "tzlocal==5.1"
   ],
-  "version": "0.1.12.1"
+  "version": "0.1.12.2"
 }
diff --git a/custom_components/petkit/util.py b/custom_components/petkit/util.py
@@ -30,7 +30,7 @@ async def async_validate_api(hass: HomeAssistant, email: str, password: str, reg
     )
     try:
         async with async_timeout.timeout(TIMEOUT):
-            devices_query = await client.get_device_roster()
+            devices_query = await client.get_device_rosters()
     except AuthError as err:
         LOGGER.error(f'Could not authenticate on PetKit servers: {err}')
         raise AuthError(err)
@@ -51,9 +51,10 @@ async def async_validate_api(hass: HomeAssistant, email: str, password: str, reg
         LOGGER.error(f'Failed to get information from PetKit servers: {err}')
         raise ConnectionError from err
 
-    devices: dict[str, Any] = devices_query['result']['devices']
+    devices = sum(len(value['result']['devices']) for value in devices_query.values())
 
-    if not devices:
+
+    if devices == 0:
         LOGGER.error("Could not retrieve any devices from PetKit servers")
         raise NoDevicesError
     return True