Update utils/enrollment.py

- Use grade (Union[str, Grade])

utils/enrollment.py

@@ -1,44 +1,44 @@
-from typing import Optional
-from utils.enums.grade import Grade
-
-from .student import Student
-
-from sqlmodel import SQLModel, Field, Relationship
-
-
-class Enrollment(SQLModel, table=True):
-    """
-    Represents an Enrollment, including the student, the course they are enrolled in, and the assigned grade.
-
-    Attributes:
-        id: (int): The enrollment ID
-        student_id (str): The ID of the student who is enrolled in the course.
-        course_id (str): The ID of the course in which the student is enrolled.
-        grade (Grade): The grade assigned to the student for the course. Default if NO_GRADE with enum value of None if no grade has been assigned yet.
-    """
-
-    id: int = Field(primary_key=True)
-    student_id: str = Field(foreign_key="student.id")
-    course_id: str = Field(foreign_key="course.id")
-    grade: Grade = Field(sa_column=Field(sa_type=Grade))
-
-    course: "Course" = Relationship(
-        back_populates="enrollments")
-
-    def assign_grade(self, grade: Grade) -> None:
-        """
-        Assigns a grade to the student for the course.
-
-        Args:
-            grade (Grade): The grade to assign to the student.
-        """
-        self.grade = grade
-
-    def __str__(self) -> str:
-        """
-        Returns a string representation of the Enrollment.
-
-        Returns:
-            str: A description of the enrollment including the student id_number, course_id, and grade.
-        """
-        return f"Enrollment(student id: {self.student_id}, course: {self.course_id}, grade: {self.grade.value})"
+from typing import Optional
+from utils.enums.grade import Grade
+
+from .student import Student
+
+from sqlmodel import SQLModel, Field, Relationship
+
+
+class Enrollment(SQLModel, table=True):
+    """
+    Represents an Enrollment, including the student, the course they are enrolled in, and the assigned grade.
+
+    Attributes:
+        id: (int): The enrollment ID
+        student_id (str): The ID of the student who is enrolled in the course.
+        course_id (str): The ID of the course in which the student is enrolled.
+        grade (Union[str, Grade]): The grade assigned to the student for the course. Default if NO_GRADE with enum value of None if no grade has been assigned yet.
+    """
+
+    id: int = Field(primary_key=True)
+    student_id: str = Field(foreign_key="student.id")
+    course_id: str = Field(foreign_key="course.id")
+    grade: Union[str, Grade] = Field(default=Grade.A_PLUS)
+
+    course: "Course" = Relationship(
+        back_populates="enrollments")
+
+    def assign_grade(self, grade: Grade) -> None:
+        """
+        Assigns a grade to the student for the course.
+
+        Args:
+            grade (Grade): The grade to assign to the student.
+        """
+        self.grade = grade
+
+    def __str__(self) -> str:
+        """
+        Returns a string representation of the Enrollment.
+
+        Returns:
+            str: A description of the enrollment including the student id_number, course_id, and grade.
+        """
+        return f"Enrollment(student id: {self.student_id}, course: {self.course_id}, grade: {self.grade.value})"