"""
This module is used to manage the output files of the program.
"""

from utils.settings.SettingManager import SettingManager


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

    Example:
        >>> data = {
        ...     'col1': [1, 2, 3],
        ...     'col2': [4, 5, 6],
        ...     'col3': [7, 8, 9]
        ... }
        >>> format_data(data, separator=';')
        'col1;col2;col3
         1   ;4   ;7
         2   ;5   ;8
         3   ;6   ;9
        '
    """
    isPrettier = SettingManager.get_instance().get_setting('pretiffy_output_file')
    output = ''
    if selected_columns is None:
        selected_columns = list(data.keys())
    for column_name in selected_columns:
        if isPrettier:
            output += column_name.ljust(len(column_name) if len(column_name) > 8 else 9 )  + separator
        else:
            output += column_name + separator
    output += '\n'
    for i in range(len(data[selected_columns[0]])):
        for column in selected_columns:
            if isPrettier:
                output += str(data[column][i]).ljust(len(column) if len(column) > 8 else 9 ) + separator
            else:
                output += str(data[column][i]) + 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)