comparison svgsitemap/middleware.py @ 9:aa4eab6dc994

* dont set node width, height; * move save() to its own function
author Jeff Hammel <jhammel@mozilla.com>
date Sun, 02 Jan 2011 11:39:45 -0800
parents 1d16bde33eee
children 1329127709de
comparison
equal deleted inserted replaced
8:1d16bde33eee 9:aa4eab6dc994
109 self.graph.node_attr['color'] = self.nodecolor 109 self.graph.node_attr['color'] = self.nodecolor
110 self.graph.node_attr['fontcolor'] = self.fontcolor 110 self.graph.node_attr['fontcolor'] = self.fontcolor
111 self.graph.node_attr['fontname'] = self.fontname 111 self.graph.node_attr['fontname'] = self.fontname
112 self.graph.node_attr['fontsize'] = self.fontsize 112 self.graph.node_attr['fontsize'] = self.fontsize
113 self.graph.node_attr['shape'] = self.shape 113 self.graph.node_attr['shape'] = self.shape
114 self.graph.node_attr['width'] = '0.1'
115 self.graph.node_attr['height'] = '0.1'
116 self.graph.edge_attr['color'] = self.edgecolor 114 self.graph.edge_attr['color'] = self.edgecolor
117 self.graph.edge_attr['fontcolor'] = self.fontcolor 115 self.graph.edge_attr['fontcolor'] = self.fontcolor
118 self.graph.edge_attr['fontname'] = self.fontname 116 self.graph.edge_attr['fontname'] = self.fontname
119 self.graph.edge_attr['fontsize'] = self.fontsize 117 self.graph.edge_attr['fontsize'] = self.fontsize
120 self.graph.edge_attr['len'] = self.len 118 self.graph.edge_attr['len'] = self.len
121 self.graph.edge_attr['arrowsize'] = self.arrowsize 119 self.graph.edge_attr['arrowsize'] = self.arrowsize
120
121 if self.edges:
122 self.save()
122 123
123 def __call__(self, environ, start_response): 124 def __call__(self, environ, start_response):
124 request = Request(environ) 125 request = Request(environ)
125 if request.referer: 126 if request.referer:
126 127
174 for url in from_url, to_url: 175 for url in from_url, to_url:
175 node = self.graph.get_node(url) 176 node = self.graph.get_node(url)
176 node.attr['label'] = url 177 node.attr['label'] = url
177 node.attr['href'] = url 178 node.attr['href'] = url
178 179
179 if self.file: 180 self.save()
180 self.graph.write(self.file)
181 if self.output:
182 self.graph.draw(self.output, prog='neato')
183 181
184 def exclude(self, *urls): 182 def exclude(self, *urls):
185 """tell whether the edge is excluded""" 183 """tell whether the edge is excluded"""
186 for pattern in self.excluded: 184 for pattern in self.excluded:
187 for url in urls: 185 for url in urls:
202 edge.attr['style'] = 'setlinewidth(%s)' % width 200 edge.attr['style'] = 'setlinewidth(%s)' % width
203 201
204 else: 202 else:
205 for edge in self.graph.edges(): 203 for edge in self.graph.edges():
206 edge.attr['style'] = '' 204 edge.attr['style'] = ''
205
206 def save(self):
207 if self.file:
208 self.graph.write(self.file)
209 if self.output:
210 self.graph.draw(self.output, prog='neato')