瀏覽代碼

add a helper to skip some non-exist fields

Xi Ruoyao 5 年之前
父節點
當前提交
e78bcf4e14
共有 1 個文件被更改,包括 3 次插入1 次删除
  1. 3 1
      csv_parser.py

+ 3 - 1
csv_parser.py

@@ -18,6 +18,8 @@ def do_convert(s, fmt):
         return s
     if fmt == 'f':
         return float(s)
+    if fmt == 'n':
+        return "null"
     throw("what is this format?")
 
 def data_from_row(d):
@@ -32,7 +34,7 @@ def data_from_row(d):
             st = d['\ufeff' + name]
         else:
             st = ''
-        if st == '':
+        if st == '' and fmt != 'n':
             raise KeyError
         d1[field] = do_convert(st, fmt)
     if d1['lat'] == 0: