...

1 files changed, 74 insertions, 0 deletions

diff --git a/menu.py b/menu.py
new file mode 100644
index 0000000..e0e376a
--- /dev/null
+++ b/menu.py
@@ -0,0 +1,74 @@
+import simplejson as json
+from urlparse import urlparse
+import cherrypy
+import cfg
+
+class Menu():
+    """One menu item."""
+    def __init__(self, label="", url="#", order=50):
+        """label is the text that is displayed on the menu.
+
+        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.order = order
+        self.url = url
+        self.items = []
+
+    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_item(self, label, url, order=50):
+        """This method creates a menu item with the parameters, adds
+        that menu item to this menu, and returns the item."""
+        item = Menu(label=label, url=url, order=order)
+        self.items.append(item)
+        self.sort_items()
+        return item
+
+    def active_p(self):
+        """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 urlparse(cherrypy.url()).path.startswith(self.url)
+
+    def active_item(self):
+        """Return item list (e.g. submenu) of active menu item."""
+        path = urlparse(cherrypy.url()).path
+        for item in self.items:
+            if path.startswith(item.url):
+                return item
+
+    def serializable(self, render_subs=False):
+        """Return the items in self.items as a serializable object we can pass to json.
+        Note: this doesn't serialize all the data in this object."""
+
+        so = []
+        for item in self.items:
+            i = { 'label':item.label, 'url':item.url}
+            if item.active_p():
+                i['active']=True
+            if item.items and render_subs:
+                i['subs'] = item.serializable()
+            so.append(i)
+        return so
+    def encode(self, name="", render_subs=False):
+        """return a string containing a javascript data structure
+        assigned to the menu name
+
+        if render_subs is True, we render submenus too"""
+
+        return ('<SCRIPT LANGUAGE="JavaScript">\n    <!--\n      var %s_items=' % name
+                #+ json.dumps(self.serializable(render_subs=render_subs), separators=(',',':')) # compact
+                + "\n"+ json.dumps(self.serializable(render_subs=render_subs), sort_keys=True, indent=4) # pretty print
+                + ';\n    // -->\n    </SCRIPT>')