Add exception for redirect to marketing consent (#284)

* Add exception for redirect to marketing consent during login the process can halt because the user is redirected to a marketing constent page. Without catching this condition a KeyError Location is raised because it's the final page but doesn't start with `myskoda://`. --------- Co-authored-by: Jonas Seydel <[email protected]>
skodaconnect · Dec 16, 2024 · 03d373f · 03d373f
1 parent fd45e77
commit 03d373f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/myskoda/auth/authorization.py b/myskoda/auth/authorization.py
@@ -163,6 +163,8 @@ async def _enter_password(self, csrf: CSRFState) -> IDKAuthorizationCode:
                 while not location.startswith("myskoda://"):
                     if "terms-and-conditions" in location:
                         raise TermsAndConditionsError(location)
+                    if "consent/marketing" in location:
+                        raise MarketingConsentError(location)
                     async with self.session.get(location, allow_redirects=False) as response:
                         location = response.headers["Location"]
                 codes = location.replace("myskoda://redirect/login/#", "")
@@ -324,3 +326,6 @@ class AuthorizationFailedError(Exception):
 
 class TermsAndConditionsError(Exception):
     """Redirect to Terms and Conditions was encountered."""
+
+class MarketingConsentError(Exception):
+    """Redirect to Marketing Consent encountered."""