Simple MenuItem and Menu classes for keeping track of ordered menus.
authorTodd Larsen <tlarsen@google.com>
Wed, 24 Sep 2008 02:23:32 +0000
changeset 192 f6bf679dab26
parent 191 80f08751f1e5
child 193 3d30a7a96ce7
Simple MenuItem and Menu classes for keeping track of ordered menus.
app/soc/logic/menu.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/app/soc/logic/menu.py	Wed Sep 24 02:23:32 2008 +0000
@@ -0,0 +1,85 @@
+#!/usr/bin/python2.5
+#
+# Copyright 2008 the Melange authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Representations and manipulation of arbitrarily nested menus.
+"""
+
+__authors__ = [
+  '"Todd Larsen" <tlarsen@google.com>',
+  ]
+
+
+from google.appengine.api import users
+
+from django.utils import datastructures
+
+
+class Menu:
+  """Ordered collection of MenuItem objects.
+  
+  MenuItems are retrievable as an ordered list or individually by their
+  MenuItem.text key.
+  """
+  
+  def __init__(self, items=None):
+    """Initializes ordered list of MenuItems.
+    
+    Args:
+      items: list of MenuItem objects in display order
+    """
+    if not items:
+      items = []
+    
+    items = [(i.name, i) for i in items]
+    self._items = datastructures.SortedDict(data=items)
+
+  def getItem(self, name):
+    """Returns a MenuItem retrieved by its MenuItem.text."""
+    return self._items.get(name)
+
+  def setItem(self, item):
+    """Overwrites an existing MenuItem, or appends a new one."""
+    self._items[item.name] = item
+
+  def delItem(self, name):
+    """Removes an existing MenuItem."""
+    del self._items[name]
+
+  def _getItems(self):
+    """Returns an ordered list of the MenuItems."""
+    return self._items.values()
+
+  items = property(_getItems, doc=
+    """Read-only list of MenuItems, for use in templates.""")
+
+
+class MenuItem:
+  """Provides menu item properties as easily-accessible attributes.
+  """
+  
+  def __init__(self, name, selected=False, sub_menu=None):
+    """Initializes the menu item attributes from supplied arguments.
+    
+    Args:
+      name: name of the menu item
+      selected: Boolean indicating if this menu item is selected;
+        default is False
+      sub_menu: a Menu of sub-items to display below this menu item;
+        default is None, indicating no sub-menu
+    """
+    self.name = name
+    self.selected = selected
+    self.sub_menu = sub_menu