105 lines
2.6 KiB
Python
105 lines
2.6 KiB
Python
from tbm_api.utils import get_data_from_json
|
|
from urllib.parse import quote
|
|
from tbm_api.stop_point import StopPoint
|
|
from re import search
|
|
from tbm_api.route import Route
|
|
|
|
|
|
INFO_URL = "https://ws.infotbm.com/ws/1.0/network/stoparea-informations/%s"
|
|
|
|
LINE_TRANSLATE = {
|
|
"Tram A": "A",
|
|
"Tram B": "B",
|
|
"Tram C": "C",
|
|
"Tram D": "D",
|
|
"TBNight": "58",
|
|
"BAT3": "69",
|
|
}
|
|
|
|
LINE_TYPES = (
|
|
"Tram",
|
|
"Corol",
|
|
"Lianes",
|
|
"Ligne",
|
|
"Bus Relais",
|
|
"Citéis",
|
|
)
|
|
|
|
|
|
class Stop:
|
|
def __init__(self, id, name, latitude, longitude, city):
|
|
self.id = id
|
|
self.name = name
|
|
self.latitude = latitude
|
|
self.longitude = longitude
|
|
self.city = city
|
|
self.stopPoints = []
|
|
|
|
def getId(self):
|
|
return self.id
|
|
|
|
def getName(self):
|
|
return self.name
|
|
|
|
def getLatitude(self):
|
|
return self.latitude
|
|
|
|
def getLongitude(self):
|
|
|
|
return self.longitude
|
|
|
|
def getCity(self):
|
|
return self.city
|
|
|
|
def getStopPoints(self):
|
|
return self.stopPoints
|
|
|
|
def setStopPoints(self, stopPoints):
|
|
self.stopPoints = stopPoints
|
|
|
|
def __repr__(self):
|
|
return self.name + " (" + self.city + ")" + " (id: " + self.id + ")"
|
|
|
|
def __str__(self):
|
|
return self.name + " (" + self.city + ")" + " (id: " + self.id + ")"
|
|
|
|
|
|
def get_stop_by_id(id):
|
|
data = get_data_from_json(INFO_URL % quote(id))
|
|
stop = Stop(
|
|
data["id"],
|
|
data["name"],
|
|
float(data["latitude"]),
|
|
float(data["longitude"]),
|
|
data["city"],
|
|
)
|
|
stopPoints = []
|
|
for i in data["stopPoints"]:
|
|
stopPoint = StopPoint(i["name"])
|
|
routes = []
|
|
stopPoint.setId(int(search("[0-9]+$", i["id"]).group()))
|
|
for j in i["routes"]:
|
|
route = Route(j["name"], j["line"]["name"])
|
|
add = False
|
|
if route.getLineName() in LINE_TRANSLATE:
|
|
line_id = LINE_TRANSLATE[route.getLineName()]
|
|
add = True
|
|
else:
|
|
try:
|
|
line_id = search("[0-9]+$", route.getLineName()).group()
|
|
except AttributeError:
|
|
continue
|
|
line_id = "%02d" % int(line_id)
|
|
for i in LINE_TYPES:
|
|
if route.getLineName()[0 : len(i)] == i:
|
|
add = True
|
|
break
|
|
if add:
|
|
route.setId(line_id)
|
|
routes.append(route)
|
|
stopPoint.setRoutes(routes)
|
|
if stopPoint.getRoutes() != []:
|
|
stopPoints.append(stopPoint)
|
|
stop.setStopPoints(stopPoints)
|
|
return stop
|