aswi2020sebela-gitlab/python-module/Utilities/CSV/CSVutils.py @ 04a2b5a4

# Path to processed data
PROCESSED_DATA_PATH = "ProcessedData/"


def get_unique_names_from_file(filename, column_number):
    """

    Args:
        filename:
        column_number:

    Returns:

    """
    f = open(filename, "r")

    # create set of unique names
    name_set = set()

    # go through every line of line
    for x in f:
        # split by csv splitter ;
        array = x.split(";")
        # add string from chosen column to set
        name_set.add(array[column_number])

    f.close()

    return name_set


def export_data_to_csv(filename, data_dict):
    with open(PROCESSED_DATA_PATH + filename[12:], "w+") as file:

        for date in data_dict:
            for data in data_dict[date]:
                file.write(data_dict[date][data].to_csv() + '\n')