checks if url is substring of path for root items

Author: lucaskuzma

menu_generator/menu.py

@@ -78,10 +78,20 @@ def _get_url(self, item_dict):
 
     def _is_selected(self, item_dict):
         """
-        Given a menu item dictionary, it returns true if `url` is on path.
+        Given a menu item dictionary, it returns true if `url` is on path,
+        unless the item is marked as a root, in which case returns true if `url` is part of path.
         """
         url = self._get_url(item_dict)
-        return url == self.path
+        if self._is_root(item_dict):
+            return url in self.path
+        else:
+            return url == self.path
+
+    def _is_root(self, item_dict):
+        """
+        Given a menu item dictionary, it returns true if item is marked as a `root`.
+        """
+        return item_dict.get('root', False)
 
     def _process_breadcrums(self, menu_list):
         """