Spaces:
Sleeping
Sleeping
File size: 1,107 Bytes
4d1c551 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
from typing import Optional
from pydantic import model_validator
from sqlmodel import Field
from typing_extensions import Self
from .person import Person
from utils.enums.major import Major
class Student(Person, table=True):
"""
Represents a Student, inheriting from Person, with an additional major field.
Attributes:
major (Major): The major that the student is pursuing.
Methods:
set_id() -> Self: An SQLModel model validator that automatically sets the student's ID with a "STU" prefix through handle_id method in the Person class.
"""
major: Major = Field(sa_column=Field(sa_type=Major))
@model_validator(mode='after')
def set_id(self) -> Self:
self.handle_id(prefix="STU")
return self
def __str__(self) -> str:
"""
Returns a string representation of the student.
Returns:
str: A description of the student including their name, ID number, and major.
"""
return f"Student(name: {self.name}, id: {self.id}, major: {self.major.value})"
|