Merge pull request #20 from StyleShare/fix-totaltime

Fix an error raised when some node data are empty.
amcfague · Jan 18, 2012 · a01c32c · a01c32c
2 parents 7637864 + 7574c90
commit a01c32c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/linesman/middleware.py b/linesman/middleware.py
@@ -413,6 +413,11 @@ def prepare_graph(source_graph, cutoff_time, break_cycles=False):
     # Always use a copy for destructive changes
     graph = source_graph.copy()
 
+    # Some node data could be empty dict
+    for node, data in graph.nodes(data=True):
+        if not data:
+            data['totaltime'] = 0
+
     max_totaltime = max(data['totaltime']
                         for node, data in graph.nodes(data=True))
     for node, data in graph.nodes(data=True):