85 lines
1.9 KiB
Python
85 lines
1.9 KiB
Python
from math import floor
|
|
from datetime import datetime
|
|
from dateutil import parser
|
|
|
|
class Departure:
|
|
|
|
id_pool = {}
|
|
storage = dict()
|
|
|
|
@staticmethod
|
|
def get(stop_id: str):
|
|
resp = []
|
|
for id in Departure.storage:
|
|
d = Departure.storage[id]
|
|
if d.stop_id == stop_id:
|
|
resp.append(d)
|
|
return resp
|
|
|
|
@staticmethod
|
|
def get_id(stop_id: str):
|
|
if stop_id not in Departure.id_pool:
|
|
Departure.id_pool[stop_id] = []
|
|
i = 0
|
|
while True:
|
|
i += 1
|
|
if i not in Departure.id_pool[stop_id]:
|
|
Departure.id_pool[stop_id].append(i)
|
|
return i
|
|
|
|
@staticmethod
|
|
def clear():
|
|
remove = []
|
|
for did in Departure.storage:
|
|
d = Departure.storage[did]
|
|
if d.get_departure() < -3.5:
|
|
Departure.id_pool[d.stop_id].remove(d.id)
|
|
remove.append(did)
|
|
for did in remove:
|
|
Departure.storage.pop(did)
|
|
|
|
def __init__(self, did, stop_id, type, line, last_stop, departure, delay):
|
|
departure = (parser.parse(departure)).timestamp()
|
|
if -(datetime.now().timestamp() - (departure + delay*60))/60 <= -1:
|
|
return
|
|
self.did = did
|
|
self.stop_id = stop_id
|
|
self.id = Departure.get_id(stop_id)
|
|
self.line = line
|
|
self.type = type
|
|
self.last_stop = last_stop
|
|
self.departure = departure
|
|
self.delay = delay
|
|
self.updated = 1
|
|
Departure.storage[did] = self
|
|
|
|
def update(self, delay):
|
|
if delay != self.delay:
|
|
self.updated += 1
|
|
self.delay = delay
|
|
|
|
def get_departure(self):
|
|
return -(floor(((datetime.now().timestamp() - (self.departure + self.delay*60))/60)*10)/10)
|
|
|
|
def data(self):
|
|
if self.updated != 0:
|
|
self.updated = 0
|
|
return f"{self}"
|
|
else:
|
|
return None
|
|
|
|
def __repr__(self):
|
|
last_stop = self.last_stop
|
|
if len(last_stop) >= 21:
|
|
last_stop = last_stop[:20].strip() + "..."
|
|
return f"{self.id}|{self.type}|{self.line}|{last_stop}|{(self.get_departure()*10):.0f}"
|
|
|
|
def json(self):
|
|
return {
|
|
'id': self.id,
|
|
'line': self.line,
|
|
'type': self.type,
|
|
'last_stop': self.last_stop,
|
|
'departure': self.get_departure(),
|
|
'delay': self.delay
|
|
} |