44"""
55# Standard library
66import argparse
7- import csv
87import os
98import sys
109import textwrap
@@ -82,16 +81,6 @@ def parse_arguments():
8281 return args
8382
8483
85- def data_to_csv (args , data , file_path ):
86- if not args .enable_save :
87- return
88- os .makedirs (PATHS ["data_phase" ], exist_ok = True )
89- # emulate csv.unix_dialect
90- data .to_csv (
91- file_path , index = False , quoting = csv .QUOTE_ALL , lineterminator = "\n "
92- )
93-
94-
9584def process_product_totals (args , count_data ):
9685 """
9786 Processing count data: totals by product
@@ -131,7 +120,7 @@ def process_product_totals(args, count_data):
131120 data .items (), columns = ["CC legal tool product" , "Count" ]
132121 )
133122 file_path = shared .path_join (PATHS ["data_phase" ], "gcs_product_totals.csv" )
134- data_to_csv (args , data , file_path )
123+ shared . data_to_csv (args , data , file_path )
135124
136125
137126def process_latest_prior_retired_totals (args , count_data ):
@@ -212,7 +201,7 @@ def process_latest_prior_retired_totals(args, count_data):
212201 file_path = shared .path_join (
213202 PATHS ["data_phase" ], f"gcs_status_{ key } _totals.csv"
214203 )
215- data_to_csv (args , dataframe , file_path )
204+ shared . data_to_csv (args , dataframe , file_path )
216205
217206
218207def process_totals_by_free_cultural (args , count_data ):
@@ -245,7 +234,7 @@ def process_totals_by_free_cultural(args, count_data):
245234 file_path = shared .path_join (
246235 PATHS ["data_phase" ], "gcs_totals_by_free_cultural.csv"
247236 )
248- data_to_csv (args , data , file_path )
237+ shared . data_to_csv (args , data , file_path )
249238
250239
251240def process_totals_by_restrictions (args , count_data ):
@@ -279,7 +268,7 @@ def process_totals_by_restrictions(args, count_data):
279268 file_path = shared .path_join (
280269 PATHS ["data_phase" ], "gcs_totals_by_restrictions.csv"
281270 )
282- data_to_csv (args , data , file_path )
271+ shared . data_to_csv (args , data , file_path )
283272
284273
285274def process_totals_by_language (args , data ):
@@ -300,7 +289,7 @@ def process_totals_by_language(args, data):
300289 file_path = shared .path_join (
301290 PATHS ["data_phase" ], "gcs_totals_by_language.csv"
302291 )
303- data_to_csv (args , data , file_path )
292+ shared . data_to_csv (args , data , file_path )
304293
305294
306295def process_totals_by_country (args , data ):
@@ -321,14 +310,14 @@ def process_totals_by_country(args, data):
321310 file_path = shared .path_join (
322311 PATHS ["data_phase" ], "gcs_totals_by_country.csv"
323312 )
324- data_to_csv (args , data , file_path )
313+ shared . data_to_csv (args , data , file_path )
325314
326315
327316def main ():
328317 args = parse_arguments ()
329318 shared .paths_log (LOGGER , PATHS )
330319 shared .git_fetch_and_merge (args , PATHS ["repo" ])
331- shared .check_for_data_files (args , FILE_PATHS , QUARTER )
320+ shared .check_completion_file_exists (args , FILE_PATHS )
332321
333322 # Count data
334323 file1_count = shared .path_join (PATHS ["data_1-fetch" ], "gcs_1_count.csv" )
0 commit comments