first commit

code/system/apis/dept.py

@@ -0,0 +1,79 @@
+# from application.ninja_cof import api
+# Author 臧成龙
+# coding=utf-8
+# @Time    : 2022/5/15 21:47
+# @File    : dept.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 Dept
+from utils.fu_crud import create, delete, retrieve, update
+from utils.fu_ninja import FuFilters, MyPagination
+from utils.fu_response import FuResponse
+from utils.list_to_tree import list_to_route, list_to_tree
+
+router = Router()
+
+
+class Filters(FuFilters):
+    name: str = Field(None, alias="name")
+    status: bool = Field(None, alias="status")
+    id: str = Field(None, alias="id")
+
+
+class SchemaIn(ModelSchema):
+    parent_id: int = None
+
+    class Config:
+        model = Dept
+        model_exclude = ['id', 'parent', 'create_datetime', 'update_datetime']
+
+
+class SchemaOut(ModelSchema):
+    class Config:
+        model = Dept
+        model_fields = "__all__"
+    # model_fields = []
+
+
+@router.post("/dept", response=SchemaOut)
+def create_dept(request, data: SchemaIn):
+    dept = create(request, data, Dept)
+    return dept
+
+
+@router.delete("/dept/{dept_id}")
+def delete_dept(request, dept_id: int):
+    delete(dept_id, Dept)
+    return {"success": True}
+
+
+@router.put("/dept/{dept_id}", response=SchemaOut)
+def update_dept(request, dept_id: int, data: SchemaIn):
+    dept = update(request, dept_id, data, Dept)
+    return dept
+
+
+@router.get("/dept", response=List[SchemaOut])
+@paginate(MyPagination)
+def list_dept(request, filters: Filters = Query(...)):
+    qs = retrieve(request, Dept, filters)
+    return qs
+
+
+@router.get("/dept/{dept_id}", response=SchemaOut)
+def get_dept(request, dept_id: int):
+    dept = get_object_or_404(Dept, id=dept_id)
+    return dept
+
+
+@router.get("/dept/list/tree")
+def list_dept_tree(request, filters: Filters = Query(...)):
+    qs = retrieve(request, Dept, filters).values()
+    dept_tree = list_to_tree(list(qs))
+    return FuResponse(data=dept_tree)