← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to dotgraph.py

  • Committer: abentley
  • Date: 2005-10-16 00:01:58 UTC
  • Revision ID: abentley@lappy-20051016000158-7e35573ad3792faa
Added .tmp to --detrius

Show diffs side-by-side

added added

removed removed

Lines of Context:
dotgraph.py
1
 
# Copyright (C) 2004, 2005 Aaron Bentley
2
 
# <aaron@aaronbentley.com>
3
 
#
4
 
#    This program is free software; you can redistribute it and/or modify
5
 
#    it under the terms of the GNU General Public License as published by
6
 
#    the Free Software Foundation; either version 2 of the License, or
7
 
#    (at your option) any later version.
8
 
#
9
 
#    This program is distributed in the hope that it will be useful,
10
 
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
#    GNU General Public License for more details.
13
 
#
14
 
#    You should have received a copy of the GNU General Public License
15
 
#    along with this program; if not, write to the Free Software
16
 
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
 
 
 
1
#!/usr/bin/env python2.4
18
2
from subprocess import Popen, PIPE
 
3
from urllib import urlencode
 
4
from xml.sax.saxutils import escape
19
5
import os.path
20
6
import errno
21
7
import tempfile
22
8
import shutil
 
9
import time
23
10
 
24
11
RSVG_OUTPUT_TYPES = ('png', 'jpg')
25
 
DOT_OUTPUT_TYPES = ('svg', 'svgz', 'gif', 'jpg', 'ps', 'fig', 'mif', 'png',
 
12
DOT_OUTPUT_TYPES = ('svg', 'svgz', 'gif', 'jpg', 'ps', 'fig', 'mif', 'png', 
26
13
                    'cmapx')
27
14
 
28
15
class NoDot(Exception):
49
36
        self.message = message
50
37
        self.href = None
51
38
 
52
 
    @staticmethod
53
 
    def get_attribute(name, value):
54
 
        if value is None:
55
 
            return ''
56
 
        value = value.replace("\\", "\\\\")
57
 
        value = value.replace('"', '\\"')
58
 
        value = value.replace('\n', '\\n')
59
 
        return '%s="%s"' % (name, value)
60
 
 
61
39
    def define(self):
62
40
        attributes = []
63
41
        style = []
71
49
        if label is not None:
72
50
            attributes.append('label="%s"' % label)
73
51
        attributes.append('shape="box"')
74
 
        tooltip = None
 
52
        tooltip = ''
75
53
        if self.message is not None:
76
 
            tooltip = self.message
77
 
        attributes.append(self.get_attribute('tooltip', tooltip))
 
54
            tooltip += self.message.replace('"', '\\"')
 
55
        if tooltip:
 
56
            attributes.append('tooltip="%s"' % tooltip)
78
57
        if self.href is not None:
79
58
            attributes.append('href="%s"' % self.href)
80
59
        elif tooltip:
203
182
        shutil.rmtree(tempdir)
204
183
    return status
205
184
 
206
 
def invoke_dot(input, out_file=None, file_type='svg', antialias=None,
 
185
def invoke_dot(input, out_file=None, file_type='svg', antialias=None, 
207
186
               fontname="Helvetica", fontsize=11):
208
 
    cmdline = ['dot', '-T%s' % file_type, '-Nfontname=%s' % fontname,
 
187
    cmdline = ['dot', '-T%s' % file_type, '-Nfontname=%s' % fontname, 
209
188
               '-Efontname=%s' % fontname, '-Nfontsize=%d' % fontsize,
210
189
               '-Efontsize=%d' % fontsize]
211
190
    if out_file is not None:
218
197
        else:
219
198
            raise
220
199
    for line in input:
221
 
        dot_proc.stdin.write(line.encode('utf-8'))
 
200
        dot_proc.stdin.write(line)
222
201
    dot_proc.stdin.close()
223
202
    return dot_proc.wait()
224
203