from . import poBaseURL
from .po_station import PoStation
from .urlreader import UrlReader


class PoStationReader(UrlReader):

    def __init__(self):
        super().__init__(poBaseURL + 'stations.json')

    def get_stations(self):
        print("PoStationReader::get_stations: Lade Stationen herunter...")

        stations_json = self.getJsonResponse()
        if stations_json is None or len(stations_json) == 0:
            print("PoStationReader::get_stations: Keine Stationen erhalten")
            return None

        stations = []
        for station_json in stations_json:
            try:
                stations.append(PoStation(station_json))
            except Exception as e:
                print("PoStationReader::get_stations: Fehler: error=%s, json=%s" % (e, station_json))

        print("PoStationReader::get_stations: %d Stationen erhalten" % (len(stations),))

        return stations