import csv

from data.sig_ref import SIGRef
from sigrefdataproviders.local_file_sig_ref_data_provider import LocalFileSIGRefDataProvider


class TOTA(LocalFileSIGRefDataProvider):
    """SIG ref data provider for Toilets on the Air"""

    SIG = "TOTA"
    PATH = "datafiles/tota.csv"

    def __init__(self, provider_config):
        super().__init__(self.SIG, provider_config, self.PATH)

    def _file_to_data(self, path):
        new_data = {}
        f = open(path)
        csv_data = f.read()
        dr = csv.DictReader(csv_data.splitlines())
        for row in dr:
            new_data[row["ref"]] = SIGRef(sig=self.SIG, id=row["ref"], name=row["ref"], latitude=float(row["lat"]),
                                          longitude=float(row["lon"]))
        return new_data