formatted files

Author: Zohair-coder

src/helpers.py

@@ -5,7 +5,11 @@
 
 
 def send_request(
-    session: Session, url: str, method: str = "GET", json: dict[str, Any] = {}, headers: dict[str, str] = {}
+    session: Session,
+    url: str,
+    method: str = "GET",
+    json: dict[str, Any] = {},
+    headers: dict[str, str] = {},
 ) -> Response:
     try:
         resp = send_request_helper(session, url, method, json, headers)
@@ -42,7 +46,11 @@ def send_request(
 
 
 def send_request_helper(
-    session: Session, url: str, method: str, json: dict[str, Any], headers: dict[str, str]
+    session: Session,
+    url: str,
+    method: str,
+    json: dict[str, Any],
+    headers: dict[str, str],
 ) -> Response:
     timeout = 2
     if method == "GET":

src/parse.py

@@ -5,8 +5,12 @@
 import re
 from typing import Any
 
+
 def parse_subject_page(
-    html: str, data: dict[str, dict[str, Any]], include_ratings: bool = False, ratings_cache: dict[str, dict[str, int] | None] = {}
+    html: str,
+    data: dict[str, dict[str, Any]],
+    include_ratings: bool = False,
+    ratings_cache: dict[str, dict[str, int] | None] = {},
 ) -> dict[str, str]:
     soup = BeautifulSoup(html, "html.parser")
     table_rows = soup.find_all("tr", class_=["odd", "even"])
@@ -72,7 +76,9 @@ def parse_crn_page(html: str, data: dict[str, dict[str, Any]]) -> None:
 
 
 def get_instructors(
-    instructors_str: str, include_ratings: bool, ratings_cache: dict[str, dict[str, int] | None]
+    instructors_str: str,
+    include_ratings: bool,
+    ratings_cache: dict[str, dict[str, int] | None],
 ) -> list[dict[str, Any]] | None:
     if instructors_str == "STAFF":
         return None
@@ -133,7 +139,9 @@ def get_max_enroll(td: Tag) -> str:
     span = td.contents[0]
 
     if not isinstance(span, Tag):
-        raise Exception("Max enrollment HTML span tag inside td not structured as expected")
+        raise Exception(
+            "Max enrollment HTML span tag inside td not structured as expected"
+        )
 
     title_attr = span.attrs["title"]
 

src/ratings.py

@@ -40,7 +40,10 @@ def search_professors(professor_name: str) -> list[dict[str, dict[str, str]]]:
         headers={"Authorization": AUTHORIZATION_HEADER},
     )
 
-    return cast(list[dict[str, dict[str, str]]], response.json()["data"]["newSearch"]["teachers"]["edges"])
+    return cast(
+        list[dict[str, dict[str, str]]],
+        response.json()["data"]["newSearch"]["teachers"]["edges"],
+    )
 
 
 def get_ratings(id: str) -> dict[str, Any]:

src/scrape.py

@@ -9,7 +9,9 @@
 import config
 
 
-def scrape(include_ratings: bool = False, all_colleges: bool = False) -> dict[str, dict[str, Any]]:
+def scrape(
+    include_ratings: bool = False, all_colleges: bool = False
+) -> dict[str, dict[str, Any]]:
     session = Session()
     data: dict[str, dict[str, Any]] = {}
 
@@ -46,7 +48,9 @@ def go_to_college_page(session: Session, college_code: str) -> Response:
     return send_request(session, config.get_college_page_url(college_code))
 
 
-def scrape_all_subjects(session: Session, data: dict[str, dict[str, Any]], html: str, include_ratings: bool) -> dict[str, dict[str, Any]]:
+def scrape_all_subjects(
+    session: Session, data: dict[str, dict[str, Any]], html: str, include_ratings: bool
+) -> dict[str, dict[str, Any]]:
     try:
         with open("cache/extra_course_data_cache.json", "r") as f:
             extra_course_data_cache = json.load(f)