Ubuntu
114 lines
3.3 KiB
Python
114 lines
3.3 KiB
Python
"""
|
|
分类管理路由
|
|
"""
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.orm import Session
|
|
from typing import List
|
|
import sys
|
|
import os
|
|
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from database import get_db
|
|
from models import Category, Post
|
|
from schemas import CategoryCreate, CategoryResponse
|
|
from routers.auth import get_current_admin
|
|
|
|
router = APIRouter(prefix="/api/categories", tags=["分类"])
|
|
|
|
|
|
@router.get("", response_model=List[CategoryResponse])
|
|
def get_categories(db: Session = Depends(get_db)):
|
|
"""获取所有分类"""
|
|
categories = db.query(Category).all()
|
|
# 添加文章数量
|
|
result = []
|
|
for cat in categories:
|
|
post_count = db.query(Post).filter(Post.category_id == cat.id).count()
|
|
cat_dict = {
|
|
"id": cat.id,
|
|
"name": cat.name,
|
|
"description": cat.description,
|
|
"created_at": cat.created_at,
|
|
"post_count": post_count
|
|
}
|
|
result.append(cat_dict)
|
|
return result
|
|
|
|
|
|
@router.post("", response_model=CategoryResponse)
|
|
def create_category(
|
|
category_data: CategoryCreate,
|
|
current_admin=Depends(get_current_admin),
|
|
db: Session = Depends(get_db)
|
|
):
|
|
"""创建分类"""
|
|
# 检查数量限制
|
|
count = db.query(Category).count()
|
|
if count >= 25:
|
|
raise HTTPException(status_code=400, detail="分类数量已达上限(25个)")
|
|
|
|
# 检查名称重复
|
|
existing = db.query(Category).filter(Category.name == category_data.name).first()
|
|
if existing:
|
|
raise HTTPException(status_code=400, detail="分类名称已存在")
|
|
|
|
category = Category(**category_data.dict())
|
|
db.add(category)
|
|
db.commit()
|
|
db.refresh(category)
|
|
|
|
return {
|
|
"id": category.id,
|
|
"name": category.name,
|
|
"description": category.description,
|
|
"created_at": category.created_at,
|
|
"post_count": 0
|
|
}
|
|
|
|
|
|
@router.put("/{category_id}", response_model=CategoryResponse)
|
|
def update_category(
|
|
category_id: int,
|
|
category_data: CategoryCreate,
|
|
current_admin=Depends(get_current_admin),
|
|
db: Session = Depends(get_db)
|
|
):
|
|
"""更新分类"""
|
|
category = db.query(Category).filter(Category.id == category_id).first()
|
|
if not category:
|
|
raise HTTPException(status_code=404, detail="分类不存在")
|
|
|
|
category.name = category_data.name
|
|
category.description = category_data.description
|
|
db.commit()
|
|
db.refresh(category)
|
|
|
|
post_count = db.query(Post).filter(Post.category_id == category.id).count()
|
|
return {
|
|
"id": category.id,
|
|
"name": category.name,
|
|
"description": category.description,
|
|
"created_at": category.created_at,
|
|
"post_count": post_count
|
|
}
|
|
|
|
|
|
@router.delete("/{category_id}")
|
|
def delete_category(
|
|
category_id: int,
|
|
current_admin=Depends(get_current_admin),
|
|
db: Session = Depends(get_db)
|
|
):
|
|
"""删除分类"""
|
|
category = db.query(Category).filter(Category.id == category_id).first()
|
|
if not category:
|
|
raise HTTPException(status_code=404, detail="分类不存在")
|
|
|
|
# 将该分类下的文章设为未分类
|
|
db.query(Post).filter(Post.category_id == category_id).update({"category_id": None})
|
|
|
|
db.delete(category)
|
|
db.commit()
|
|
return {"message": "分类已删除"}
|