FreedomBox/menu.py

78 lines
2.8 KiB
Python

from django.core.urlresolvers import reverse
import util
import cfg
class Menu(object):
"""One menu item."""
def __init__(self, label="", icon="", url="#", order=50):
"""label is the text that is displayed on the menu.
icon is the icon to be displayed next to the label.
Choose from the Glyphicon set:
http://twitter.github.com/bootstrap/base-css.html#icons
url is the url location that will be activated when the menu
item is selected.
order is the numerical rank of this item within the menu.
Lower order items appear closest to the top/left of the menu.
By convention, we use the spectrum between 0 and 100 to rank
orders, but feel free to disregard that. If you need more
granularity, don't bother renumbering things. Feel free to
use fractional orders.
"""
self.label = label
self.icon = icon
self.url = url
self.order = order
self.items = []
def get(self, urlname, url_args=None, url_kwargs=None):
"""Return a menu item with given URL name"""
url = reverse(urlname, args=url_args, kwargs=url_kwargs)
url = util.rel_urljoin([cfg.server_dir, url])
for item in self.items:
if item.url == url:
return item
raise KeyError('Menu item not found')
def sort_items(self):
"""Sort the items in self.items by order."""
self.items = sorted(self.items, key=lambda x: x.order, reverse=False)
def add_urlname(self, label, icon, urlname, order=50, url_args=None,
url_kwargs=None):
""" Add a named URL to the menu (via add_item)
url_args and url_kwargs will be passed on to url reverse """
url = reverse(urlname, args=url_args, kwargs=url_kwargs)
return self.add_item(label, icon, url, order, add_url_prefix=True)
def add_item(self, label, icon, url, order=50, add_url_prefix=False):
"""This method creates a menu item with the parameters, adds
that menu item to this menu, and returns the item.
"""
if add_url_prefix:
url = util.rel_urljoin([cfg.server_dir, url])
item = Menu(label=label, icon=icon, url=url, order=order)
self.items.append(item)
self.sort_items()
return item
def is_active(self, request_path):
"""
Returns True if this menu item is active, otherwise False.
We can tell if a menu is active if the menu item points
anywhere above url we are visiting in the url tree.
"""
return request_path.startswith(self.url)
def active_item(self, request):
"""Return the first active item (e.g. submenu) that is found"""
for item in self.items:
if request.path.startswith(item.url):
return item