feat: add error check for invalid user agreement (#49)

* minor: add error check for invalid user agreement

* fix: lint

* feat: add no user agreement error

* fix: version issue

* fix: added account to str

Author: Lash-L

pyproject.toml

@@ -27,7 +27,7 @@ pycryptodome = "~3.17.0"
 pycryptodomex = {version = "~3.17.0", markers = "sys_platform == 'darwin'"}
 paho-mqtt = "~1.6.1"
 dacite = "~1.8.0"
-construct = "^2.10.68"
+construct = "^2.10.56"
 
 
 [build-system]

roborock/api.py

@@ -39,6 +39,8 @@
     RoborockException,
     RoborockInvalidCode,
     RoborockInvalidEmail,
+    RoborockInvalidUserAgreement,
+    RoborockNoUserAgreement,
     RoborockTimeout,
     RoborockUrlException,
     VacuumError,
@@ -474,6 +476,12 @@ async def code_login(self, code) -> UserData:
         if response_code != 200:
             if response_code == 2018:
                 raise RoborockInvalidCode("Invalid code - check your code and try again.")
+            if response_code == 3009:
+                raise RoborockNoUserAgreement("You must accept the user agreement in the Roborock app to continue.")
+            if response_code == 3006:
+                raise RoborockInvalidUserAgreement(
+                    "User agreement must be accepted again - or you are attempting to use the Mi Home app account."
+                )
             raise RoborockException(f"{login_response.get('msg')} - response code: {response_code}")
         user_data = login_response.get("data")
         if not isinstance(user_data, dict):

roborock/exceptions.py

@@ -52,3 +52,11 @@ class RoborockInvalidCode(RoborockException):
 
 class RoborockInvalidEmail(RoborockException):
     """Class for Roborock invalid formatted email exceptions."""
+
+
+class RoborockInvalidUserAgreement(RoborockException):
+    """Class for Roborock invalid user agreement exceptions."""
+
+
+class RoborockNoUserAgreement(RoborockException):
+    """Class for Roborock no user agreement exceptions."""