Bump pyloadapi to v1.2.0 (#120218)

home-assistant · Jun 23, 2024 · f0d5640 · f0d5640
1 parent bc45dcb
commit f0d5640
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 10 deletions.
diff --git a/homeassistant/components/pyload/manifest.json b/homeassistant/components/pyload/manifest.json
@@ -6,5 +6,5 @@
   "integration_type": "service",
   "iot_class": "local_polling",
   "loggers": ["pyloadapi"],
-  "requirements": ["PyLoadAPI==1.1.0"]
+  "requirements": ["PyLoadAPI==1.2.0"]
 }
diff --git a/homeassistant/components/pyload/sensor.py b/homeassistant/components/pyload/sensor.py
@@ -6,11 +6,15 @@
 from enum import StrEnum
 import logging
 from time import monotonic
-from typing import Any
 
 from aiohttp import CookieJar
-from pyloadapi.api import PyLoadAPI
-from pyloadapi.exceptions import CannotConnect, InvalidAuth, ParserError
+from pyloadapi import (
+    CannotConnect,
+    InvalidAuth,
+    ParserError,
+    PyLoadAPI,
+    StatusServerResponse,
+)
 import voluptuous as vol
 
 from homeassistant.components.sensor import (
@@ -132,7 +136,7 @@ def __init__(
         self.api = api
         self.entity_description = entity_description
         self._attr_available = False
-        self.data: dict[str, Any] = {}
+        self.data: StatusServerResponse
 
     async def async_update(self) -> None:
         """Update state of sensor."""
@@ -167,7 +171,7 @@ async def async_update(self) -> None:
             self._attr_available = False
             return
         else:
-            self.data = status.to_dict()
+            self.data = status
             _LOGGER.debug(
                 "Finished fetching pyload data in %.3f seconds",
                 monotonic() - start,

diff --git a/requirements_all.txt b/requirements_all.txt
@@ -60,7 +60,7 @@ PyFlume==0.6.5
 PyFronius==0.7.3
 
 # homeassistant.components.pyload
-PyLoadAPI==1.1.0
+PyLoadAPI==1.2.0
 
 # homeassistant.components.mvglive
 PyMVGLive==1.1.4

diff --git a/requirements_test_all.txt b/requirements_test_all.txt
@@ -51,7 +51,7 @@ PyFlume==0.6.5
 PyFronius==0.7.3
 
 # homeassistant.components.pyload
-PyLoadAPI==1.1.0
+PyLoadAPI==1.2.0
 
 # homeassistant.components.met_eireann
 PyMetEireann==2021.8.0

diff --git a/tests/components/pyload/conftest.py b/tests/components/pyload/conftest.py
@@ -47,7 +47,7 @@ def mock_pyloadapi() -> Generator[AsyncMock, None, None]:
     ):
         client = mock_client.return_value
         client.username = "username"
-        client.login.return_value = LoginResponse.from_dict(
+        client.login.return_value = LoginResponse(
             {
                 "_permanent": True,
                 "authenticated": True,
@@ -59,7 +59,8 @@ def mock_pyloadapi() -> Generator[AsyncMock, None, None]:
                 "_flashes": [["message", "Logged in successfully"]],
             }
         )
-        client.get_status.return_value = StatusServerResponse.from_dict(
+
+        client.get_status.return_value = StatusServerResponse(
             {
                 "pause": False,
                 "active": 1,
@@ -71,5 +72,6 @@ def mock_pyloadapi() -> Generator[AsyncMock, None, None]:
                 "captcha": False,
             }
         )
+
         client.free_space.return_value = 99999999999
         yield client