Changeset 2567

Timestamp:

03/10/10 10:31:02 (5 months ago)

Author:

JensDiemer

Message:

mark first/last <a> in main menu, see:  http://trac.pylucid.net/ticket/365

Location:

branches/0.9/pylucid_project

Files:

• apps/pylucid/tree_model.py (modified) (5 diffs)
• pylucid_plugins/main_menu/templates/main_menu/main_menu_items.html (modified) (1 diff)

branches/0.9/pylucid_project/apps/pylucid/tree_model.py

@@ -18 +18 @@
     $Author: JensDiemer $
 
-    :copyleft: 2007-2009 by the PyLucid team, see AUTHORS for more details.
+    :copyleft: 2007-2010 by the PyLucid team, see AUTHORS for more details.
     :license: GNU GPL v3 or above, see LICENSE for more details.
 """
@@ -25 +25 @@
     # run unittest for this module
     import sys
-    import test_tools # before django imports!
+    from pylucid_project.tests import test_tools # before django imports!
     from django.core import management
     management.call_command('test', "test_TreeModel.TreeModelTest")
+#    management.call_command('test', "pylucid_project.pylucid_plugins.main_menu")
     sys.exit()
 
@@ -44 +45 @@
         self.current = False # current node in main menu?
         self.visible = True  # Seen in main menu?
-        self.level = None
+
+        self.level = None # The level count, starts with 0
+        self.first = False # Is the first node in the current level
+        self.last = False # Is the last node in the current level
+
+    def css_class(self):
+        """
+        returns a css class string, for: ... class="{{ node.css_class }}"
+        So it's not needed to add a few {% if %} template statements. 
+        """
+        class_names = "level_%s" % self.level
+        if self.active:
+            class_names += " active"
+        if self.current:
+            class_names += " current"
+        if self.first:
+            class_names += " first"
+        if self.last:
+            class_names += " last"
+        return class_names
 
     def add(self, node):
@@ -118 +138 @@
             parent.add(self.nodes[node_data["id"]])
 
-        # add level number to all nodes
-        self.setup_level()
+        # add level number to all nodes and mark first/last nodes
+        self.setup_nodes()
 
     def get_first_nodes(self, nodes=None):
@@ -137 +157 @@
                     return nodes2
 
-    def setup_level(self, nodes=None, level=0):
-        """ add level number to all nodes """
+    def setup_nodes(self, nodes=None, level=0):
+        """ add level number to all nodes and mark first/last nodes """
         if nodes == None:
             nodes = self.root.subnodes
+
+        nodes[0].first = True
+        nodes[-1].last = True
+
         for node in nodes:
             node.level = level
             if node.subnodes:
-                self.setup_level(node.subnodes, level + 1)
+                self.setup_nodes(node.subnodes, level + 1)
 
     def debug(self, nodes=None):

branches/0.9/pylucid_project/pylucid_plugins/main_menu/templates/main_menu/main_menu_items.html

@@ -4 +4 @@
             {% if node.visible %}
                 {% if node.current %}<strong>{% endif %}
-                <a href="{{ node.pagemeta.get_absolute_url }}" hreflang="{{ node.pagemeta.language.code }}" title="{{ node.pagemeta.get_title }}" class="level_{{ node.level }}{% if node.active %} active{% endif %}{% if node.current %} current{% endif %}">{{ node.pagemeta.get_name }}</a>
+                <a href="{{ node.pagemeta.get_absolute_url }}" hreflang="{{ node.pagemeta.language.code }}" title="{{ node.pagemeta.get_title }}" class="{{ node.css_class }}">{{ node.pagemeta.get_name }}</a>
                 {% if node.current %}</strong>{% endif %}
             {% endif %}