from urllib.parse import quote

from . import poBaseURL
from .urlreader import UrlReader


class PoGraphReader(UrlReader):

    def __init__(self, station: str, days: int):
        """
        Initialisiert die Super-Klasse mit einer URL aus gegebener Station und Anzahl Tage
        :param station: Kurzname der gewünschten Station
        :type station: str
        :param days: gewünschte Anzahl an Tagen
        :type days: int
        """
        super().__init__(poBaseURL + 'stations/%s/W/measurements.png?start=P%dD' % (quote(station), days))

    def download(self):
        """
        Versucht die Grafik über die super.get_data_response Methode herunterzuladen
        """
        print("PoGraphReader::download: Lade Bild herunter...")

        image_data = self.get_data_response()
        if image_data is None or len(image_data) == 0:
            # Keine Daten erhalten → Abbruch
            print("PoGraphReader::download: Fehler: Keine Daten erhalten")
            return None

        print("PoGraphReader::download: Vollständig")
        return image_data