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):
+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, "有序号相同的教程存在")
+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": "删除成功"}

tutorial/migrations/0002_alter_tutorial_display.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1.6 on 2025-03-03 14:40
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('tutorial', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='tutorial',
+            name='display',
+            field=models.IntegerField(unique=True),
+        ),
+    ]

tutorial/models.py

@@ -3,7 +3,7 @@ from django_extensions.db.models import TimeStampedModel
 
 
 class Tutorial(TimeStampedModel):
-    display = models.CharField(max_length=10, unique=True)
+    display = models.IntegerField(unique=True)
     title = models.CharField(max_length=100)
     content = models.TextField()
     is_public = models.BooleanField(default=False)

tutorial/schemas.py

@@ -4,7 +4,7 @@ from .models import Tutorial
 
 
 class TutorialSlim(Schema):
-    display: str
+    display: int
     title: str
     is_public: bool
 
@@ -16,13 +16,12 @@ class TutorialAll(ModelSchema):
 
 
 class TutorialReturn(Schema):
-    total: int
     list: List[TutorialSlim]
     first: Optional[TutorialAll]
 
 
 class TutorialIn(Schema):
-    display: str
+    display: int
     title: str
     content: str
     is_public: bool = False