From 4a554a56ee6ac3079d933da330d8011287c2e904 Mon Sep 17 00:00:00 2001
From: Rishabh Singh <rishabhsingh971@gmail.com>
Date: Tue, 12 Sep 2023 00:16:00 +0530
Subject: [PATCH] auto detect fieldnames

---
 pyutils/csvu/main.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/pyutils/csvu/main.py b/pyutils/csvu/main.py
index 869abde..edd52fd 100644
--- a/pyutils/csvu/main.py
+++ b/pyutils/csvu/main.py
@@ -96,6 +96,11 @@ def save_csv(file_path, rows, mode, fieldnames=None):
         return
     if isinstance(rows, dict):
         rows = list(rows.values())
+    if isinstance(rows, dict) and fieldnames is None:
+        fields = set()
+        for key in rows:
+            fields.add(key)
+        fieldnames = list(fields)
     write_header = False
     if not os.path.exists(file_path) or mode == "w":
         parent_dir = os.path.dirname(file_path)