Make CSV Better

twint/output.py

@@ -7,6 +7,7 @@ import asyncio
 import csv
 import datetime
 import json
+import os
 import re
 import sys
 import time
@@ -15,23 +16,42 @@ def write(entry, f):
 		print(entry, file=open(f, "a", encoding="utf-8"))
 
 def writeCSV(Tweet, file):
-	data = [
-			Tweet.id,
-			Tweet.datestamp,
-			Tweet.timestamp,
-			Tweet.timezone,
-			Tweet.user_id,
-			Tweet.username,
-			Tweet.tweet,
-			Tweet.replies,
-			Tweet.retweets,
-			Tweet.likes,
-			Tweet.location,
-			",".join(Tweet.hashtags),
-			Tweet.link]
+		fieldnames = [
+					"id",
+					"date",
+					"time",
+					"timezone",
+					"user_id",
+					"username",
+					"tweet",
+					"replies",
+					"retweets",
+					"likes",
+					"location",
+					"hashtags",
+					"link"]
+		row = {
+				"id": Tweet.id,
+				"date": Tweet.datestamp,
+				"time": Tweet.timestamp,
+				"timezone": Tweet.timezone,
+				"user_id": Tweet.user_id,
+				"username": Tweet.username,
+				"tweet": Tweet.tweet,
+				"replies": Tweet.replies,
+				"retweets": Tweet.retweets,
+				"likes": Tweet.likes,
+				"location": Tweet.location,
+				"hashtags": Tweet.hashtags,
+				"link": Tweet.link
+				}
+		if not (os.path.exists(file)):
+			with open(file, "w", newline='', encoding="utf-8") as csv_file:
+				writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
+				writer.writeheader()
 		with open(file, "a", newline='', encoding="utf-8") as csv_file:
-		writer = csv.writer(csv_file, quoting=csv.QUOTE_ALL, delimiter=",")
-		writer.writerow(data)
+				writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
+				writer.writerow(row)
 
 def writeJSON(Tweet, file):
 		data = {