first commit

code/system/apis/button.py

@@ -0,0 +1,79 @@
+# from application.ninja_cof import api
+# Author 臧成龙
+# coding=utf-8
+# @Time    : 2022/5/21 02:36
+# @File    : button.py
+# @Software: PyCharm
+# @qq: 939589097
+
+from typing import List
+
+from django.shortcuts import get_object_or_404
+from ninja import Field, ModelSchema, Query, Router, Schema
+from ninja.pagination import paginate
+from system.models import Button
+from utils.fu_ninja import MyPagination
+
+router = Router()
+
+
+class Filters(Schema):
+    name: str = Field(None, alias="name")
+    code: str = Field(None, alias="code")
+    id: str = Field(None, alias="id")
+
+
+class SchemaIn(ModelSchema):
+
+    class Config:
+        model = Button
+        model_exclude = ['id', 'create_datetime', 'update_datetime']
+
+
+class SchemaOut(ModelSchema):
+    class Config:
+        model = Button
+        model_fields = "__all__"
+    # model_fields = []
+
+
+@router.post("/button", response=SchemaOut)
+def create_button(request, data: SchemaIn):
+    post = Button.objects.create(**data.dict())
+    return post
+
+
+@router.get("/button/{button_id}", response=SchemaOut)
+def get_button(request, button_id: int):
+    post = get_object_or_404(Button, id=button_id)
+    return post
+
+
+@router.get("/button", response=List[SchemaOut])
+@paginate(MyPagination)
+def list_button(request, filters: Filters = Query(...)):
+    qs = Button.objects.filter(**filters.dict(exclude_none=True))
+    return qs
+
+
+@router.get("/button/all/list", response=List[SchemaOut])
+def all_list_button(request):
+    qs = Button.objects.all()
+    return qs
+
+
+@router.put("/button/{button_id}", response=SchemaOut)
+def update_button(request, button_id: int, payload: SchemaIn):
+    post = get_object_or_404(Button, id=button_id)
+    for attr, value in payload.dict().items():
+        setattr(post, attr, value)
+    post.save()
+    return post
+
+
+@router.delete("/button/{button_id}")
+def delete_button(request, button_id: int):
+    post = get_object_or_404(Button, id=button_id)
+    post.delete()
+    return {"success": True}
+