Add a CaptchaException()

EdupageAPI · Apr 20, 2024 · ecc1767 · ecc1767
1 parent 1f35aed
commit ecc1767
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/edupage_api/exceptions.py b/edupage_api/exceptions.py
@@ -72,3 +72,7 @@ class SecondFactorFailedException(Exception):
 
 class InsufficientPermissionsException(Exception):
     pass
+
+
+class CaptchaException(Exception):
+    pass
diff --git a/edupage_api/login.py b/edupage_api/login.py
@@ -5,7 +5,7 @@
 
 from requests import Response
 
-from edupage_api.exceptions import (BadCredentialsException,
+from edupage_api.exceptions import (BadCredentialsException, CaptchaException,
                                     MissingDataException, RequestError,
                                     SecondFactorFailedException)
 from edupage_api.module import EdupageModule, Module
@@ -175,6 +175,9 @@ def login(
 
         response = self.edupage.session.post(request_url, parameters)
 
+        if "cap=1" in response.url or "lerr=b43b43" in response.url:
+            raise CaptchaException()
+
         if "bad=1" in response.url:
             raise BadCredentialsException()
-Original file line number
+Diff line change
@@ Expand Up / @@ -72,3 +72,7 @@ class SecondFactorFailedException(Exception): @@
     class InsufficientPermissionsException(Exception):
         pass
+    class CaptchaException(Exception):
+        pass