refactor(移除生产部署文件)

applications/rights/roles.py

@@ -0,0 +1,189 @@
+from flask_restful import Resource, reqparse, marshal
+
+from common.utils.http import table_api, success_api, fail_api
+from extensions import db
+from models import RightModels, RoleModels, UserModels
+
+
+def remove_role(role_id):
+    """ 删除角色 """
+    role = RoleModels.query.filter_by(id=role_id).first()
+    # 删除该角色的权限
+    power_id_list = []
+    for p in role.power:
+        power_id_list.append(p.id)
+
+    powers = RightModels.query.filter(RightModels.id.in_(power_id_list)).all()
+    for p in powers:
+        role.power.remove(p)
+    user_id_list = []
+    for u in role.user:
+        user_id_list.append(u.id)
+    users = UserModels.query.filter(UserModels.id.in_(user_id_list)).all()
+    for u in users:
+        role.user.remove(u)
+    r = RoleModels.query.filter_by(id=role_id).delete()
+    db.session.commit()
+    return r
+
+
+def batch_remove_role(role_ids):
+    """ 批量删除 """
+    for role_id in role_ids:
+        remove_role(role_id)
+
+
+class RoleRolesResource(Resource):
+    # 表格数据
+    def get(self):
+        parser = reqparse.RequestParser(bundle_errors=True)
+        parser.add_argument('page', type=int, default=1)
+        parser.add_argument('limit', type=int, default=10)
+        parser.add_argument('roleName', type=str, dest='role_name', default="")
+        parser.add_argument('roleCode', type=str, dest='role_code', default="")
+
+        res = parser.parse_args()
+
+        filters = []
+        if res.role_name:
+            filters.append(RoleModels.name.like('%' + res.role_name + '%'))
+        if res.role_code:
+            filters.append(RoleModels.code.like('%' + res.role_code + '%'))
+
+        paginate = RoleModels.query.filter(*filters).paginate(page=res.page, per_page=res.limit, error_out=False)
+
+        return table_api(result={'items': [{'id': item.id,
+                                            'roleName': item.name,
+                                            'roleCode': item.code,
+                                            'enable': item.enable,
+                                            'comment': item.comment,
+                                            'details': item.details,
+                                            'sort': item.sort,
+                                             } for item in paginate.items],
+                                 'total': paginate.total}
+                         , code=0)
+
+    def delete(self):
+        parser = reqparse.RequestParser()
+        parser.add_argument('ids[]', action='append', dest='ids')
+
+        res = parser.parse_args()
+
+        batch_remove_role(res.ids)
+        return success_api(message="批量删除成功")
+
+
+class RoleRoleResource(Resource):
+
+    def post(self, role_id):
+        parser = reqparse.RequestParser()
+        parser.add_argument('details', type=str)
+        parser.add_argument('enable', type=int)
+        parser.add_argument('roleCode', type=str, dest='role_code')
+        parser.add_argument('roleName', type=str, dest='role_name')
+        parser.add_argument('sort', type=int)
+
+        res = parser.parse_args()
+
+        role = RoleModels(
+            details=res.details,
+            enable=res.enable,
+            code=res.role_code,
+            name=res.role_name,
+            sort=res.sort
+        )
+        db.session.add(role)
+        db.session.commit()
+        return success_api(message="成功")
+
+    # 更新角色
+    def put(self, role_id):
+        parser = reqparse.RequestParser()
+        parser.add_argument('roleId', dest='role_id', type=int)
+        parser.add_argument('roleCode', dest='role_code', type=str)
+        parser.add_argument('roleName', dest='role_name', type=str)
+        parser.add_argument('sort', type=int)
+        parser.add_argument('enable', type=int)
+        parser.add_argument('details', type=str)
+
+        res = parser.parse_args()
+
+        data = {
+            "code": res.role_code,
+            "name": res.role_name,
+            "sort": res.sort,
+            "enable": res.enable,
+            "details": res.details
+        }
+
+        role = RoleModels.query.filter_by(id=role_id).update(data)
+        db.session.commit()
+        if not role:
+            return fail_api(message="更新角色失败")
+        return success_api(message="更新角色成功")
+
+
+class RoleEnableResource(Resource):
+    """启用用户"""
+
+    def put(self, role_id):
+        ret = RoleModels.query.get(role_id)
+        ret.enable = not ret.enable
+        db.session.commit()
+
+        message = "修改成功"
+        if not ret:
+            return fail_api(message="出错啦")
+        return success_api(message=message)
+
+
+class RolePowerResource(Resource):
+
+    def get(self, role_id):
+        # 获取角色权限
+        role = RoleModels.query.filter_by(id=role_id).first()
+        # 获取权限列表的 id
+        check_powers_list = [rp.id for rp in role.power]
+        powers = RightModels.query.all()  # 获取所有的权限
+        powers = marshal(powers, RightModels.fields())
+        for i in powers:
+            if int(i.get("powerId")) in check_powers_list:
+                i["checkArr"] = "1"
+            else:
+                i["checkArr"] = "0"
+        return {
+            "data": powers,
+            "status": {"code": 200, "message": "默认"}
+        }
+
+    # 保存角色权限
+    def put(self, role_id):
+        parser = reqparse.RequestParser()
+        parser.add_argument('powerIds', dest='power_ids')
+        parser.add_argument('roleId', dest='role_id')
+
+        res = parser.parse_args()
+        power_list = res.power_ids.split(',')
+
+        """ 更新角色权限 """
+        role = RoleModels.query.filter_by(id=role_id).first()
+        power_id_list = []
+        for p in role.power:
+            power_id_list.append(p.id)
+        powers = RightModels.query.filter(RightModels.id.in_(power_id_list)).all()
+        for p in powers:
+            role.power.remove(p)
+        powers = RightModels.query.filter(RightModels.id.in_(power_list)).all()
+        for p in powers:
+            role.power.append(p)
+        db.session.commit()
+        return success_api(message="授权成功")
+
+    # 角色删除
+    def delete(self, role_id):
+        print(role_id)
+        res = remove_role(role_id)
+        print(res)
+        if not res:
+            return fail_api(message="角色删除失败")
+        return success_api(message="角色删除成功")