Write data to CSV file 🐻

Author: zkan

postgres/python/main.py

@@ -1,4 +1,5 @@
 import configparser
+import csv
 
 import psycopg2
 
@@ -23,6 +24,19 @@
     results = cursor.fetchall()
     print(results)
 
+    with open("products.csv", "w") as f:
+        writer = csv.writer(f)
+        columns = [(
+            "UPC",
+            "DESCRIPTION",
+            "MANUFACTURER",
+            "CATEGORY",
+            "SUB_CATEGORY",
+            "PRODUCT_SIZE",
+        )]
+        writer.writerows(columns)
+        writer.writerows(results)
+
     query = "SELECT * FROM product;"
     cursor = conn.cursor()
     cursor.execute(query)