Refactor request_degreeworks to management script

backend/degree/management/commands/import_degreeworks.py

@@ -0,0 +1,17 @@
+from django.core.management.base import BaseCommand
+import tqdm
+
+from degree.models import DegreePlan, Requirement, Rule
+from degree.utils.request_degreeworks import get_programs, degree_plans_of, audit, write_dp
+
+
+class Command(BaseCommand):
+    help = "TODO: ADD HELP TEXT"
+
+    def handle(self, *args, **kwargs):
+        for year in range(2017, 2023 + 1):
+            print(year)
+            for program in get_programs(year=year):
+                print("\t" + program)
+                for degree_plan in tqdm(degree_plans_of(program), year=year):
+                    write_dp(degree_plan, audit(degree_plan))

backend/degree/models.py

@@ -58,7 +58,7 @@ class DegreePlan(models.Model):
     )
 
     def __str__(self) -> str:
-        return f"DegreePlan: {self.program} {self.degree} in {self.major} with conc. {self.concentration} ({self.year})"
+        return f"{self.program} {self.degree} in {self.major} with conc. {self.concentration} ({self.year})"
 
 
 class Requirement(models.Model):
@@ -103,7 +103,7 @@ class Requirement(models.Model):
     )
 
     def __str__(self) -> str:
-        return f"Requirement: {self.name} ({self.code}), min_cus={self.min_cus}, degree_plan={self.degree_plan}"
+        return f"{self.name} ({self.code}), min_cus={self.min_cus}, degree_plan={self.degree_plan}"
 
 
 class Rule(models.Model):
@@ -168,4 +168,4 @@ class Rule(models.Model):
     )
 
     def __str__(self) -> str:
-        return f"Rule: {self.q}, min_num={self.min_num}, max_num={self.max_num}, min_cus={self.min_cus}, max_cus={self.max_cus}, requirement={self.requirement}"
+        return f"{self.q}, min_num={self.min_num}, max_num={self.max_num}, min_cus={self.min_cus}, max_cus={self.max_cus}, requirement={self.requirement}"

backend/degree/scripts/request_degreeworks.py

@@ -24,10 +24,7 @@
 s.headers.update(headers)
 
 
-def audit(degree_plan: DegreePlan, timeout=30) -> dict:
-    """
-    
-    """
+def audit(degree_plan: DegreePlan, timeout=30):
     payload = {
         "studentId": env["PENN_ID"],
         "isIncludeInprogress": True,
@@ -58,7 +55,7 @@ def audit(degree_plan: DegreePlan, timeout=30) -> dict:
     return res.json()
 
 
-def degree_plans_of(program_code: str, year: int=2023) -> list[DegreePlan]:
+def degree_plans_of(program_code):
     goals_payload = [
         {
             "id": "programCollection",
@@ -67,7 +64,7 @@ def degree_plans_of(program_code: str, year: int=2023) -> list[DegreePlan]:
             "goals": [
                 {
                     "name": "catalogYear",
-                    "selectedChoices": [str(year)],
+                    "selectedChoices": ["2023"],
                     "ruleGoalCode": None,
                     "links": [],
                 },
@@ -384,7 +381,7 @@ def degree_plans_of(program_code: str, year: int=2023) -> list[DegreePlan]:
     return degree_plans
 
 
-def get_programs(timeout=30, year: int=2023) -> str:
+def get_programs(timeout=30):
     goals_payload = [
         {
             "id": "programCollection",
@@ -406,7 +403,7 @@ def get_programs(timeout=30, year: int=2023) -> str:
                     "source": "api/catalogYears",
                     "errorMessage": "",
                     "choices": [],
-                    "selectedChoices": [str(year)],
+                    "selectedChoices": ["2023"],
                     "ruleGoalCode": None,
                     "links": [],
                 },
@@ -2345,18 +2342,15 @@ def get_programs(timeout=30, year: int=2023) -> str:
     return [program["key"] for program in res.json()[0]["goals"][1]["choices"]]
 
 
-def write_dp(dp: DegreePlan, json: dict, dir: str | Path="degreeplans"):
-    with open(Path(
-        dir, 
-        f"{dp.year}-{dp.program}-{dp.degree}-{dp.major}-{dp.concentration}"
-    )) as f:
+def write_dp(dp: DegreePlan, json: dict, dir: str | Path = "degreeplans"):
+    with open(Path(dir, f"{dp.year}-{dp.program}-{dp.degree}-{dp.major}-{dp.concentration}")) as f:
         json.dump(json, f, indent=4)
 
+
 if __name__ == "__main__":
     for year in range(2017, 2023 + 1):
         print(year)
         for program in get_programs(year=year):
             print("\t" + program)
             for degree_plan in tqdm(degree_plans_of(program), year=year):
                 write_dp(degree_plan, audit(degree_plan))
-