fix when group has no child

Author: wuyue92tree

adminlteui/core.py

@@ -49,6 +49,7 @@ def make(self, request, models=None, deep=1, deep_limit=0):
             menu_item['name'] = menu_item['label']
 
         menu_item['target_blank'] = self.target_blank
+        menu_item['menu_type'] = self.menu_type or 'group'
 
         if self.child:
             if deep_limit == 0 or deep <= deep_limit:
@@ -57,6 +58,10 @@ def make(self, request, models=None, deep=1, deep_limit=0):
                     deep += 1
                     child_menu = child.make(request, models, deep, deep_limit)
                     if child_menu:
+                        # menu_type: group and child is empty will hide the menu
+                        if child_menu.get('menu_type', 'group') == 'group':
+                            if len(child_menu.get('child')) == 0:
+                                continue
                         child_list.append(child_menu)
                 menu_item['child'] = child_list
             else:
@@ -109,6 +114,10 @@ def build_main_menu(self, request, app_list):
         for menu_item in self.main_menu:
             menu_item_ = menu_item.make(request, models)
             if menu_item_:
+                # menu_type: group and child is empty will hide the menu
+                if menu_item_.get('menu_type', 'group') == 'group':
+                    if len(menu_item_.get('child')) == 0:
+                        continue
                 menu.append(menu_item_)
         return menu
 
@@ -127,5 +136,9 @@ def build_top_menu(self, request, app_list):
         for menu_item in self.top_menu:
             menu_item_ = menu_item.make(request, models, deep_limit=2)
             if menu_item_:
+                # menu_type: group and child is empty will hide the menu
+                if menu_item_.get('menu_type', 'group') == 'group':
+                    if len(menu_item_.get('child')) == 0:
+                        continue
                 menu.append(menu_item_)
         return menu