35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
"""
|
|
This module is used to manage the output files of the program.
|
|
"""
|
|
|
|
def format_data(data:dict, separator:str, selected_columns:list = None) -> str:
|
|
"""
|
|
Format the data to be saved in the output file.
|
|
|
|
:param data: Data to be formatted
|
|
:param selected_columns: Columns to be saved
|
|
:param separator: Separator of the columns
|
|
:return: Formatted data
|
|
"""
|
|
output = ''
|
|
if selected_columns is None:
|
|
selected_columns = list(data.keys())
|
|
for column_name in selected_columns:
|
|
output += column_name + separator
|
|
output += '\n'
|
|
for i in range(len(data[selected_columns[0]])):
|
|
for column in selected_columns:
|
|
output += str(data[column][i]).ljust(len(column)) + separator
|
|
output += '\n'
|
|
return output
|
|
|
|
def save_output_file(output_file:str, content:str):
|
|
"""
|
|
Save the output file.
|
|
|
|
:param output_file: Path to the output file
|
|
:param content: Content of the output file
|
|
"""
|
|
with open(output_file, 'w') as f:
|
|
f.write(content)
|