97 lines
2.8 KiB
Python
97 lines
2.8 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
|
|
from core.database import get_db
|
|
from schemas.student import StudentCreate, StudentUpdate
|
|
from dao import student as student_dao
|
|
student_router = APIRouter()
|
|
|
|
@student_router.post("",summary='添加学生')
|
|
def create_student(request:StudentCreate,db:Session = Depends(get_db)):
|
|
d = request.model_dump()
|
|
r = student_dao.create_students(d,db)
|
|
if not r:
|
|
raise HTTPException(status_code=500,detail='服务器繁忙，请稍后添加！')
|
|
return {'code':200,'detail':"添加学生成功"}
|
|
|
|
@student_router.put("/{id}", summary="修改学生")
|
|
def update_student(
|
|
id: int,
|
|
request: StudentUpdate,
|
|
db: Session = Depends(get_db),
|
|
):
|
|
# 只取前端真正传来的字段
|
|
# 例如只传 age，就只修改 age
|
|
data = request.model_dump(exclude_unset=True)
|
|
|
|
student = student_dao.update_student(id, data, db)
|
|
|
|
# None 表示学生不存在
|
|
if student is None:
|
|
raise HTTPException(status_code=404, detail="学生不存在")
|
|
|
|
# False 表示操作数据库失败
|
|
if student is False:
|
|
raise HTTPException(status_code=500, detail="修改学生失败")
|
|
|
|
return {
|
|
"code": 200,
|
|
"detail": "修改学生成功",
|
|
"data": student,
|
|
}
|
|
|
|
@student_router.get("", summary="查询学生列表")
|
|
def get_students(
|
|
# 三个查询条件都可以不传
|
|
student_no: str | None = None,
|
|
student_name: str | None = None,
|
|
class_id: int | None = None,
|
|
# 获取本次请求使用的数据库连接
|
|
db: Session = Depends(get_db),
|
|
):
|
|
# 调用 DAO 查询数据库
|
|
students = student_dao.get_students(
|
|
db=db,
|
|
student_no=student_no,
|
|
student_name=student_name,
|
|
class_id=class_id,
|
|
)
|
|
|
|
return {
|
|
"code": 200,
|
|
"detail": "查询成功",
|
|
"total": len(students),
|
|
"data": students,
|
|
}
|
|
|
|
@student_router.get("/{id}", summary="查询学生详情")
|
|
def get_student(id: int, db: Session = Depends(get_db)):
|
|
# 根据路径中的 ID 查询学生
|
|
student = student_dao.get_student_by_id(id, db)
|
|
|
|
if not student:
|
|
# 404 表示请求的数据不存在
|
|
raise HTTPException(status_code=404, detail="学生不存在")
|
|
|
|
return {
|
|
"code": 200,
|
|
"detail": "查询成功",
|
|
"data": student,
|
|
}
|
|
|
|
@student_router.delete("/{id}", summary="逻辑删除学生")
|
|
def delete_student(id: int, db: Session = Depends(get_db)):
|
|
student = student_dao.delete_student(id, db)
|
|
|
|
# None 表示不存在，或者以前已经删除
|
|
if student is None:
|
|
raise HTTPException(status_code=404, detail="学生不存在或已经删除")
|
|
|
|
if student is False:
|
|
raise HTTPException(status_code=500, detail="删除学生失败")
|
|
|
|
return {
|
|
"code": 200,
|
|
"detail": "删除学生成功",
|
|
}
|