from sqlalchemy.orm import Session
from model.student_info import Student

# 根据学生id查询学生
def get_student_by_id(db: Session, student_id: str):
    return db.query(Student).filter(Student.id == student_id).first()

# 根据班级名称查询该班级全部学生
def get_student_by_class(db: Session, class_name: str):
    return db.query(Student).filter(Student.class_name == class_name).all()

# 创建学生
def create_student(db: Session, student_id: str, student_name: str, class_name: str):
    new_stu = Student(id=student_id, name=student_name, class_name=class_name, status="待就业")
    db.add(new_stu)
    db.commit()
    db.refresh(new_stu)
    return new_stu