75 lines
3.0 KiB
Python
75 lines
3.0 KiB
Python
"""Informe les usagers des prochains passages des transports en communs de la RATP"""
|
|
|
|
# PYTHON STUFFS #######################################################
|
|
|
|
from nemubot.exception import IMException
|
|
from nemubot.hooks import hook
|
|
from nemubot.module.more import Response
|
|
|
|
from nextstop import ratp
|
|
|
|
@hook.command("ratp",
|
|
help="Affiche les prochains horaires de passage",
|
|
help_usage={
|
|
"TRANSPORT": "Affiche les lignes du moyen de transport donné",
|
|
"TRANSPORT LINE": "Affiche les stations sur la ligne de transport donnée",
|
|
"TRANSPORT LINE STATION": "Affiche les prochains horaires de passage à l'arrêt donné",
|
|
"TRANSPORT LINE STATION DESTINATION": "Affiche les prochains horaires de passage dans la direction donnée",
|
|
})
|
|
def ask_ratp(msg):
|
|
l = len(msg.args)
|
|
|
|
transport = msg.args[0] if l > 0 else None
|
|
line = msg.args[1] if l > 1 else None
|
|
station = msg.args[2] if l > 2 else None
|
|
direction = msg.args[3] if l > 3 else None
|
|
|
|
if station is not None:
|
|
times = sorted(ratp.getNextStopsAtStation(transport, line, station, direction), key=lambda i: i[0])
|
|
|
|
if len(times) == 0:
|
|
raise IMException("la station %s n'existe pas sur le %s ligne %s." % (station, transport, line))
|
|
|
|
(time, direction, stationname) = times[0]
|
|
return Response(message=["\x03\x02%s\x03\x02 direction %s" % (time, direction) for time, direction, stationname in times],
|
|
title="Prochains passages du %s ligne %s à l'arrêt %s" % (transport, line, stationname),
|
|
channel=msg.channel)
|
|
|
|
elif line is not None:
|
|
stations = ratp.getAllStations(transport, line)
|
|
|
|
if len(stations) == 0:
|
|
raise IMException("aucune station trouvée.")
|
|
return Response(stations, title="Stations", channel=msg.channel)
|
|
|
|
elif transport is not None:
|
|
lines = ratp.getTransportLines(transport)
|
|
if len(lines) == 0:
|
|
raise IMException("aucune ligne trouvée.")
|
|
return Response(lines, title="Lignes", channel=msg.channel)
|
|
|
|
else:
|
|
raise IMException("précise au moins un moyen de transport.")
|
|
|
|
|
|
@hook.command("ratp_alert",
|
|
help="Affiche les perturbations en cours sur le réseau")
|
|
def ratp_alert(msg):
|
|
if len(msg.args) == 0:
|
|
raise IMException("précise au moins un moyen de transport.")
|
|
|
|
l = len(msg.args)
|
|
transport = msg.args[0] if l > 0 else None
|
|
line = msg.args[1] if l > 1 else None
|
|
|
|
if line is not None:
|
|
d = ratp.getDisturbanceFromLine(transport, line)
|
|
if "date" in d and d["date"] is not None:
|
|
incidents = "Au {date[date]}, {title}: {message}".format(**d)
|
|
else:
|
|
incidents = "{title}: {message}".format(**d)
|
|
else:
|
|
incidents = ratp.getDisturbance(None, transport)
|
|
|
|
return Response(incidents, channel=msg.channel, nomore="No more incidents", count=" (%d more incidents)")
|