fix: improve login handling

Improve login handling

library_test.py

@@ -1,5 +1,6 @@
 import argparse
 import asyncio
+import logging
 
 from aiovodafone.api import VodafoneStationApi
 
@@ -35,15 +36,21 @@ async def main() -> None:
     api = VodafoneStationApi(args.router, args.ssl, args.username, args.password)
     logged = await api.login()
     print("Logged:", logged)
+    if not logged:
+        exit(1)
+
     print("-" * 20)
     devices = await api.get_all_devices()
     print("Devices:", devices)
     print("-" * 20)
     data = await api.get_user_data()
     print("Data:", data)
     print("-" * 20)
+    print("Logout & close session")
     await api.logout()
+    await api.close()
 
 
 if __name__ == "__main__":
+    logging.basicConfig(level=logging.DEBUG)
     asyncio.run(main())

src/aiovodafone/api.py

@@ -3,7 +3,6 @@
 import hashlib
 import hmac
 import html
-import logging
 import re
 import urllib.parse
 from dataclasses import dataclass
@@ -13,9 +12,8 @@
 
 import aiohttp
 
-from .exceptions import CannotAuthenticate, CannotConnect
-
-_LOGGER = logging.getLogger(__package__)
+from .const import _LOGGER, LOGIN
+from .exceptions import AlreadyLogged, CannotAuthenticate, CannotConnect
 
 
 @dataclass
@@ -255,10 +253,19 @@ async def login(self) -> bool:
             verify_ssl=False,
             allow_redirects=True,
         )
-        if reply.status != 200:
+        reply_json = await reply.json(content_type="text/html")
+        _LOGGER.debug("Login result: %s[%s]", LOGIN[int(reply_json)], reply_json)
+
+        if reply_json == "1":
+            return True
+
+        if reply_json == "2":
+            raise AlreadyLogged
+
+        if reply_json in ["3", "4"]:
             raise CannotAuthenticate
 
-        return True
+        return False
 
     async def logout(self) -> None:
         """Router logout."""

src/aiovodafone/const.py

@@ -0,0 +1,13 @@
+"""Constants for Vodafone Station."""
+import logging
+
+_LOGGER = logging.getLogger(__package__)
+
+LOGIN = [
+    "-",
+    "logged",
+    "already logged",
+    "credential error",
+    "credential error",
+    "password mismatch",
+]

src/aiovodafone/exceptions.py

@@ -12,3 +12,7 @@ class CannotConnect(VodafoneError):
 
 class CannotAuthenticate(VodafoneError):
     """Exception raised when credentials are incorrect."""
+
+
+class AlreadyLogged(VodafoneError):
+    """Exception raised if a user is already logged."""