加了带班信息查询，查询老师list表优化

api/wl_teacher_api.py

@@ -8,7 +8,7 @@ from dao.wl_teacher_dao import (
     list_teachers,
     update_teacher,
     delete_teacher_logic,
-    generate_teacher_no,
+    generate_teacher_no, get_teacher_class_info,
 )
 from database import get_db
 from scheme.wl_teacher_scheme import (
@@ -75,4 +75,21 @@ def edit_teacher(
 # 逻辑删除教师
 @router.delete("/{t_id}")
 def remove_teacher(t_id: int, db: Session = Depends(get_db)):
-    return delete_teacher_logic(db, t_id)
+    teacher = delete_teacher_logic(db, t_id)
+    if not teacher:
+        raise HTTPException(status_code=404, detail="老师不存在")
+    return {"msg": "删除成功"}
+
+
+# 查询老师所带的班级ID
+@router.get("/{t_id}/class")
+def get_class_by_t_id(t_id: int, db: Session = Depends(get_db)):
+    """
+    根据老师ID查询其负责的班级ID
+    """
+    class_id = get_teacher_class_info(db, t_id)
+    if class_id is None:
+        # 如果返回 None，说明老师不存在或者没有分配班级
+        raise HTTPException(status_code=404, detail="老师不存在或未分配班级")
+
+    return {"class_id": class_id}