Extend rows that are too short with None

weecology · Jan 17, 2017 · 540a157 · 540a157
1 parent 13993aa
commit 540a157
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/table.py b/lib/table.py
@@ -128,6 +128,12 @@ def values_from_line(self, line):
                 # too many values for columns; ignore
                 pass
             column += 1
+
+        # make sure we have enough values by padding with None
+        keys = self.get_insert_columns(join=False, create=False)
+        if len(linevalues) < len(keys):
+            linevalues.extend([None for _ in range(len(keys) - len(linevalues))])
+
         return linevalues
 
     def get_insert_columns(self, join=True, create=False):