import csv

from data.sig_ref import SIGRef
from sigrefdataproviders.file_download_sig_ref_data_provider import FileDownloadSIGRefDataProvider


class ARLHS(FileDownloadSIGRefDataProvider):
    """SIG ref data provider for Amateur Radio Light House Society"""

    POLL_INTERVAL_DAYS = 30
    SIG = "ARLHS"
    DATA_URL = "https://www.gma.rocks/download/lighthouse.csv"

    def __init__(self, provider_config):
        super().__init__(self.SIG, provider_config, self.DATA_URL, self.POLL_INTERVAL_DAYS)

    def _http_response_to_data(self, http_response):
        new_data = {}
        for row in csv.DictReader(http_response.content.decode("utf-8-sig").splitlines()[1:]):
            if "ARLHS" in row:
                ref_id = row["ARLHS"]
                new_data[ref_id] = SIGRef(sig=self.SIG, id=ref_id, name=row["Name"] if "Name" in row else None,
                                          url="https://www.cqgma.org/zinfo.php?ref=" + ref_id,
                                          latitude=float(row["Latitude"]) if "Latitude" in row and row[
                                              "Latitude"] != "" else None,
                                          longitude=float(row["Longitude"]) if "Longitude" in row and row[
                                              "Longitude"] != "" else None,
                                          grid=row["Maidenhead Locator"])

        return new_data