调整魔目录结构

applications/view/rights/view.py

@@ -1,6 +1,7 @@
-from flask import render_template, request, jsonify
+from flask import render_template, request, jsonify, make_response
+from flask_restful import Resource, reqparse
 
-from . import rights_bp
+from . import rights_bp, rights_api
 from ...common.admin import rights_curd
 from ...common.utils.http import success_api, fail_api
 from ...common.utils.rights import authorize
@@ -25,12 +26,14 @@ def data():
 @rights_bp.get('/add')
 @authorize("admin:power:add", log=True)
 def add():
+    """获取增加视图"""
     return render_template('admin/power/add.html')
 
 
 @rights_bp.get('/selectParent')
 @authorize("admin:power:main", log=True)
 def select_parent():
+    """获取选择父节点"""
     power_data = rights_curd.select_parent()
     res = {
         "status": {"code": 200, "message": "默认"},
@@ -49,36 +52,41 @@ def save():
     return success_api(msg="成功")
 
 
-# 权限编辑
-@rights_bp.get('/edit/<int:_id>')
-@authorize("admin:power:edit", log=True)
-def edit(_id):
-    power = rights_curd.get_power_by_id(_id)
-    icon = str(power.icon).split()
-    if len(icon) == 2:
-        icon = icon[1]
-    else:
-        icon = None
-    return render_template('admin/power/edit.html', power=power, icon=icon)
+@rights_api.resource('/power/<int:power_id>')
+class RightsURD(Resource):
+    @authorize("admin:power:edit", log=True)
+    def get(self, power_id):
+        power = rights_curd.get_power_by_id(power_id)
+        icon = str(power.icon).split()
+        if len(icon) == 2:
+            icon = icon[1]
+        else:
+            icon = None
+        return make_response(render_template('admin/power/edit.html', power=power, icon=icon))
 
+    @authorize("admin:power:remove", log=True)
+    def delete(self, power_id):
+        r = rights_curd.remove_power(power_id)
+        if r:
+            return success_api(msg="删除成功")
+        else:
+            return fail_api(msg="删除失败")
 
-# 权限更新
-@rights_bp.put('/update')
-@authorize("admin:power:edit", log=True)
-def update():
-    res = rights_curd.update_power(request.json)
-    if not res:
-        return fail_api(msg="更新权限失败")
-    return success_api(msg="更新权限成功")
+    @authorize("admin:power:edit", log=True)
+    def put(self, power_id):
+        res = rights_curd.update_power(request.json)
+        if not res:
+            return fail_api(msg="更新权限失败")
+        return success_api(msg="更新权限成功")
 
 
 # 启用权限
 @rights_bp.put('/enable')
 @authorize("admin:power:edit", log=True)
 def enable():
-    _id = request.json.get('powerId')
+    power_id = request.json.get('powerId')
     if id:
-        res = rights_curd.enable_status(_id)
+        res = rights_curd.enable_status(power_id)
         if not res:
             return fail_api(msg="出错啦")
         return success_api(msg="启用成功")
@@ -98,17 +106,6 @@ def dis_enable():
     return fail_api(msg="数据错误")
 
 
-# 权限删除
-@rights_bp.delete('/remove/<int:_id>')
-@authorize("admin:power:remove", log=True)
-def remove(_id):
-    r = rights_curd.remove_power(_id)
-    if r:
-        return success_api(msg="删除成功")
-    else:
-        return fail_api(msg="删除失败")
-
-
 # 批量删除
 @rights_bp.delete('/batchRemove')
 @authorize("admin:power:remove", log=True)