refactor(重构程序&目录结构)

applications/api/rights/department.py

@@ -1,20 +1,19 @@
 from flask import jsonify
 from flask_restful import Resource, reqparse
 
-from applications.common.utils.http import success_api, fail_api
-from applications.extensions import db
-from applications.models import CompanyDepartment, CompanyUser
+from common.utils.http import success_api, fail_api
+from extensions import db
+from models import DepartmentModels, UserModels
 
 
 class DepartmentsResource(Resource):
 
     def get(self):
-        dept_data = CompanyDepartment.query.order_by(CompanyDepartment.sort).all()
+        dept_data = DepartmentModels.query.order_by(DepartmentModels.sort).all()
         # TODO dtree 需要返回状态信息
         res = {
             "status": {"code": 200, "message": "默认"},
             "data": [
-
                 {
                     'deptId': item.id,
                     'parentId': item.parent_id,
@@ -45,7 +44,7 @@ class DepartmentsResource(Resource):
 
         res = parser.parse_args()
 
-        dept = CompanyDepartment(
+        dept = DepartmentModels(
             parent_id=res.parent_id,
             dept_name=res.dept_name,
             sort=res.sort,
@@ -63,7 +62,7 @@ class DepartmentsResource(Resource):
 
 class DepartmentResource(Resource):
     def get(self, dept_id):
-        dept = CompanyDepartment.query.filter_by(id=dept_id).first()
+        dept = DepartmentModels.query.filter_by(id=dept_id).first()
         dept_data = {
             'id': dept.id,
             'dept_name': dept.dept_name,
@@ -96,15 +95,15 @@ class DepartmentResource(Resource):
             "status": res.status,
             "address": res.address
         }
-        res = CompanyDepartment.query.filter_by(id=dept_id).update(data)
+        res = DepartmentModels.query.filter_by(id=dept_id).update(data)
         if not res:
             return fail_api(message="更新失败")
         db.session.commit()
         return success_api(message="更新成功")
 
     def delete(self, dept_id):
-        ret = CompanyDepartment.query.filter_by(id=dept_id).delete()
-        CompanyUser.query.filter_by(dept_id=dept_id).update({"dept_id": None})
+        ret = DepartmentModels.query.filter_by(id=dept_id).delete()
+        UserModels.query.filter_by(dept_id=dept_id).update({"dept_id": None})
         db.session.commit()
         if ret:
             return success_api(message="删除成功")
@@ -113,7 +112,7 @@ class DepartmentResource(Resource):
 
 class DeptEnableResource(Resource):
     def put(self, dept_id):
-        d = CompanyDepartment.query.get(dept_id)
+        d = DepartmentModels.query.get(dept_id)
         if d:
             d.status = not d.status
             db.session.commit()