Small fix

raphv · Aug 22, 2018 · 646a05f · 646a05f
1 parent f80fffd
commit 646a05f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gexf2json.py b/gexf2json.py
@@ -10,6 +10,7 @@
 PADDING = 30
 MIN_EDGE_WIDTH = 1
 MAX_EDGE_WIDTH = 50
+DEFAULT_COLOR = {'r': 127, 'g': 127, 'b': 127}
 
 parser = argparse.ArgumentParser()
 parser.add_argument('input', type=argparse.FileType('r'), help='Input GEXF File')
@@ -60,7 +61,7 @@
     x = float(pos.get('x'))
     y = float(pos.get('y'))
     size = float(node.xpath('viz:size', namespaces=nsmap)[0].get('value'))
-    color = node.xpath('viz:color', namespaces=nsmap)[0]
+    color = (node.xpath('viz:color', namespaces=nsmap) or [DEFAULT_COLOR])[0]
     r = int(color.get('r'))
     g = int(color.get('g'))
     b = int(color.get('b'))