调整目录结构

applications/common/admin/dept_curd.py

@@ -0,0 +1,87 @@
+from applications.common.utils.validate import xss_escape
+from applications.extensions import db
+from applications.models import Dept, DeptSchema
+from applications.models import User
+from applications.common.curd import model_to_dicts
+
+
+def get_dept_dict():
+    dept = Dept.query.order_by(Dept.sort).all()
+    res = model_to_dicts(Schema=DeptSchema, model=dept)
+    return res
+
+
+def save_dept(req):
+    address = xss_escape(req.get("address"))
+    deptName = xss_escape(req.get("deptName"))
+    email = xss_escape(req.get("email"))
+    leader = xss_escape(req.get("leader"))
+    parentId = xss_escape(req.get("parentId"))
+    phone = xss_escape(req.get("phone"))
+    sort = xss_escape(req.get("sort"))
+    status = xss_escape(req.get("status"))
+    dept = Dept(
+        parent_id=parentId,
+        dept_name=deptName,
+        sort=sort,
+        leader=leader,
+        phone=phone,
+        email=email,
+        status=status,
+        address=address
+    )
+    r = db.session.add(dept)
+    db.session.commit()
+    return r
+
+
+def get_dept_by_id(id):
+    d = Dept.query.filter_by(id=id).first()
+    return d
+
+
+# 启动权限
+def enable_status(id):
+    enable = 1
+    d = Dept.query.filter_by(id=id).update({"status": enable})
+    if d:
+        db.session.commit()
+        return True
+    return False
+
+
+# 停用权限
+def disable_status(id):
+    enable = 0
+    d = Dept.query.filter_by(id=id).update({"status": enable})
+    if d:
+        db.session.commit()
+        return True
+    return False
+
+
+def update_dept(json):
+    id = json.get("deptId"),
+    data = {
+        "dept_name": xss_escape(json.get("deptName")),
+        "sort": xss_escape(json.get("sort")),
+        "leader": xss_escape(json.get("leader")),
+        "phone": xss_escape(json.get("phone")),
+        "email": xss_escape(json.get("email")),
+        "status": xss_escape(json.get("status")),
+        "address": xss_escape(json.get("address"))
+    }
+    d = Dept.query.filter_by(id=id).update(data)
+    if not d:
+        return False
+    db.session.commit()
+    return True
+
+
+def remove_dept(id):
+    d = Dept.query.filter_by(id=id).delete()
+    if not d:
+        return False
+    User.query.filter_by(dept_id=id).update({"dept_id": None})
+    db.session.commit()
+    return True