feature: 部门增删改查

pear_admin/apis/__init__.py

@@ -1,5 +1,6 @@
 from flask import Blueprint, Flask
 
+from .department import department_api
 from .passport import passport_api
 from .rights import rights_api
 from .role import role_api
@@ -11,5 +12,6 @@ def register_apis(app: Flask):
     apis.register_blueprint(passport_api)
     apis.register_blueprint(rights_api)
     apis.register_blueprint(role_api)
+    apis.register_blueprint(department_api)
 
     app.register_blueprint(apis)

pear_admin/apis/department.py

@@ -0,0 +1,68 @@
+from flask import Blueprint, request
+from flask_sqlalchemy.pagination import Pagination
+
+from pear_admin.extensions import db
+from pear_admin.orms import DepartmentORM
+
+department_api = Blueprint("department", __name__)
+
+
+@department_api.get("/department")
+def department_list():
+    page = request.args.get("page", default=1, type=int)
+    per_page = request.args.get("limit", default=10, type=int)
+
+    _type = request.args.get("type")
+    ret = []
+    if _type == "tree":
+        dept_obj: DepartmentORM = DepartmentORM.query.get(1)
+        for child in dept_obj.children:
+            child_data = child.json()
+            child_data["children"] = []
+            if child.children:
+                child_data["isParent"] = True
+            for son in child.children:
+                son_data = son.json()
+                # son_data["isParent"] = True
+                child_data["children"].append(son_data)
+            ret.append(child_data)
+        return {"code": 0, "message": "请求权限数据成功", "data": ret}
+    q = db.select(DepartmentORM)
+
+    pages: Pagination = db.paginate(q, page=page, per_page=per_page)
+
+    return {
+        "code": 0,
+        "message": "获取部门数据成功",
+        "data": [item.json() for item in pages.items],
+        "count": pages.total,
+    }
+
+
+@department_api.post("/department")
+def create_department():
+    data = request.get_json()
+    if data["id"]:
+        del data["id"]
+    department = DepartmentORM(**data)
+    department.save()
+    return {"code": 0, "message": "新增部门成功"}
+
+
+@department_api.put("/department/<int:rid>")
+def change_department(rid):
+    data = request.get_json()
+    del data["id"]
+
+    department_obj = DepartmentORM.query.get(rid)
+    for key, value in data.items():
+        setattr(department_obj, key, value)
+    department_obj.save()
+    return {"code": 0, "message": "修改部门成功"}
+
+
+@department_api.delete("/department/<int:rid>")
+def del_department(rid):
+    department_obj = DepartmentORM.query.get(rid)
+    department_obj.delete()
+    return {"code": 0, "message": "删除删除成功"}