formatted with ruff

Author: Zohair-coder

src/login.py

@@ -24,23 +24,44 @@ def login_with_drexel_connect(session: Session) -> Session:
     }
 
     # this should send the credentials and send the MFA request
-    response = send_request(session, config.drexel_connect_base_url + form_action_path, data=login_payload, method="POST")
-    assert response.status_code == 200, "Failed to send request to Drexel Connect with username and password"
+    response = send_request(
+        session,
+        config.drexel_connect_base_url + form_action_path,
+        data=login_payload,
+        method="POST",
+    )
+    assert (
+        response.status_code == 200
+    ), "Failed to send request to Drexel Connect with username and password"
 
     soup = BeautifulSoup(response.text, "html.parser")
     data = parse_initial_mfa_page(soup)
 
-    response = send_request(session, config.drexel_connect_base_url + data["url"], data=data["form-data"], method="POST")
-    assert response.status_code == 200, "Failed to request MFA code page from Drexel Connect"
+    response = send_request(
+        session,
+        config.drexel_connect_base_url + data["url"],
+        data=data["form-data"],
+        method="POST",
+    )
+    assert (
+        response.status_code == 200
+    ), "Failed to request MFA code page from Drexel Connect"
     json_response = response.json()
 
     data = {
         json_response["csrfN"]: json_response["csrfV"],
         "_eventId": json_response["actValue"],
     }
 
-    response = send_request(session, config.drexel_connect_base_url + json_response["flowExURL"], data=data, method="POST")
-    assert response.status_code == 200, "Failed to receive MFA code page from Drexel Connect"
+    response = send_request(
+        session,
+        config.drexel_connect_base_url + json_response["flowExURL"],
+        data=data,
+        method="POST",
+    )
+    assert (
+        response.status_code == 200
+    ), "Failed to receive MFA code page from Drexel Connect"
     soup = BeautifulSoup(response.text, "html.parser")
 
     parsed_data = parse_final_mfa_page(soup)
@@ -53,8 +74,15 @@ def login_with_drexel_connect(session: Session) -> Session:
         "j_mfaToken": totp_code,
     }
 
-    response = send_request(session, config.drexel_connect_base_url + parsed_data["url"], data=data, method="POST")
-    assert response.status_code == 200, "Failed to send MFA code to Drexel Connect (final step)"
+    response = send_request(
+        session,
+        config.drexel_connect_base_url + parsed_data["url"],
+        data=data,
+        method="POST",
+    )
+    assert (
+        response.status_code == 200
+    ), "Failed to send MFA code to Drexel Connect (final step)"
 
     return session
 

src/scrape.py

@@ -29,12 +29,12 @@ def scrape(
             print("Error logging in to Drexel Connect: ")
             print(traceback.format_exc())
             print("Trying again...")
-        
+
         failiure_count += 1
         if failiure_count > 5:
             raise Exception("Failed to log in to Drexel Connect after 5 attempts")
         time.sleep(1)
-        
+
     data: dict[str, dict[str, Any]] = {}
 
     if not all_colleges: