from tbm_api.utils import get_data_from_json
from urllib.parse import quote


STOP_URL = "https://ws.infotbm.com/ws/1.0/get-schedule/%s"


class StopArea:
    def __init__(self, id, name, city):
        self.id = id
        self.name = name
        self.city = city

    def getId(self):
        return self.id

    def getName(self):
        return self.name

    def getCity(self):
        return self.city

    def __repr__(self):
        return self.name + " (" + self.city + ")" + " (id: " + self.id + ")"

    def __str__(self):
        return self.name + " (" + self.city + ")" + " (id: " + self.id + ")"


# we on only treat stops of type "stop_area"
def get_stop_areas_by_name(keyword):
    data = get_data_from_json(STOP_URL % quote(keyword))
    stopAreas = []
    for s in data:
        if s["type"] == "stop_area":
            stopAreas.append(StopArea(s["id"], s["name"], s["city"]))
    return stopAreas