Use DictWriter instead of plain loop

ieasybooks · Aug 25, 2023 · a4fc6b3 · a4fc6b3
1 parent f47546a
commit a4fc6b3
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tafrigh/writer.py b/tafrigh/writer.py
@@ -82,11 +82,9 @@ def write_csv(
         delimiter=',',
     ) -> None:
         with open(file_path, 'w', encoding='utf-8') as fp:
-            writer = csv.writer(fp, delimiter=delimiter)
-            writer.writerow(['text', 'start', 'end'])
-
-            for segment in segments:
-                writer.writerow([segment['text'], segment['start'], segment['end']])
+            writer = csv.DictWriter(fp, fieldnames=['text', 'start', 'end'], delimiter=delimiter)
+            writer.writeheader()
+            writer.writerows(segments)
 
     def write_json(
         self,