diff options
| -rw-r--r-- | .gitmodules | 3 | ||||
| -rwxr-xr-x | bin/nvm | 41 | ||||
| m--------- | ext/transport-apis | 0 |
3 files changed, 43 insertions, 1 deletions
diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 0000000..e6c1dd2 --- /dev/null +++ b/.gitmodules @@ -0,0 +1,3 @@ +[submodule "transport-apis"] + path = ext/transport-apis + url = https://github.com/public-transport/transport-apis.git @@ -14,8 +14,10 @@ import dateutil.parser from jinja2 import Environment, FileSystemLoader, select_autoescape +import geojson import json import os +import shapely.geometry headers = { "Access-Control-Allow-Origin": "*", @@ -30,6 +32,31 @@ ajax_headers = { db_rest_api = os.getenv("NVM_DB_REST_API", "https://v5.db.transport.rest") env = Environment(loader=FileSystemLoader("templates"), autoescape=select_autoescape()) +apis = None + + +class TransportAPIs: + def __init__(self): + self.apis = list() + base = "ext/transport-apis/data/de" + for filename in os.listdir(base): + with open(f"{base}/{filename}", "r") as f: + data = json.load(f) + if data["type"].get("efa", False): + try: + area = data["coverage"]["realtimeCoverage"]["area"] + except KeyError: + continue + # surely there must be a more elegant way to load a JSON sub-dict as GeoJSON + area = geojson.loads(json.dumps(area)) + self.apis.append((data["options"], shapely.geometry.shape(area))) + + def get_efa(self, location): + location = shapely.geometry.Point(*location) + for api, area in self.apis: + if area.contains(location): + return api + return None class Departure: @@ -42,11 +69,18 @@ class Departure: self.classes = str() self.station_name = None - self.stop_name = None self.stop_name = obj.get("stop", dict()).get("name", None) self.station_name = obj.get("station", dict()).get("name", self.stop_name) + try: + self.location = ( + obj["stop"]["location"]["longitude"], + obj["stop"]["location"]["latitude"], + ) + except KeyError: + self.location = None + if "," in self.direction: self.direction, self.suffix = self.direction.split(",", maxsplit=1) else: @@ -126,6 +160,9 @@ async def show_departure_board(request): departures = list(map(Departure, departures)) + if len(departures): + efa_endpoint = apis.get_efa(departures[0].location) + station_name_freq = dict() now = datetime.now().timestamp() for departure in departures: @@ -190,6 +227,8 @@ if __name__ == "__main__": parser.add_argument("--prefix", type=str, metavar="PATH", default="/") args = parser.parse_args() + apis = TransportAPIs() + app = web.Application() app.router.add_get(args.prefix, show_landing_page) app.router.add_get(f"{args.prefix}board/{{eva}}", show_departure_board) diff --git a/ext/transport-apis b/ext/transport-apis new file mode 160000 +Subproject 45a498e5cb3dd5c03967f7684433caaa7ba45f1 |