pysmarlaapi 0.7.1__tar.gz → 0.7.2__tar.gz

{pysmarlaapi-0.7.1 → pysmarlaapi-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pysmarlaapi
-Version: 0.7.1
+Version: 0.7.2
 Summary: Swing2Sleep Smarla API
 Author-email: Robin Lintermann <robin.lintermann@explicatis.com>
 Requires-Python: >=3.11

{pysmarlaapi-0.7.1 → pysmarlaapi-0.7.2}/pysmarlaapi/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.7.1"
+__version__ = "0.7.2"
 
 from .classes import Connection
 from .federwiege import Federwiege

{pysmarlaapi-0.7.1 → pysmarlaapi-0.7.2}/pysmarlaapi/classes/connection.py

@@ -27,22 +27,26 @@ class Connection:
     async def refresh_token(self) -> bool:
         try:
             async with aiohttp.ClientSession(self.url) as session:
-                response = await session.post(
-                    "/api/AppParing/getToken",
-                    headers={"accept": "*/*", "Content-Type": "application/json"},
-                    data=jsonpickle.encode(self.token, unpicklable=False),
-                )
+                content = await self._get_token(session)
         except (aiohttp.ClientError, asyncio.TimeoutError):
             return False
 
-        if response.status != 200:
+        if not content:
             return False
 
-        content = await response.json()
-
         try:
             self.token = AuthToken.from_json(content)
         except ValueError:
             return False
 
         return True
+
+    async def _get_token(self, session: aiohttp.ClientSession):
+        async with await session.post(
+            "/api/AppParing/getToken",
+            headers={"accept": "*/*", "Content-Type": "application/json"},
+            data=jsonpickle.encode(self.token, unpicklable=False),
+        ) as response:
+            if response.status != 200:
+                return None
+            return await response.json()