menus.modifiers: 73 total statements, 0.0% covered

Generated: Wed 2013-03-13 10:33 CET

Source file: /media/Envs/Envs/filer-gallery/lib/python2.7/site-packages/menus/modifiers.py

Stats: 0 executed, 71 missed, 2 excluded, 38 ignored

1. # -*- coding: utf-8 -*-
2. from menus.base import Modifier
3. from menus.menu_pool import menu_pool
5. class Marker(Modifier):
6. """
7. searches the current selected node and marks them.
8. current_node: selected = True
9. siblings: sibling = True
10. descendants: descendant = True
11. ancestors: ancestor = True
12. """
13. def modify(self, request, nodes, namespace, root_id, post_cut, breadcrumb):
14. if post_cut or breadcrumb:
15. return nodes
16. selected = None
17. root_nodes = []
18. for node in nodes:
19. if not hasattr(node, "descendant"):
20. node.descendant = False
21. if not hasattr(node, "ancestor"):
22. node.ancestor = False
23. if not node.parent:
24. if selected and not selected.parent:
25. node.sibling = True
26. root_nodes.append(node)
27. if node.selected:
28. if node.parent:
29. newnode = node
30. while newnode.parent:
31. newnode = newnode.parent
32. newnode.ancestor = True
33. for sibling in node.parent.children:
34. if not sibling.selected:
35. sibling.sibling = True
36. else:
37. for root_node in root_nodes:
38. if not root_node.selected:
39. root_node.sibling = True
40. if node.children:
41. self.mark_descendants(node.children)
42. selected = node
43. if node.children:
44. node.is_leaf_node = False
45. else:
46. node.is_leaf_node = True
47. return nodes
49. def mark_descendants(self, nodes):
50. for node in nodes:
51. node.descendant = True
52. self.mark_descendants(node.children)
56. class Level(Modifier):
57. """
58. marks all node levels
59. """
60. post_cut = True
62. def modify(self, request, nodes, namespace, root_id, post_cut, breadcrumb):
63. if breadcrumb:
64. return nodes
65. for node in nodes:
67. if not node.parent:
68. if post_cut:
69. node.menu_level = 0
70. else:
71. node.level = 0
72. self.mark_levels(node, post_cut)
74. return nodes
77. def mark_levels(self, node, post_cut):
78. for child in node.children:
79. if post_cut:
80. child.menu_level = node.menu_level + 1
81. else:
82. child.level = node.level + 1
83. self.mark_levels(child, post_cut)
87. class AuthVisibility(Modifier):
88. """
89. Remove nodes that are login required or require a group
90. """
91. def modify(self, request, nodes, namespace, root_id, post_cut, breadcrumb):
92. if post_cut or breadcrumb:
93. return nodes
94. final = []
95. for node in nodes:
96. if (node.attr.get('visible_for_authenticated', True) and \
97. request.user.is_authenticated()) or \
98. (node.attr.get('visible_for_anonymous', True) and \
99. not request.user.is_authenticated()):
100. final.append(node)
101. else:
102. if node.parent and node in node.parent.children:
103. node.parent.children.remove(node)
104. return final
107. def register():
108. menu_pool.register_modifier(Marker)
109. menu_pool.register_modifier(AuthVisibility)
110. menu_pool.register_modifier(Level)