支持隐藏不同角色下的菜单

app/models/Resource.py

@@ -31,6 +31,8 @@ class Resource(db.Model, UserMixin):
 
     STATUS = db.Column(db.String(10))
 
+    HIDDEN = False
+
     def get_id(self):
         return str(self.ID)
 
@@ -66,9 +68,9 @@ class Resource(db.Model, UserMixin):
 
     def to_router_json(self):
         router = {
-            'name': self.PATH.capitalize(),
+            'name': self.PATH.capitalize() if self.PATH else '',
             'path': self.PATH,
-            'hidden': False,
+            'hidden': self.HIDDEN,
             'redirect': 'noRedirect',
             'component': self.URL,
             'alwaysShow': True,

app/routes/index.py

@@ -4,13 +4,36 @@ from flask_login import login_user, logout_user, login_required, \
     current_user
 from flask import g, jsonify
 from ..models import Resource, Organization, ResourceType
+from sqlalchemy import text
+
+def setAccessable(resource, permitedIDList):
+    if resource.ID not in permitedIDList:
+        resource.HIDDEN = True
+    
+    for child in resource.children:
+        setAccessable(child, permitedIDList)
 
 @base.route('/getRouters')
 @login_required
 def getRouters():
-    resources = Resource.query.join(ResourceType, Resource.type).filter(Resource.SYRESOURCE_ID == None).all()
+    resources = []
+    resources += [res for org in current_user.organizations for res in org.resources if org.resources]
+    resources += [res for role in current_user.roles for res in role.resources if role.resources]
+    
+    ids = [res.ID for res in resources]
+
+    #allResources = Resource.query.filter(
+    #    text("FIND_IN_SET(ID, :ids) > 0")).params(ids = ','.join(ids)) .all()
     
-    return jsonify({'msg': '操作成功', 'code': 200, "data": [res.to_router_json() for res in resources]})    
+    allResources = Resource.query.join(ResourceType, Resource.type).filter(
+        Resource.SYRESOURCE_ID == None).all()
+    # filter resource not allowed
+    for res in allResources:
+       setAccessable(res, ids)
+
+    json = [res.to_router_json() for res in allResources]
+
+    return jsonify({'msg': '操作成功', 'code': 200, "data": json})    
 
 
     # return jsonify({'msg': '操作成功', 'code': 200, "data":[{"name":"System","path":"/system","hidden":False,"redirect":"noRedirect","component":"Layout","alwaysShow":True,"meta":{"title":"系统管理","icon":"system","noCache":False,"link":''},"children":[{"name":"User","path":"user","hidden":False,"component":"system/user/index","meta":{"title":"用户管理","icon":"user","noCache":False,"link":''}},{"name":"Role","path":"role","hidden":False,"component":"system/role/index","meta":{"title":"角色管理","icon":"peoples","noCache":False,"link":''}},{"name":"Menu","path":"menu","hidden":False,"component":"system/menu/index","meta":{"title":"菜单管理","icon":"tree-table","noCache":False,"link":''}},{"name":"Dept","path":"dept","hidden":False,"component":"system/dept/index","meta":{"title":"部门管理","icon":"tree","noCache":False,"link":''}},{"name":"Log","path":"log","hidden":False,"redirect":"noRedirect","component":"ParentView","alwaysShow":True,"meta":{"title":"日志管理","icon":"log","noCache":False,"link":''},"children":[{"name":"Operlog","path":"operlog","hidden":False,"component":"monitor/operlog/index","meta":{"title":"操作日志","icon":"form","noCache":False,"link":''}},{"name":"Logininfor","path":"logininfor","hidden":False,"component":"monitor/logininfor/index","meta":{"title":"登录日志","icon":"logininfor","noCache":False,"link":''}}]}]}]})