from . import poBaseURL
from .po_waterlevel import PoWaterlevel
from .urlreader import UrlReader


class PoWaterlevelReader(UrlReader):

    def __init__(self):
        super().__init__(poBaseURL + 'stations.json?timeseries=W&includeTimeseries=true&includeCurrentMeasurement=true')

    def get_waterlevels(self):
        print("PoWaterlevelReader::get_waterlevels: Lade Pegelstände herunter...")

        stations_json = self.getJsonResponse()
        if stations_json is None or len(stations_json) == 0:
            print("PoWaterlevelReader::get_waterlevels: FEHLER: Keine Pegelstände erhalten")
            return None

        stations = []
        for station_json in stations_json:
            stations.append(PoWaterlevel(station_json))

        print("PoWaterlevelReader::get_waterlevels: %d Pegelstände erhalten" % (len(stations),))

        return stations