import datetime class Connection: def __init__(self, details = False): self.services = [] self.details = details self.via = [] def addService(self, service): self.services.append(service) def addVia(self, station): self.via.append(station) def depst(self): try: return self.services[0].depst except: return None def arrst(self): try: return self.services[-1].arrst except: return None def deptime(self): try: return self.services[0].deptime except: return None def arrtime(self): try: return self.services[-1].arrtime except: return None def currdep(self): try: return self.services[0].currdep except: return None def currarr(self): try: return self.services[-1].currarr except: return None def duration(self): try: return self.services[-1].arrtime - self.services[0].deptime except: return None def durationString(self): hrs, scs = divmod(self.duration().total_seconds(), 3600) mns, rmd = divmod(scs, 60) return "%i:%s" % (hrs, str(int(mns)).zfill(2)) def changes(self): return max(len([service for service in self.services if service.name != "Walk"]) - 1, 0) if self.details else max(len([service for service in self.services[0].name.split("/") if service != "Walk"]) - 1, 0) def xml(self, indent = 0, cid = False, frm = True, to = True, deptime = True, arrtime = True, duration = True, changes = True, services = True, via = True, servicekwargs = {}, stationkwargs = {}): out = " " * indent + "\n" % ("" if cid is False else " id=\"%i\"" % cid) if frm: out += " " * indent + " \n" out += self.depst().xml(indent + 2, **stationkwargs) + "\n" out += " " * indent + " \n" if via and self.via != 0: out += " " * indent + " \n" for vst in self.via: out += vst.xml(indent + 2, **stationkwargs) + "\n" out += " " * indent + " \n" if to: out += " " * indent + " \n" out += self.arrst().xml(indent + 2, **stationkwargs) + "\n" out += " " * indent + " \n" if deptime or arrtime or duration or changes: out += " " * indent + "
\n" if deptime: out += " " * indent + " \n" out += " " * indent + " %s\n" % datetime.datetime.strftime(self.deptime(), "%d.%m.%Y") out += " " * indent + " \n" % datetime.datetime.strftime(self.deptime(), "%H:%M") out += " " * indent + " \n" if arrtime: out += " " * indent + " \n" out += " " * indent + " %s\n" % datetime.datetime.strftime(self.arrtime(), "%d.%m.%Y") out += " " * indent + " \n" % datetime.datetime.strftime(self.arrtime(), "%H:%M") out += " " * indent + " \n" out += (" " * indent + " %s\n" % self.durationString()) if duration else "" out += (" " * indent + " %i\n" % self.changes()) if changes else "" out += " " * indent + "
\n" if services: out += " " * indent + " \n" for i in range(len(self.services)): out += self.services[i].xml(indent + 2, i, **servicekwargs) + "\n" out += " " * indent + " \n" out += " " * indent + "" return out def json(self, indent = 0, cid = False, frm = True, to = True, deptime = True, arrtime = True, duration = True, changes = True, services = True, servicekwargs = {}, stationkwargs = {}): out = " " * indent + "{\n" out += (" " * indent + " \"@id\": %i,\n" % cid) if cid is not False else "" if frm: out += " " * indent + " \"from\":\n" out += self.depst().json(indent + 2, **stationkwargs) + ",\n" if to: out += " " * indent + " \"to\":\n" out += self.arrst().json(indent + 2, **stationkwargs) + ",\n" if deptime or arrtime or duration or changes: det = "" det += " " * indent + " \"details\": {\n" if deptime: det += " " * indent + " \"departure\": {\n" det += " " * indent + " \"date\": \"%s\",\n" % datetime.datetime.strftime(self.deptime(), "%d.%m.%Y") det += " " * indent + " \"time\": \"%s\"\n" % datetime.datetime.strftime(self.deptime(), "%H:%M") det += " " * indent + " },\n" if arrtime: det += " " * indent + " \"arrival\": {\n" det += " " * indent + " \"date\": \"%s\",\n" % datetime.datetime.strftime(self.arrtime(), "%d.%m.%Y") det += " " * indent + " \"time\": \"%s\"\n"% datetime.datetime.strftime(self.arrtime(), "%H:%M") det += " " * indent + " },\n" det += (" " * indent + " \"duration\": \"%s\",\n" % self.durationString()) if duration else "" det += (" " * indent + " \"changes\": %i,\n" % self.changes()) if changes else "" det = "".join(det.rsplit(",", 1)) out += det out += " " * indent + " },\n" if services: out += " " * indent + " \"services\": [\n" for i in range(len(self.services)): out += self.services[i].json(indent + 2, i, **servicekwargs) + (",\n" if not i == len(self.services) - 1 else "\n") out += " " * indent + " ],\n" out += " " * indent + "}" out = "".join(out.rsplit(",", 1)) return out