Added data folder to path for file export

dnaeye · Dec 31, 2020 · 8bc4152 · 8bc4152
1 parent 744bfcc
commit 8bc4152
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/boomkat.py b/boomkat.py
@@ -91,4 +91,6 @@ def get_page(year):
         print("Error getting chart item")
 
 filename = "boomkat_" + user_year + ".csv"
-df.to_csv(filename, index=False, encoding='utf-8-sig')
+if not os.path.exists('data'):
+    os.makedirs('data')
+df.to_csv("data/" + filename, index=False, encoding='utf-8-sig')
diff --git a/pitchfork.py b/pitchfork.py
@@ -67,4 +67,6 @@ def get_page(year):
 df = df.sort_values('rank', ascending=True)
 
 filename = "pitchfork_" + user_year + ".csv"
-df.to_csv(filename, index=False, encoding='utf-8-sig')
+if not os.path.exists('data'):
+    os.makedirs('data')
+df.to_csv("data/" + filename, index=False, encoding='utf-8-sig')
diff --git a/spotify.py b/spotify.py
@@ -94,4 +94,4 @@ def get_input():
 
 df['html'] = pd.Series(html)
 
-df.to_csv(site + "_" + year + "_html.csv", index=False)
+df.to_csv("data/" + site + "_" + year + "_html.csv", index=False)
Original file line number	Diff line number	Diff line change
Expand Up		@@ -94,4 +94,4 @@ def get_input():

		df['html'] = pd.Series(html)

		df.to_csv(site + "_" + year + "_html.csv", index=False)
		df.to_csv("data/" + site + "_" + year + "_html.csv", index=False)