From 61d67961ee952510d4fd78ba6b0edcd60dee1a58 Mon Sep 17 00:00:00 2001
From: Jeff Moe <moe@libre.is>
Date: Thu, 19 Sep 2024 18:44:51 -0600
Subject: [PATCH] Output CSV fields nicer

---
 hsparse/parse_csv_contacts.py | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/hsparse/parse_csv_contacts.py b/hsparse/parse_csv_contacts.py
index a10177b..f305dc2 100644
--- a/hsparse/parse_csv_contacts.py
+++ b/hsparse/parse_csv_contacts.py
@@ -22,9 +22,9 @@
 
 """ Read CSV contacts file exported from hubspot."""
 
-import csv
 import argparse
-
+import csv
+import pandas as pd
 
 def parse_args():
     parser = argparse.ArgumentParser(description="Parse Hubspot Contacts CSV Export")
@@ -56,16 +56,13 @@ def parse_args():
 
 
 def csv_dump(CSV):
-    with open(CSV, newline="") as csvfile:
-        contactreader = csv.reader(csvfile, delimiter=",", quotechar='"')
-        for row in contactreader:
-            print(", ".join(row))
+    df = pd.read_csv(CSV, low_memory=False)
+    print(df)
+
 
 def csv_fields(CSV):
-    with open(CSV, newline="") as csvfile:
-        contactreader = csv.reader(csvfile, delimiter=",", quotechar='"')
-        header = next(csvfile)
-        print(header)
+    df = pd.read_csv(CSV, low_memory=False, header=0)
+    print("\n".join([col for col in df.columns]))
 
 
 def main():
@@ -78,5 +75,6 @@ def main():
     if args.fields:
         csv_fields(CSV)
 
+
 if __name__ == "__main__":
     main()