summaryrefslogtreecommitdiffstats
path: root/cobbler/Cheetah/Tools/SiteHierarchy.py
blob: d4a92e1e7a450e4128074b8f0449e000ad719e7c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
#!/usr/bin/env python
# $Id: SiteHierarchy.py,v 1.1 2001/10/11 03:25:54 tavis_rudd Exp $
"""Create menus and crumbs from a site hierarchy.

You define the site hierarchy as lists/tuples.  Each location in the hierarchy
is a (url, description) tuple.  Each list has the base URL/text in the 0
position, and all the children coming after it.  Any child can be a list,
representing further depth to the hierarchy.  See the end of the file for an
example hierarchy.

Use Hierarchy(contents, currentURL), where contents is this hierarchy, and
currentURL is the position you are currently in.  The menubar and crumbs methods
give you the HTML output.

There are methods you can override to customize the HTML output.

Meta-Data
================================================================================
Author: Ian Bicking <ianb@colorstudy.com>
Version: $Revision: 1.1 $
Start Date: 2001/07/23
Last Revision Date: $Date: 2001/10/11 03:25:54 $
"""
__author__ = "Ian Bicking <ianb@colorstudy.com>"
__version__ = "$Revision: 1.1 $"[11:-2]

##################################################
## DEPENDENCIES
import string
try:
	from cStringIO import StringIO
except ImportError:
	from StringIO import StringIO


##################################################
## GLOBALS & CONSTANTS

True, False = (1==1), (0==1)

##################################################
## CLASSES

class Hierarchy:
	def __init__(self, hierarchy, currentURL, prefix='', menuCSSClass=None,
				 crumbCSSClass=None):
		"""
		hierarchy is described above, currentURL should be somewhere in
		the hierarchy.  prefix will be added before all of the URLs (to
		help mitigate the problems with absolute URLs), and if given,
		cssClass will be used for both links *and* nonlinks.
		"""

		self._contents = hierarchy
		self._currentURL = currentURL
		if menuCSSClass:
			self._menuCSSClass = ' class="%s"' % menuCSSClass
		else:
			self._menuCSSClass = ''
		if crumbCSSClass:
			self._crumbCSSClass = ' class="%s"' % crumbCSSClass
		else:
			self._crumbCSSClass = ''
		self._prefix=prefix

	
	## Main output methods
	
	def menuList(self, menuCSSClass=None):
		"""An indented menu list"""
		if menuCSSClass:
			self._menuCSSClass = ' class="%s"' % menuCSSClass
		
		stream = StringIO()
		for item in self._contents[1:]:
			self._menubarRecurse(item, 0, stream)
		return stream.getvalue()

	def crumbs(self, crumbCSSClass=None):
		"""The home>where>you>are crumbs"""
		if crumbCSSClass:
			self._crumbCSSClass = ' class="%s"' % crumbCSSClass
		
		path = []
		pos = self._contents
		while 1:
			## This is not the fastest algorithm, I'm afraid.
			## But it probably won't be for a huge hierarchy anyway.
			foundAny = False
			path.append(pos[0])
			for item in pos[1:]:
				if self._inContents(item):
					if type(item) is type(()):
						path.append(item)
						break
					else:
						pos = item
						foundAny = True
						break
			if not foundAny:
				break
		if len(path) == 1:
			return self.emptyCrumb()
		return string.join(map(lambda x, self=self: self.crumbLink(x[0], x[1]),
							   path), self.crumbSeperator()) + \
							   self.crumbTerminator()

	## Methods to control the Aesthetics
	#  - override these methods for your own look
		
	def menuLink(self, url, text, indent):
		if url == self._currentURL or self._prefix + url == self._currentURL:
			return '%s<B%s>%s</B> <BR>\n' % ('&nbsp;'*2*indent,
							 self._menuCSSClass, text)
		else:
			return '%s<A HREF="%s%s"%s>%s</A> <BR>\n' % \
				   ('&nbsp;'*2*indent, self._prefix, url,
					self._menuCSSClass, text)
		
	def crumbLink(self, url, text):
		if url == self._currentURL or self._prefix + url == self._currentURL:
			return '<B%s>%s</B>' % (text, self._crumbCSSClass)
		else:
			return '<A HREF="%s%s"%s>%s</A>' % \
				   (self._prefix, url, self._crumbCSSClass, text)
		
	def crumbSeperator(self):
		return '&nbsp;&gt;&nbsp;'
	
	def crumbTerminator(self):
		return ''
	
	def emptyCrumb(self):
		"""When you are at the homepage"""
		return ''
				
	## internal methods
	
	def _menubarRecurse(self, contents, indent, stream):
		if type(contents) is type(()):
			url, text = contents
			rest = []
		else:
			url, text = contents[0]
			rest = contents[1:]
		stream.write(self.menuLink(url, text, indent))
		if self._inContents(contents):
			for item in rest:
				self._menubarRecurse(item, indent+1, stream)

	def _inContents(self, contents):
		if type(contents) is type(()):
			return self._currentURL == contents[0]
		for item in contents:
			if self._inContents(item):
				return True
		return False

##################################################
## from the command line

if __name__ == '__main__':
	hierarchy = [('/', 'home'),
		    ('/about', 'About Us'),
		    [('/services', 'Services'),
		     [('/services/products', 'Products'),
		      ('/services/products/widget', 'The Widget'),
		      ('/services/products/wedge', 'The Wedge'),
		      ('/services/products/thimble', 'The Thimble'),
		      ],
		     ('/services/prices', 'Prices'),
		     ],
		    ('/contact', 'Contact Us'),
		    ]

	for url in ['/', '/services', '/services/products/widget', '/contact']:
		print '<p>', '='*50
		print '<br> %s: <br>\n' % url
		n = Hierarchy(hierarchy, url, menuCSSClass='menu', crumbCSSClass='crumb',
			      prefix='/here')
		print n.menuList()
		print '<p>', '-'*50
		print n.crumbs()