From a4fc6b387834b1638a6dad232ba5dcc03c3f4805 Mon Sep 17 00:00:00 2001 From: Ali Hamdi Ali Fadel Date: Fri, 25 Aug 2023 16:27:30 +0300 Subject: [PATCH] Use DictWriter instead of plain loop --- tafrigh/writer.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/tafrigh/writer.py b/tafrigh/writer.py index ab5d56d..47ed889 100644 --- 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,