Refactor csv.py

wgmaster/import.py

@@ -44,17 +44,13 @@ def sqlite_regexp(pattern: str, value: Optional[str]) -> Optional[bool]:
 
 
 def import_csv(cur: sqlite3.Cursor, table_name: str) -> None:
-    rows = list(csv.parse(f'{args.dir}/{table_name}.csv'))
-    if len(rows) == 0:
-        return
-
-    names = tuple(rows[0].keys())
-    values = [tuple(row.values()) for row in rows]
+    rows = csv.parse(f'{args.dir}/{table_name}.csv')
+    names = next(rows)
 
     sql = f'INSERT INTO {table_name} ({", ".join(names)}) VALUES ({", ".join(["?"] * len(names))})'
     print(sql)
-    cur.executemany(sql, values)
-    print(f'{len(values)} inserts')
+    cur.executemany(sql, rows)
+    print(f'{cur.rowcount} inserts')
 
     cur.close()