update

tutorial/api.py

@@ -1,9 +1,7 @@
+from typing import List
 from ninja import Router
 from ninja.errors import HttpError
-
 from account.decorators import super_required
-
-# from account.decorators import super_required
 from .schemas import TutorialAll, TutorialIn, TutorialReturn
 from .models import Tutorial
 
@@ -14,28 +12,41 @@ router = Router()
 @super_required
 def tutorial(request):
     return {
-        "total": Tutorial.objects.count(),
         "list": Tutorial.objects.all(),
         "first": Tutorial.objects.first(),
     }
 
 
+@router.get("/display", response=List[int])
+def get_all_public_display(request):
+    return Tutorial.objects.filter(is_public=True).values_list("display", flat=True)
+
+
 @router.get("/{display}", response=TutorialAll)
-def get(request, display: str):
-    return Tutorial.objects.get(display=display)
+def get(request, display: int):
+    try:
+        return Tutorial.objects.get(display=display)
+    except Tutorial.DoesNotExist:
+        raise HttpError(404, "此序号无教程")
 
 
 @router.post("/")
 @super_required
-def create(request, payload: TutorialIn):
-    if Tutorial.objects.filter(display=payload.display):
-        raise HttpError(400, "有序号相同的教程存在")
-    Tutorial.objects.create(**payload.dict())
-    return {"message": "创建成功"}
+def create_or_update(request, payload: TutorialIn):
+    try:
+        item = Tutorial.objects.get(display=payload.display)
+        item.title = payload.title
+        item.content = payload.content
+        item.is_public = payload.is_public
+        item.save()
+        return {"message": "更新成功"}
+    except Tutorial.DoesNotExist:
+        Tutorial.objects.create(**payload.dict())
+        return {"message": "创建成功"}
 
 
 @router.delete("/{display}")
 @super_required
-def remove(request, display: str):
+def remove(request, display: int):
     Tutorial.objects.filter(display=display).delete()
     return {"message": "删除成功"}