change logic to create_new_columnnames_filename

sfbrigade · Mar 5, 2020 · 330efe2 · 330efe2
1 parent 5e22655
commit 330efe2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/data/create_column_names.py b/data/create_column_names.py
@@ -40,7 +40,9 @@ def create_new_unique_header_mappings(headers, unique_headers_dict, filename):
 def create_new_columnnames_filename(filename):
     # File names are currently in this format SF_Development_Pipeline_2017_Q1.csv
     # parse filename and create a new filename to this format YYYYQ#.txt
-    new_file_arr = filename.split(".")[0].split("_")[-2:]
+
+    new_file_arr = filename.split(".")[0].split("_")
+    new_file_arr = new_file_arr[-3] + new_file_arr[-2][0] + new_file_arr[-1]
     return ''.join(new_file_arr) + '.txt'