tfa/tclBot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Ajout des horaires théoriques si les estimés pas dispo

Browse Source
This commit is contained in:
tfa
2023-01-04 16:37:03 +01:00
parent bdc2a516d6
commit d02eee7340
1 changed files with 63 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

103
tcl.py
View File

@@ -14,9 +14,19 @@ class Terminus:
def __init__(self, name): def __init__(self, name):
self.name = name self.name = name
self.horaires = [] self.horaires = []
self.estimated = False
def add_horaire(self, horaire): def add_horaire(self, horaire, est):
self.horaires.append(horaire) if self.estimated == False:
if est == "E":
self.horaires = []
self.horaires.append(horaire)
self.estimated = True
else:
self.horaires.append(horaire)
else:
if est == "E":
self.horaires.append(horaire)
def get_horaires(self): def get_horaires(self):
return self.sort_horaires() return self.sort_horaires()
@@ -34,6 +44,9 @@ class Terminus:
inth.sort() inth.sort()
return inth return inth
def get_estimated(self):
return self.estimated
class Ligne: class Ligne:
def __init__(self, name): def __init__(self, name):
self.name = name self.name = name
@@ -57,7 +70,6 @@ class Ligne:
return self.name return self.name
class Config(BaseProxyConfig): class Config(BaseProxyConfig):
def do_update(self, helper: ConfigUpdateHelper) -> None: def do_update(self, helper: ConfigUpdateHelper) -> None:
helper.copy("mail") helper.copy("mail")
@@ -80,6 +92,9 @@ class Tcl(Plugin):
- Funis : F1, F2 - Funis : F1, F2
ATTENTION : Le bot ne vérifie pas que la ligne existe ATTENTION : Le bot ne vérifie pas que la ligne existe
!tcl horaires arrêt -> Retourne les horaires à l'arrêt !tcl horaires arrêt -> Retourne les horaires à l'arrêt
La réponse indique si les horaires sont théoriques (T) ou estimés (E)
Les horaires estimés sont préférés, si pas dispo, les horaires théoriques
sont affichés
""" """
) )
@@ -139,6 +154,9 @@ class Tcl(Plugin):
await evt.respond( await evt.respond(
""" """
!tcl horaires arrêt -> Retourne les horaires à l'arrêt !tcl horaires arrêt -> Retourne les horaires à l'arrêt
La réponse indique si les horaires sont théoriques (T) ou estimés (E)
Les horaires estimés sont préférés, si pas dispo, les horaires théoriques
sont affichés
""" """
) )
return return
@@ -181,43 +199,44 @@ class Tcl(Plugin):
for value in values: for value in values:
if value["id"] in Ids: if value["id"] in Ids:
if value["type"] == "E": #Filtré sur les estimés et non théoriques # Les metros ont leur propre code
# Les metros ont leur propre code if value["coursetheorique"].split('-')[0] == "301A" :
if value["coursetheorique"].split('-')[0] == "301A" : transport = "A"
transport = "A" elif value["coursetheorique"].split('-')[0] == "303" :
elif value["coursetheorique"].split('-')[0] == "303" : transport = "C"
transport = "C" elif value["coursetheorique"].split('-')[0] == "325" :
elif value["coursetheorique"].split('-')[0] == "325" : transport = "F1"
transport = "F1" elif value["coursetheorique"].split('-')[0] == "326" :
elif value["coursetheorique"].split('-')[0] == "326" : transport = "F2"
transport = "F2" elif value["coursetheorique"].split('-')[0] == "304" :
elif value["coursetheorique"].split('-')[0] == "304" : transport = "D"
transport = "D" elif value["coursetheorique"].split('-')[0] == "302A" :
elif value["coursetheorique"].split('-')[0] == "302A" : transport = "B"
transport = "B" else:
l = value["coursetheorique"].split('-')[0]
if l[-1:].isdigit():
transport = l
else: else:
l = value["coursetheorique"].split('-')[0] transport = value["coursetheorique"].split('-')[0][:-1]
if l[-1:].isdigit(): if value["delaipassage"].find('min') < 0 :
transport = l continue
found = False
for line in lines:
if line.get_name() == transport:
term = line.get_terminus(value["direction"])
if term is None:
newterm = Terminus(value["direction"])
newterm.add_horaire(value["delaipassage"], value["type"])
line.add_terminus(newterm)
else: else:
transport = value["coursetheorique"].split('-')[0][:-1] term.add_horaire(value["delaipassage"], value["type"])
found = False found = True
for line in lines: if not found :
if line.get_name() == transport: newLine = Ligne(transport)
term = line.get_terminus(value["direction"]) newterm = Terminus(value["direction"])
if term is None: newterm.add_horaire(value["delaipassage"], value["type"])
newterm = Terminus(value["direction"]) newLine.add_terminus(newterm)
newterm.add_horaire(value["delaipassage"]) lines.append(newLine)
line.add_terminus(newterm)
else:
term.add_horaire(value["delaipassage"])
found = True
if not found :
newLine = Ligne(transport)
newterm = Terminus(value["direction"])
newterm.add_horaire(value["delaipassage"])
newLine.add_terminus(newterm)
lines.append(newLine)
for line in lines: for line in lines:
self.log.info("Ligne " + line.get_name()) self.log.info("Ligne " + line.get_name())
respText += "#### Ligne " + line.get_name() + "\n" respText += "#### Ligne " + line.get_name() + "\n"
@@ -226,7 +245,11 @@ class Tcl(Plugin):
self.log.info(t.get_name()) self.log.info(t.get_name())
if t.get_name() == nomArret: if t.get_name() == nomArret:
continue continue
respText += "***Direction " + t.get_name() + "***\n" respText += "***Direction " + t.get_name()
if t.get_estimated() == True:
respText += " (E)***\n"
else:
respText += " (T)***\n"
hs = t.get_horaires() hs = t.get_horaires()
hs.sort() hs.sort()
for idx, h in enumerate(hs): for idx, h in enumerate(hs):
@@ -238,7 +261,7 @@ class Tcl(Plugin):
if idx == 2: ##On limite à 3 items if idx == 2: ##On limite à 3 items
break break
respText += "\n" respText += "\n"
await evt.respond(respText)#, allow_html=True) await evt.respond(respText)
else: else:
await evt.respond(self.help_msg()) await evt.respond(self.help_msg())