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": "删除删除成功"}

pear_admin/orms/department.py

@@ -8,17 +8,15 @@ class DepartmentORM(BaseORM):
     id = db.Column(db.Integer, primary_key=True, comment="部门ID")
     name = db.Column(db.String(50), comment="部门名称")
     leader = db.Column(db.String(50), comment="负责人")
-    phone = db.Column(db.String(20), comment="联系方式")
-    email = db.Column(db.String(50), comment="邮箱")
-    enable = db.Column(db.Boolean, comment="状态(1开启,0关闭)")
-    comment = db.Column(db.Text, comment="备注")
-    address = db.Column(db.String(255), comment="详细地址")
-    sort = db.Column(db.Integer, comment="排序")
+    enable = db.Column(db.Boolean, comment="状态(1开启,0关闭)", default=True)
 
     users = db.relationship("UserORM", backref="department")
 
-    pid = db.Column(db.Integer, db.ForeignKey("ums_department.id"))
-    parent = db.relationship("DepartmentORM", remote_side=[id], backref="child")  # 自关联
+    pid = db.Column(db.Integer, db.ForeignKey("ums_department.id"), default=1)
+    parent = db.relationship(
+        "DepartmentORM", back_populates="children", remote_side=[id]
+    )  # 自关联
+    children = db.relationship("DepartmentORM", back_populates="parent")
 
     def json(self):
         return {
@@ -26,8 +24,5 @@ class DepartmentORM(BaseORM):
             "pid": self.pid,
             "name": self.name,
             "leader": self.leader,
-            "email": self.email,
-            "phone": self.phone,
-            "sort": self.sort,
             "enable": self.enable,
         }

pear_admin/views/views.py

@@ -11,3 +11,8 @@ def rights_view():
 @views_bp.get("/role")
 def role_view():
     return render_template("views/role/index.html")
+
+
+@views_bp.get("/department")
+def department_view():
+    return render_template("views/department/index.html")