graph: only replace allowed characters which are -,+,_

2011-06-09 11:58:49 +00:00
parent 156935eece
commit 4f4fcff807
1 changed files with 1 additions and 2 deletions
@@ -139,8 +139,7 @@ class Digraph(object):
    def id_str(self, u):
        # Graph format only accepts underscores as key values
        # Sanitize the values. This is 2x faster than the old method.
-        return ''.join([ch if ch not in '!"#$%&\'()*+,-./:;<=>?@[\\]^_`{|}~' \
-                            else '_' for ch in u])
+        return u.replace("-", "_").replace("+", "_")

    def write_graphviz(self, f):
        f.write('digraph G {\n')