141 lines
5.4 KiB
Python
141 lines
5.4 KiB
Python
import datetime
|
|
|
|
|
|
class Service:
|
|
def __init__(self, name, depst, deptime, arrst, arrtime, dest=None, deppf=None, currdep=None, arrpf=None, curarr=None):
|
|
self.name = name
|
|
self.dest = dest
|
|
self.depst = depst
|
|
self.deptime = deptime
|
|
self.arrst = arrst
|
|
self.arrtime = arrtime
|
|
self.deppf = deppf
|
|
self.currdep = currdep
|
|
self.arrpf = arrpf
|
|
self.curarr = curarr
|
|
|
|
def duration(self):
|
|
return self.arrtime - self.deptime
|
|
|
|
def xml(self, indent=0, iid=False, name=True, depst=True, deptime=True, arrst=True, arrtime=True, deppf=True, currdep=True, arrpf=True, curarr=True, duration=True, dest=True, stationkwargs={}):
|
|
out = " " * indent + \
|
|
"<service%s>\n" % ("" if iid is False else " id=\"%i\"" % iid)
|
|
|
|
out += (" " * indent + " <name>%s</name>\n" %
|
|
self.name) if name else ""
|
|
|
|
if dest and self.dest:
|
|
out += " " * indent + " <destination>\n"
|
|
out += self.dest.xml(indent + 2, **stationkwargs) + "\n"
|
|
out += " " * indent + " </destination>\n"
|
|
|
|
if depst or deptime or deppf or currdep:
|
|
out += " " * indent + " <departure>\n"
|
|
|
|
out += (self.depst.xml(indent + 2, **stationkwargs) +
|
|
"\n") if depst else ""
|
|
|
|
if deptime:
|
|
out += " " * indent + \
|
|
" <date>%s</date>\n" % datetime.datetime.strftime(
|
|
self.deptime, "%d.%m.%Y")
|
|
out += " " * indent + \
|
|
" <time>%s</time>\n" % datetime.datetime.strftime(
|
|
self.deptime, "%H:%M")
|
|
|
|
out += (" " * indent + " <current>%s</current>\n" %
|
|
self.currdep) if currdep and self.currdep else ""
|
|
out += (" " * indent + " <platform>%s</platform>\n" %
|
|
self.deppf) if deppf and self.deppf else ""
|
|
|
|
out += " " * indent + " </departure>\n"
|
|
|
|
if arrst or arrtime or arrpf or curarr:
|
|
out += " " * indent + " <arrival>\n"
|
|
|
|
out += (self.arrst.xml(indent + 2, **stationkwargs) +
|
|
"\n") if arrst else ""
|
|
|
|
if arrtime:
|
|
out += " " * indent + \
|
|
" <date>%s</date>\n" % datetime.datetime.strftime(
|
|
self.arrtime, "%d.%m.%Y")
|
|
out += " " * indent + \
|
|
" <time>%s</time>\n" % datetime.datetime.strftime(
|
|
self.arrtime, "%H:%M")
|
|
|
|
out += (" " * indent + " <current>%s</current>\n" %
|
|
self.curarr) if curarr and self.curarr else ""
|
|
out += (" " * indent + " <platform>%s</platform>\n" %
|
|
self.arrpf) if arrpf and self.arrpf else ""
|
|
|
|
out += " " * indent + " </arrival>\n"
|
|
|
|
out += " " * indent + "</service>"
|
|
|
|
return out
|
|
|
|
def json(self, indent=0, iid=False, name=True, depst=True, deptime=True, arrst=True, arrtime=True, deppf=True, currdep=True, arrpf=True, curarr=True, duration=True, dest=True, stationkwargs={}):
|
|
out = " " * indent + "{\n"
|
|
|
|
out += (" " * indent + " \"@id\": %i,\n" %
|
|
iid) if iid is not False else ""
|
|
out += (" " * indent + " \"name\": \"%s\",\n" %
|
|
self.name) if name else ""
|
|
|
|
if dest and self.dest:
|
|
out += " " * indent + " \"destination\":\n"
|
|
out += self.dest.json(indent + 2, **stationkwargs) + ",\n"
|
|
|
|
if depst or deptime or deppf or currdep:
|
|
dep = " " * indent + " \"departure\": {\n"
|
|
|
|
if depst:
|
|
dep += " " * indent + " \"station\":\n"
|
|
dep += self.depst.json(indent + 3, **stationkwargs) + ",\n"
|
|
|
|
if deptime:
|
|
dep += " " * indent + \
|
|
" \"date\": \"%s\",\n" % datetime.datetime.strftime(
|
|
self.deptime, "%d.%m.%Y")
|
|
dep += " " * indent + \
|
|
" \"time\": \"%s\",\n" % datetime.datetime.strftime(
|
|
self.deptime, "%H:%M")
|
|
|
|
dep += (" " * indent + " \"current\": \"%s\",\n" %
|
|
self.currdep) if currdep and self.currdep else ""
|
|
dep += (" " * indent + " \"platform\": \"%s\",\n" %
|
|
self.deppf) if deppf and self.deppf else ""
|
|
|
|
dep = "".join(dep.rsplit(",", 1))
|
|
out += dep + " " * indent + " },\n"
|
|
|
|
if arrst or arrtime or arrpf or curarr:
|
|
arr = " " * indent + " \"arrival\": {\n"
|
|
|
|
if arrst:
|
|
arr += " " * indent + " \"station\":\n"
|
|
arr += self.arrst.json(indent + 3, **stationkwargs) + ",\n"
|
|
|
|
if arrtime:
|
|
arr += " " * indent + \
|
|
" \"date\": \"%s\",\n" % datetime.datetime.strftime(
|
|
self.arrtime, "%d.%m.%Y")
|
|
arr += " " * indent + \
|
|
" \"time\": \"%s\",\n" % datetime.datetime.strftime(
|
|
self.arrtime, "%H:%M")
|
|
|
|
arr += (" " * indent + " \"current\": \"%s\",\n" %
|
|
self.curarr) if curarr and self.curarr else ""
|
|
arr += (" " * indent + " \"platform\": \"%s\",\n" %
|
|
self.arrpf) if arrpf and self.arrpf else ""
|
|
|
|
arr = "".join(arr.rsplit(",", 1))
|
|
out += arr + " " * indent + " },\n"
|
|
|
|
out += " " * indent + "}"
|
|
|
|
out = "".join(out.rsplit(",", 1))
|
|
|
|
return out
|