Merge pull request #98 from Zohair-coder/sso

More bandaid

Author: Zohair-coder

.gitignore

@@ -9,4 +9,5 @@ profile_output.pstat
 callgrind.out.profile
 .venv
 test.py
+test_bruteforce.py
 test.txt

.vscode/settings.json

@@ -0,0 +1,7 @@
+{
+    "python.testing.pytestArgs": [
+        "src"
+    ],
+    "python.testing.unittestEnabled": false,
+    "python.testing.pytestEnabled": true
+}

src/scrape.py

@@ -19,7 +19,9 @@ def scrape(
 
     is_logged_into_drexel_connect = False
     failiure_count = 0
+    reset_period = 1  # seconds
     while not is_logged_into_drexel_connect:
+        reset_period *= 2
         try:
             session = login.login_with_drexel_connect(session)
             if "shib_idp_session" in session.cookies:
@@ -28,12 +30,17 @@ def scrape(
         except Exception:
             print("Error logging in to Drexel Connect: ")
             print(traceback.format_exc())
-            print("Trying again...")
+            print(f"Trying again in {reset_period} seconds...")
 
         failiure_count += 1
-        if failiure_count > 5:
-            raise Exception("Failed to log in to Drexel Connect after 5 attempts")
-        time.sleep(1)
+        if failiure_count > 8:
+            raise Exception(
+                "Failed to log in to Drexel Connect after {} attempts".format(
+                    failiure_count
+                )
+            )
+
+        time.sleep(reset_period)
 
     data: dict[str, dict[str, Any]] = {}