Spaces:
Sleeping
Sleeping
File size: 1,519 Bytes
bd421ea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
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
|