Spaces:
Sleeping
Sleeping
import csv | |
import json | |
def write_json_file(file_json, dict_data): | |
""" | |
--------- | |
Arguments | |
--------- | |
file_json : str | |
full path of json file to be saved | |
dict_data : dict | |
dictionary of params to be saved in the json file | |
""" | |
with open(file_json, "w", encoding="utf-8") as fh: | |
fh.write(json.dumps(dict_data, indent=4)) | |
return | |
class CSVWriter: | |
""" | |
for writing tabular data to a csv file | |
""" | |
def __init__(self, file_name, column_names): | |
""" | |
--------- | |
Arguments | |
--------- | |
file_name : str | |
full path of csv file | |
column_names : list | |
a list of columns names to be used to create the csv file | |
""" | |
self.file_name = file_name | |
self.column_names = column_names | |
self.file_handle = open(self.file_name, "w", encoding="utf-8", newline="\n") | |
self.writer = csv.writer(self.file_handle) | |
self.write_header() | |
print(f"{self.file_name} created successfully with header row") | |
def write_header(self): | |
""" | |
writes header into csv file | |
""" | |
self.write_row(self.column_names) | |
return | |
def write_row(self, row): | |
""" | |
writes a row into csv file | |
""" | |
self.writer.writerow(row) | |
return | |
def close(self): | |
""" | |
close the file | |
""" | |
self.file_handle.close() | |
return | |