comparison svgsitemap/middleware.py @ 19:60baa5252aa4

hopefully fix formatting a bit
author Jeff Hammel <jhammel@mozilla.com>
date Sun, 02 Jan 2011 20:35:59 -0800
parents 2aeac6508dc3
children
comparison
equal deleted inserted replaced
18:2aeac6508dc3 19:60baa5252aa4
91 count = int(edge.attr['tooltip'].split(':', 1)[0].strip()) 91 count = int(edge.attr['tooltip'].split(':', 1)[0].strip())
92 else: 92 else:
93 label = edge.attr['label'] 93 label = edge.attr['label']
94 count = int(label) 94 count = int(label)
95 self.label(edge, count) 95 self.label(edge, count)
96 edge.attr['fontsize'] = self.fontsize
96 self.edges[(edge[0], edge[1])] = count 97 self.edges[(edge[0], edge[1])] = count
97 if count > self.max: 98 if count > self.max:
98 self.max = count 99 self.max = count
99 self.remove_orphans() 100 self.remove_orphans()
101 for node in self.graph.nodes():
102 node.attr['fontsize'] = self.fontsize
100 self.set_widths() 103 self.set_widths()
101 else: 104 else:
102 self.graph = AGraph(name=self.name, splines=False, directed=True) 105 self.graph = AGraph(name=self.name, splines=False, directed=True)
103 106
104 # make it pretty 107 # make it pretty
105 self.graph.graph_attr['name'] = self.name 108 self.graph.graph_attr['name'] = self.name
106 self.graph.graph_attr['label'] = self.name 109 self.graph.graph_attr['label'] = self.name
107 self.graph.graph_attr['fontname'] = self.fontname 110 self.graph.graph_attr['fontname'] = self.fontname
108 self.graph.graph_attr['fontcolor'] = self.fontcolor 111 self.graph.graph_attr['fontcolor'] = self.fontcolor
109 self.graph.graph_attr['bgcolor'] = self.bgcolor 112 self.graph.graph_attr['bgcolor'] = self.bgcolor
110 self.graph.graph_attr['overlap'] = 'false' 113 self.graph.graph_attr['overlap'] = 'compact'
111 self.graph.graph_attr['sep'] = '0' 114 self.graph.graph_attr['sep'] = '0'
112 self.graph.node_attr['color'] = self.nodecolor 115 self.graph.node_attr['color'] = self.nodecolor
113 self.graph.node_attr['fontcolor'] = self.fontcolor 116 self.graph.node_attr['fontcolor'] = self.fontcolor
114 self.graph.node_attr['fontname'] = self.fontname 117 self.graph.node_attr['fontname'] = self.fontname
115 self.graph.node_attr['fontsize'] = self.fontsize 118 self.graph.node_attr['fontsize'] = self.fontsize
169 self.label(edge, count) 172 self.label(edge, count)
170 else: 173 else:
171 count = 1 174 count = 1
172 self.edges[(from_url, to_url)] = 1 175 self.edges[(from_url, to_url)] = 1
173 labeltooltip = '1: %s -> %s' % (from_url, to_url) 176 labeltooltip = '1: %s -> %s' % (from_url, to_url)
174 self.graph.add_edge(from_url, to_url, label='', tooltip=labeltooltip, href='#') 177 self.graph.add_edge(from_url, to_url, label='', tooltip=labeltooltip, href='#', fontsize=self.fontsize)
175 178
176 if count > self.max: 179 if count > self.max:
177 self.max = count 180 self.max = count
178 181
179 if self.maxwidth: 182 if self.maxwidth:
181 184
182 for url in from_url, to_url: 185 for url in from_url, to_url:
183 node = self.graph.get_node(url) 186 node = self.graph.get_node(url)
184 node.attr['label'] = url 187 node.attr['label'] = url
185 node.attr['href'] = url 188 node.attr['href'] = url
189 node.attr['fontsize'] = self.fontsize
186 190
187 self.save() 191 self.save()
188 192
189 def label(self, edge, count): 193 def label(self, edge, count):
190 edge.attr['label'] = '' 194 edge.attr['label'] = ''
195 edge.attr['fontsize'] = self.fontsize
191 edge.attr['tooltip'] = '%d: %s -> %s' % (count, edge[0], edge[1]) 196 edge.attr['tooltip'] = '%d: %s -> %s' % (count, edge[0], edge[1])
192 edge.attr['labeltooltip'] = edge.attr['tooltip'] 197 edge.attr['labeltooltip'] = edge.attr['tooltip']
193 edge.attr['href'] = '#' 198 edge.attr['href'] = '#'
194 199
195 def exclude(self, *urls): 200 def exclude(self, *urls):