Fixes #195 max_len doesn't like empty iterator

eyeonus · Mar 7, 2015 · a587967 · a587967
1 parent 1b55106
commit a587967
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/commands/local_cmd.py b/commands/local_cmd.py
@@ -70,6 +70,7 @@ def run(results, cmdenv, tdb):
     results.summary = ResultRow()
     results.summary.near = srcSystem
     results.summary.ly = ly
+    results.summary.stations = 0
 
     distances = { srcSystem: 0.0 }
 
@@ -130,6 +131,7 @@ def station_filter(system):
         row.dist = dist
         row.stations = stations if showStations else []
         results.rows.append(row)
+        results.summary.stations += len(row.stations)
 
     return results
 

diff --git a/formatting.py b/formatting.py
@@ -1,4 +1,5 @@
 from __future__ import absolute_import, with_statement, print_function, division, unicode_literals
+import itertools
 
 class ColumnFormat(object):
     """
@@ -165,6 +166,11 @@ def format(self, rowData):
         return self.prefix + ' '.join(col.format(rowData) for col in self.columns)
 
 def max_len(iterable, key=lambda item: item):
+    iterable, readahead = itertools.tee(iter(iterable))
+    try:
+        next(readahead)
+    except StopIteration:
+        return 0
     return max(len(key(item)) for item in iterable)
 
 if __name__ == '__main__':