def _refresh(self):
from chiki.contrib.common import Item, View
menus = json.loads(Item.data('admin_menus', '[]', name='管理菜单'))
if menus:
views = dict()
for view in View.objects.all():
if view.type == view.TYPE_CATE:
views[view.name] = dict(id=view.name, name=view.label, icon=view.icon, children=[])
else:
views[view.name] = dict(id=view.name, name=view.label, icon=view.icon)
right = []
for menu in menus:
if menu['id'] in views:
item = views[menu['id']]
if 'children' in item and 'children' in menu:
for child in menu['children']:
if child['id'] in views:
item['children'].append(views[child['id']])
right.append(item)
subs = dict()
for menu in self._menu:
if isinstance(item, MenuView):
subs[item._view.__class__.__name__] = menu
elif isinstance(item, MenuCategory):
for child in menu._children:
subs[child._view.__class__.__name__] = child
views = dict((x.__class__.__name__, x) for x in self._views)
menus = []
for menu in right:
if 'children' not in menu:
if menu['id'] in subs:
item = subs[menu['id']]
item.name = menu['name']
item.menu_icon_value = menu['icon']
menus.append(item)
elif menu['id'] in views:
view = views[menu['id']]
item = MenuView(menu['name'], view)
item.menu_icon_value = menu['icon']
menus.append(item)
else:
cate = MenuCategory(menu['name'], icon_value=menu['icon'])
for child in menu['children']:
if child['id'] in subs:
item = subs[child['id']]
item.name = child['name']
item.menu_icon_value = child['icon']
cate._children.append(item)
elif child['id'] in views:
view = views[child['id']]
item = MenuView(child['name'], view)
item.menu_icon_value = child['icon']
cate._children.append(item)
menus.append(cate)
self._menu = menus
def _add_menu_item(self, menu_item, target_category):
"""
Add a view to the menu tree
:param view:
View to add
"""
if target_category:
category = self._menu_categories.get(target_category)
if category is None:
category = MenuCategory(target_category)
self._menu_categories[target_category] = category
self._menu.append(category)
category.add_child(menu_item)
else:
self._menu.append(menu_item)
def _add_menu_item(self, menu_item, target_category):
"""
Add a view to the menu tree
:param view:
View to add
"""
if target_category:
category = self._menu_categories.get(target_category)
if category is None:
category = MenuCategory(target_category)
self._menu_categories[target_category] = category
self._menu.append(category)
category.add_child(menu_item)
else:
self._menu.append(menu_item)