Merge c429bb8 into 195ee5c

paetzke · Mar 3, 2014 · a22b0b6 · a22b0b6
2 parents 195ee5c + c429bb8
commit a22b0b6
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/dotter/dotter.py b/dotter/dotter.py
@@ -170,12 +170,14 @@ def add_edge(self, node1, node2, label=None):
             fmt += ' [label="{0}"]'.format(label)
         self.execute(fmt.format(Dotter.escape(node1), Dotter.escape(node2)))
 
-    def add_node(self, node, font=None, fontsize=None, label=None, shape=None, url=None):
+    def add_node(self, node, font=None, fontsize=None, label=None, shape=None,
+                 url=None, style=None):
         self.execute('{0}'.format(Dotter.escape(node)))
         self.set_label(node, label if label else node)
-        self.node_attributes(node, font, fontsize, shape, url)
+        self.node_attributes(node, font, fontsize, shape, url, style)
 
-    def node_attributes(self, node, font=None, fontsize=None, shape=None, url=None):
+    def node_attributes(self, node, font=None, fontsize=None, shape=None,
+                        url=None, style=None):
         if font:
             cmd = '{0} [fontname="{1}"]'.format(Dotter.escape(node), font)
             self.execute(cmd)
@@ -188,6 +190,9 @@ def node_attributes(self, node, font=None, fontsize=None, shape=None, url=None):
         if url:
             cmd = '{0} [URL="{1}"]'.format(Dotter.escape(node), url)
             self.execute(cmd)
+        if style:
+            cmd = '{0} [style="{1}"]'.format(Dotter.escape(node), style)
+            self.execute(cmd)
 
     def set_label(self, node, label):
         self.execute('{0} [label="{1}"]'.format(Dotter.escape(node), label))