async def delete_post(post_id: int, db: Session = Depends(deps.get_db)):
"""文章删除"""
post_obj = db.query(models.Post).get(post_id)
if post_obj:
db.delete(post_obj)
db.commit()
return Response200()
return Response404(msg="删除失败,文章不存在")
async def delete_review(comment_id: int, db: Session = Depends(deps.get_db)):
"""删除评论"""
review_obj = db.query(models.Comment).get(comment_id)
if review_obj:
db.delete(review_obj)
db.commit()
return Response200(data=review_obj)
return Response404(msg="删除失败,评论不存在")
async def delete_sort(sort_id: int, db: Session = Depends(deps.get_db)):
"""分类删除"""
category_obj = db.query(models.Category).get(sort_id)
if category_obj:
db.delete(category_obj)
db.commit()
return Response200(data=category_obj)
return Response404(msg="删除失败,分类不存在")
async def admin_login(
auth: Auth,
db: Session = Depends(deps.get_db)
):
user = db.query(Admin).filter(Admin.username == auth.username).first()
if not user or not security.verify_password(auth.password, user.password_hash):
return Response404(msg="用户名或密码错误")
token = Token(token=f"Bearer {create_access_token(user.id)}")
return Response200(data=token)
async def create_review(review: ReviewCreate,
db: Session = Depends(deps.get_db)):
if db.query(models.Post).get(review.post_id):
review_obj = models.Comment(**review.dict())
db.add(review_obj)
db.commit()
db.refresh(review_obj)
return Response200(data={"review": review_obj})
return Response404(msg="评论失败，文章不存在")
async def create_post(post: PostCreate, db: Session = Depends(deps.get_db)):
"""文章创建"""
if db.query(models.Category).get(post.category_id):
post_obj = models.Post(**post.dict())
db.add(post_obj)
db.commit()
db.refresh(post_obj)
return Response200(data={"post": post_obj})
else:
return Response404(msg="新增失败,分类不存在")
async def update_sort(category_id: int,
category: CategoryCreate,
db: Session = Depends(deps.get_db)):
"""分类更新"""
category_obj = db.query(models.Category).filter_by(id=category_id)
if category_obj:
category_obj.update({"name": category.name})
db.commit()
return Response200(data={"category": category_obj.first()})
return Response404(msg="修改失败,分类不存在")
async def create_sort(category: CategoryCreate,
db: Session = Depends(deps.get_db)):
"""分类创建"""
category_result = db.query(
models.Category).filter_by(name=category.name).first()
if category_result:
return Response404(msg="新增失败,分类已存在")
category_obj = models.Category(name=category.name)
db.add(category_obj)
db.commit()
db.refresh(category_obj)
return Response200(data={"category": category_obj})
async def update_post(post_id: int,
post: PostUpdate,
db: Session = Depends(deps.get_db)):
"""文章更新"""
post_obj = db.query(models.Post).filter_by(id=post_id)
if post_obj:
post_obj.update({
"title": post.title,
"body": post.body,
"status": post.status,
"category_id": post.category_id
})
db.commit()
return Response200(data={"post": post_obj.first()})
return Response404(msg="更新失败,文章不存在")