58 行
2.2 KiB
Python
58 行
2.2 KiB
Python
from __future__ import annotations
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
from sqlalchemy import func as sqlfunc
|
|
|
|
from app.database import get_db
|
|
from app.auth import get_current_user
|
|
from app.models.qualification import Qualification
|
|
from app.schemas.qualification import QualificationSchema, QualificationResponse
|
|
|
|
router = APIRouter(prefix="/api/qualifications", tags=["资质荣誉"], dependencies=[Depends(get_current_user)])
|
|
|
|
|
|
@router.get("/", response_model=list[QualificationResponse])
|
|
def list_qualifications(db: Session = Depends(get_db)):
|
|
items = db.query(Qualification).order_by(Qualification.sort_order).all()
|
|
return [QualificationResponse.model_validate(item) for item in items]
|
|
|
|
|
|
@router.post("/", response_model=QualificationResponse, status_code=201)
|
|
def create_qualification(req: QualificationSchema, db: Session = Depends(get_db)):
|
|
max_order = db.query(sqlfunc.max(Qualification.sort_order)).scalar() or 0
|
|
item = Qualification(**req.model_dump(), sort_order=max_order + 1)
|
|
db.add(item)
|
|
db.commit()
|
|
db.refresh(item)
|
|
return QualificationResponse.model_validate(item)
|
|
|
|
|
|
@router.put("/{item_id}", response_model=QualificationResponse)
|
|
def update_qualification(item_id: int, req: QualificationSchema, db: Session = Depends(get_db)):
|
|
item = db.query(Qualification).get(item_id)
|
|
if not item:
|
|
raise HTTPException(status_code=404, detail="资质信息不存在")
|
|
for key, val in req.model_dump().items():
|
|
setattr(item, key, val)
|
|
db.commit()
|
|
db.refresh(item)
|
|
return QualificationResponse.model_validate(item)
|
|
|
|
|
|
@router.delete("/{item_id}")
|
|
def delete_qualification(item_id: int, db: Session = Depends(get_db)):
|
|
item = db.query(Qualification).get(item_id)
|
|
if not item:
|
|
raise HTTPException(status_code=404, detail="资质信息不存在")
|
|
db.delete(item)
|
|
db.commit()
|
|
return {"message": "删除成功"}
|
|
|
|
|
|
@router.put("/reorder/batch")
|
|
def reorder_qualifications(ids: list[int], db: Session = Depends(get_db)):
|
|
for i, qid in enumerate(ids):
|
|
db.query(Qualification).filter(Qualification.id == qid).update({"sort_order": i})
|
|
db.commit()
|
|
return {"message": "排序更新成功"}
|