made use of python 3.10+ typing features

Author: Zohair-coder

cache/ratings_cache.json

@@ -0,0 +1 @@
+{}

src/db.py

@@ -15,11 +15,10 @@
 
 from datetime import datetime
 from pytz import timezone
+from typing import Any
 
-from typing import Dict, Any
 
-
-def populate_db(data: Dict[str, Dict[str, Any]]) -> None:
+def populate_db(data: dict[str, dict[str, Any]]) -> None:
     cur, conn = connect_to_db()
 
     create_tables(cur)

src/emailer.py

@@ -5,15 +5,14 @@
 
 import boto3
 import config
-from typing import Optional
 
 
 def send_email(subject: str, body: str) -> bool:
     topic_arn = config.topic_arn
     return publish_to_sns(topic_arn, subject, body)
 
 
-def publish_to_sns(topic_arn: Optional[str], subject: str, body: str) -> bool:
+def publish_to_sns(topic_arn: str | None, subject: str, body: str) -> bool:
     sns = boto3.client("sns", endpoint_url=config.sns_endpoint)
 
     if topic_arn is None:  # will be None for local testing
@@ -25,7 +24,7 @@ def publish_to_sns(topic_arn: Optional[str], subject: str, body: str) -> bool:
     return False
 
 
-def get_sns_topic_arn(topic_name: str) -> Optional[str]:
+def get_sns_topic_arn(topic_name: str) -> str | None:
     sns = boto3.client("sns", endpoint_url=config.sns_endpoint)
     response = sns.list_topics()
     for topic in response["Topics"]:

src/parse.py

@@ -3,7 +3,6 @@
 from ratings import rating
 from datetime import datetime
 import re
-from typing import Optional
 
 
 def parse_subject_page(
@@ -74,7 +73,7 @@ def parse_crn_page(html: str, data: dict) -> None:
 
 def get_instructors(
     instructors_str: str, include_ratings: bool, ratings_cache: dict
-) -> Optional[list]:
+) -> list | None:
     if instructors_str == "STAFF":
         return None
 
@@ -149,7 +148,7 @@ def fix_encoding_issue(text: str) -> str:
     return text.replace("\xa0", " ")
 
 
-def parse_days(d: str) -> Optional[list[str]]:
+def parse_days(d: str) -> list[str] | None:
     if d == "TBD":
         return None
 
@@ -170,7 +169,7 @@ def parse_days(d: str) -> Optional[list[str]]:
     return days
 
 
-def parse_time(t: str) -> tuple[Optional[str], Optional[str]]:
+def parse_time(t: str) -> tuple[str | None, str | None]:
     if t == "TBD":
         return (None, None)
     start_str, end_str = t.split(" - ")

src/ratings.py

@@ -4,7 +4,7 @@
 
 import requests
 from helpers import send_request
-from typing import Any, List, Dict, Optional
+from typing import Any
 
 DREXEL_RMP_ID = "U2Nob29sLTE1MjE="
 
@@ -14,7 +14,7 @@
 AUTHORIZATION_HEADER = "Basic dGVzdDp0ZXN0"
 
 
-def search_professors(professor_name: str) -> List[Dict[str, Dict[str, str]]]:
+def search_professors(professor_name: str) -> list[dict[str, dict[str, str]]]:
     query = """query searchProf($query: TeacherSearchQuery!){
         newSearch {
             teachers(query: $query) {
@@ -43,7 +43,7 @@ def search_professors(professor_name: str) -> List[Dict[str, Dict[str, str]]]:
     return response.json()["data"]["newSearch"]["teachers"]["edges"]
 
 
-def get_ratings(id: str) -> Dict[str, Any]:
+def get_ratings(id: str) -> dict[str, Any]:
     query = """query TeacherRatingsPageQuery($id: ID!){
         node(id: $id) {
             __typename
@@ -69,7 +69,7 @@ def get_ratings(id: str) -> Dict[str, Any]:
     return response.json()["data"]["node"]
 
 
-def rating(professor_name: str) -> Optional[Dict[str, int]]:
+def rating(professor_name: str) -> dict[str, int] | None:
     professor = search_professors(professor_name)
 
     if len(professor) == 0: