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

tcl.py

@@ -14,8 +14,18 @@ class Terminus:
     def __init__(self, name):
         self.name = name
         self.horaires = []
+        self.estimated = False
 
-    def add_horaire(self, horaire):
+    def add_horaire(self, horaire, est):
+        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):
@@ -34,6 +44,9 @@ class Terminus:
         inth.sort()
         return inth
 
+    def get_estimated(self):
+        return self.estimated
+
 class Ligne:
     def __init__(self, name):
         self.name = name
@@ -57,7 +70,6 @@ class Ligne:
         return self.name
 
 
-
 class Config(BaseProxyConfig):
     def do_update(self, helper: ConfigUpdateHelper) -> None:
         helper.copy("mail")
@@ -80,6 +92,9 @@ class Tcl(Plugin):
                                      - Funis : F1, F2
                         ATTENTION :  Le bot ne vérifie pas que la ligne existe
                 !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( 
                         """
                         !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
@@ -181,7 +199,6 @@ class Tcl(Plugin):
                    
                     for value in values:
                         if value["id"] in Ids:
-                            if value["type"] == "E": #Filtré sur les estimés et non théoriques
                             # Les metros ont leur propre code
                             if value["coursetheorique"].split('-')[0] == "301A" :
                                 transport = "A"
@@ -201,21 +218,23 @@ class Tcl(Plugin):
                                     transport = l
                                 else:
                                     transport = value["coursetheorique"].split('-')[0][:-1]
+                            if value["delaipassage"].find('min') < 0 :
+                                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"])
+                                        newterm.add_horaire(value["delaipassage"], value["type"])
                                         line.add_terminus(newterm)
                                     else:
-                                            term.add_horaire(value["delaipassage"])
+                                        term.add_horaire(value["delaipassage"], value["type"])
                                     found = True
                             if not found :
                                 newLine = Ligne(transport)
                                 newterm = Terminus(value["direction"])
-                                    newterm.add_horaire(value["delaipassage"])
+                                newterm.add_horaire(value["delaipassage"], value["type"])
                                 newLine.add_terminus(newterm)
                                 lines.append(newLine)
                 for line in lines:
@@ -226,7 +245,11 @@ class Tcl(Plugin):
                         self.log.info(t.get_name())
                         if t.get_name() == nomArret:
                             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.sort()
                         for idx, h in enumerate(hs):
@@ -238,7 +261,7 @@ class Tcl(Plugin):
                             if idx == 2: ##On limite à 3 items
                                 break
                         respText += "\n"
-                await evt.respond(respText)#, allow_html=True)
+                await evt.respond(respText)
 
             else:
                 await evt.respond(self.help_msg())