Mercurial > hg > svgsitemap
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): |