formatted with ruff

Author: Zohair-coder

src/login.py

@@ -5,64 +5,77 @@
 import config
 import totp
 
+
 def login_with_drexel_connect(session: Session) -> Session:
     response = session.get(config.drexel_connect_base_url)
     soup = BeautifulSoup(response.text, "html.parser")
 
     csrf_token = extract_csrf_token(soup)
     form_action_path = extract_form_action_path(soup)
-    
+
     login_payload = {
         "j_username": config.drexel_username,
         "j_password": config.drexel_password,
         "csrf_token": csrf_token,
-        "_eventId_proceed": ""
+        "_eventId_proceed": "",
     }
 
     # this should send the credentials and send the MFA request
-    response = session.post(config.drexel_connect_base_url + form_action_path, data=login_payload)
+    response = session.post(
+        config.drexel_connect_base_url + form_action_path, data=login_payload
+    )
 
     soup = BeautifulSoup(response.text, "html.parser")
     data = parse_initial_mfa_page(soup)
 
-    response = session.post(config.drexel_connect_base_url + data["url"], data=data["form-data"])
+    response = session.post(
+        config.drexel_connect_base_url + data["url"], data=data["form-data"]
+    )
     json_response = response.json()
 
     data = {
         json_response["csrfN"]: json_response["csrfV"],
         "_eventId": json_response["actValue"],
     }
 
-    response = session.post(config.drexel_connect_base_url + json_response["flowExURL"], data=data)
+    response = session.post(
+        config.drexel_connect_base_url + json_response["flowExURL"], data=data
+    )
     soup = BeautifulSoup(response.text, "html.parser")
 
     parsed_data = parse_final_mfa_page(soup)
-    
+
     totp_code = totp.get_token(config.drexel_mfa_secret_key)
 
     data = {
         "csrf_token": parsed_data["csrf_token"],
         "_eventId": "proceed",
-        "j_mfaToken": totp_code
+        "j_mfaToken": totp_code,
     }
 
-    response = session.post(config.drexel_connect_base_url + parsed_data["url"], data=data)
+    response = session.post(
+        config.drexel_connect_base_url + parsed_data["url"], data=data
+    )
 
     return session
 
+
 def extract_csrf_token(soup: BeautifulSoup) -> str:
-    csrf_token_input_tag = soup.find("input", {"name": "csrf_token"}) 
+    csrf_token_input_tag = soup.find("input", {"name": "csrf_token"})
 
     if not isinstance(csrf_token_input_tag, Tag):
         raise Exception("Could not find CSRF token.")
-    
+
     csrf_token = csrf_token_input_tag["value"]
 
     if not isinstance(csrf_token, str):
-        raise Exception(f"CSRF token was not a string. Found: {csrf_token} of type: {type(csrf_token)}")
+        raise Exception(
+            f"CSRF token was not a string. Found: {csrf_token} of type: {type(csrf_token)}"
+        )
 
     return csrf_token
 
+
 def extract_form_action_path(soup: BeautifulSoup) -> str:
     # the form is a child of a div with id "login-box"
     login_box_div = soup.find("div", {"id": "login-box"})
@@ -78,10 +91,13 @@ def extract_form_action_path(soup: BeautifulSoup) -> str:
     form_action_path = login_form["action"]
 
     if not isinstance(form_action_path, str):
-        raise Exception(f"Form action path was not a string. Found: {form_action_path} of type: {type(form_action_path)}")
+        raise Exception(
+            f"Form action path was not a string. Found: {form_action_path} of type: {type(form_action_path)}"
+        )
 
     return form_action_path
 
+
 def parse_initial_mfa_page(soup: BeautifulSoup) -> dict[str, str]:
     data = {}
 
@@ -94,29 +110,35 @@ def parse_initial_mfa_page(soup: BeautifulSoup) -> dict[str, str]:
     script_content = script_tag.string
 
     if not isinstance(script_content, str):
-        raise Exception(f"Script content was not a string. Found: {script_content} of type: {type(script_content)}")
+        raise Exception(
+            f"Script content was not a string. Found: {script_content} of type: {type(script_content)}"
+        )
 
-    url_match = re.search(r"url:\s*['\"](/idp/profile/cas/login\?execution=[^'\"]+)['\"]", script_content)
+    url_match = re.search(
+        r"url:\s*['\"](/idp/profile/cas/login\?execution=[^'\"]+)['\"]", script_content
+    )
     if not url_match:
         raise Exception("Could not find MFA URL.")
 
-    event_id_match = event_id_match = re.search(r"data:\s*'_eventId=([^'&]+)&csrf_token", script_content)
+    event_id_match = event_id_match = re.search(
+        r"data:\s*'_eventId=([^'&]+)&csrf_token", script_content
+    )
     if not event_id_match:
         raise Exception("Could not find MFA event ID.")
 
     csrf_token_match = re.search(r"csrf_token=([^'&]+)", script_content)
     if not csrf_token_match:
         raise Exception("Could not find MFA CSRF token.")
-    
+
     data["url"] = url_match.group(1)
     data["form-data"] = {
         "_eventId": event_id_match.group(1),
-        "csrf_token": csrf_token_match.group(1)
+        "csrf_token": csrf_token_match.group(1),
     }
 
-
     return data
 
+
 def parse_final_mfa_page(soup: BeautifulSoup) -> dict[str, str]:
     data: dict[str, str] = {}
 
@@ -125,23 +147,25 @@ def parse_final_mfa_page(soup: BeautifulSoup) -> dict[str, str]:
 
     if not isinstance(form, Tag):
         raise Exception("Could not find OTP form.")
-    
+
     url = form["action"]
 
     if not isinstance(url, str):
         raise Exception(f"Action was not a string. Found: {url} of type: {type(url)}")
-    
+
     csrf_token_input = form.find("input", {"name": "csrf_token"})
 
     if not isinstance(csrf_token_input, Tag):
         raise Exception("Could not find CSRF token input.")
-    
+
     csrf_token = csrf_token_input["value"]
 
     if not isinstance(csrf_token, str):
-        raise Exception(f"CSRF token was not a string. Found: {csrf_token} of type: {type(csrf_token)}")
+        raise Exception(
+            f"CSRF token was not a string. Found: {csrf_token} of type: {type(csrf_token)}"
+        )
 
     data["url"] = url
     data["csrf_token"] = csrf_token
 
-    return data
+    return data

src/scrape.py

@@ -9,6 +9,7 @@
 import config
 import login
 
+
 def scrape(
     include_ratings: bool = False, all_colleges: bool = False
 ) -> dict[str, dict[str, Any]]:

src/totp.py

@@ -4,6 +4,7 @@
 import time
 import hashlib
 
+
 # https://stackoverflow.com/questions/8529265/google-authenticator-implementation-in-python
 def get_token(secret: str) -> str:
     period = 30
@@ -13,4 +14,4 @@ def get_token(secret: str) -> str:
     h = hmac.new(key, msg, hashlib.sha1).digest()
     o = h[19] & 15
     h = (struct.unpack(">I", h[o : o + 4])[0] & 0x7FFFFFFF) % 1000000
-    return str(h).zfill(6)
+    return str(h).zfill(6)