Change session handling (#8)

home-assistant-ecosystem · Sep 11, 2019 · a97d266 · a97d266
1 parent efa7bfd
commit a97d266
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/connect_box/__init__.py b/connect_box/__init__.py
@@ -77,12 +77,13 @@ async def async_get_devices(self) -> List[Device]:
 
         return self.devices
 
-    async def async_logout(self) -> None:
+    async def async_close_session(self) -> None:
         """Logout and close session."""
         if not self.token:
             return
 
         await self._async_ws_function(CMD_LOGOUT)
+        self.token = None
 
     async def async_initialize_token(self) -> None:
         """Get the token first."""

diff --git a/example.py b/example.py
@@ -15,7 +15,7 @@ async def main():
         await client.async_get_devices()
         print(client.devices)
 
-        await client.async_logout()
+        await client.async_close_session()
 
 
 loop = asyncio.get_event_loop()