84 lines
3.2 KiB
Python
84 lines
3.2 KiB
Python
|
from fastapi import APIRouter, Depends, HTTPException
|
||
|
|
from sqlalchemy.orm import Session
|
||
|
|
from typing import List, Optional
|
||
|
|
|
||
|
|
from database import get_db
|
||
|
|
from models import Student, Class, Teacher
|
||
|
|
from schemas import StudentCreate, StudentUpdate, StudentBase
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/api/students", tags=["students"])
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/", response_model=List[StudentBase])
|
||
|
|
def list_students(
|
||
|
|
class_id: Optional[int] = None,
|
||
|
|
name: Optional[str] = None,
|
||
|
|
db: Session = Depends(get_db),
|
||
|
|
skip: int = 0,
|
||
|
|
limit: int = 100,
|
||
|
|
):
|
||
|
|
query = db.query(Student).filter(Student.is_deleted == False)
|
||
|
|
if class_id is not None:
|
||
|
|
query = query.filter(Student.class_id == class_id)
|
||
|
|
if name:
|
||
|
|
query = query.filter(Student.name.contains(name))
|
||
|
|
students = query.offset(skip).limit(limit).all()
|
||
|
|
return students
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/{student_id}", response_model=StudentBase)
|
||
|
|
def get_student(student_id: int, db: Session = Depends(get_db)):
|
||
|
|
student = db.query(Student).filter(Student.id == student_id, Student.is_deleted == False).first()
|
||
|
|
if not student:
|
||
|
|
raise HTTPException(status_code=404, detail="学生不存在")
|
||
|
|
return student
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/", response_model=StudentBase)
|
||
|
|
def create_student(data: StudentCreate, db: Session = Depends(get_db)):
|
||
|
|
existing = db.query(Student).filter(Student.num == data.num, Student.is_deleted == False).first()
|
||
|
|
if existing:
|
||
|
|
raise HTTPException(status_code=400, detail="学生编号已存在")
|
||
|
|
|
||
|
|
if not db.query(Class).filter(Class.id == data.class_id, Class.is_deleted == False).first():
|
||
|
|
raise HTTPException(status_code=400, detail="班级不存在")
|
||
|
|
if data.advisor_id:
|
||
|
|
if not db.query(Teacher).filter(Teacher.id == data.advisor_id, Teacher.is_deleted == False).first():
|
||
|
|
raise HTTPException(status_code=400, detail="顾问教师不存在")
|
||
|
|
|
||
|
|
student = Student(**data.model_dump())
|
||
|
|
db.add(student)
|
||
|
|
db.commit()
|
||
|
|
db.refresh(student)
|
||
|
|
return student
|
||
|
|
|
||
|
|
|
||
|
|
@router.put("/{student_id}", response_model=StudentBase)
|
||
|
|
def update_student(student_id: int, data: StudentUpdate, db: Session = Depends(get_db)):
|
||
|
|
student = db.query(Student).filter(Student.id == student_id, Student.is_deleted == False).first()
|
||
|
|
if not student:
|
||
|
|
raise HTTPException(status_code=404, detail="学生不存在")
|
||
|
|
|
||
|
|
if data.class_id is not None:
|
||
|
|
if not db.query(Class).filter(Class.id == data.class_id, Class.is_deleted == False).first():
|
||
|
|
raise HTTPException(status_code=400, detail="班级不存在")
|
||
|
|
if data.advisor_id is not None:
|
||
|
|
if not db.query(Teacher).filter(Teacher.id == data.advisor_id, Teacher.is_deleted == False).first():
|
||
|
|
raise HTTPException(status_code=400, detail="顾问教师不存在")
|
||
|
|
|
||
|
|
for key, value in data.model_dump(exclude_unset=True).items():
|
||
|
|
setattr(student, key, value)
|
||
|
|
db.commit()
|
||
|
|
db.refresh(student)
|
||
|
|
return student
|
||
|
|
|
||
|
|
|
||
|
|
@router.delete("/{student_id}")
|
||
|
|
def delete_student(student_id: int, db: Session = Depends(get_db)):
|
||
|
|
student = db.query(Student).filter(Student.id == student_id, Student.is_deleted == False).first()
|
||
|
|
if not student:
|
||
|
|
raise HTTPException(status_code=404, detail="学生不存在")
|
||
|
|
student.is_deleted = True
|
||
|
|
db.commit()
|
||
|
|
return {"message": "删除成功"}
|