diff --git a/tcl.py b/tcl.py index 1be2d1c..0818842 100644 --- a/tcl.py +++ b/tcl.py @@ -10,6 +10,54 @@ import base64 import difflib +class Terminus: + def __init__(self, name): + self.name = name + self.horaires = [] + + def add_horaire(self, horaire): + self.horaires.append(horaire) + + def get_horaires(self): + return self.sort_horaires() + + def get_name(self): + return self.name + + def sort_horaires(self): + inth = [] + for h in self.horaires: + if h == "Proche": + inth.append(0) + else : + inth.append(int(h[:-4])) + inth.sort() + return inth + +class Ligne: + def __init__(self, name): + self.name = name + self.terminus = [] + + def add_terminus(self, newterminus): + if newterminus in self.terminus: + return + self.terminus.append(newterminus) + + def get_terminus(self, term): + for t in self.terminus: + if t.get_name() == term: + return t + return None + + def get_all_terminus(self): + return self.terminus + + def get_name(self): + return self.name + + + class Config(BaseProxyConfig): def do_update(self, helper: ConfigUpdateHelper) -> None: helper.copy("mail") @@ -98,6 +146,7 @@ class Tcl(Plugin): url = 'https://download.data.grandlyon.com/ws/rdata/tcl_sytral.tclpassagearret/all.json?maxfeatures=2000&start=1' respText = "

Prochains départs " + nomArret + " :