Support all colors of the rainbow in export_svg.py

1 files changed, 7 insertions, 4 deletions

diff --git a/python/examples/export_svg.py b/python/examples/export_svg.py
index 673aa59c..660af780 100755
--- a/python/examples/export_svg.py
+++ b/python/examples/export_svg.py
@@ -179,10 +179,13 @@ def render_svg(function, origname):
         rgb = colors['none']
         try:
             bb = block.basic_block
-            color_code = bb.highlight.color
-            color_str = bb.highlight._standard_color_to_str(color_code)
-            if color_str in colors:
-                rgb = colors[color_str]
+            if hasattr(bb.highlight, 'color'):
+                color_code = bb.highlight.color
+                color_str = bb.highlight._standard_color_to_str(color_code)
+                if color_str in colors:
+                    rgb = colors[color_str]
+            else:
+                rgb = [bb.highlight.red, bb.highlight.green, bb.highlight.blue]
         except:
             pass
         output += '			<rect class="basicblock" x="{x}" y="{y}" fill-opacity="0.4" height="{height}" width="{width}" fill="rgb({r},{g},{b})"/>\n'.format(