Initial version of geojson data

creates a geojson file in config/www for use in map-card
vingerha · Nov 17, 2023 · b4837cc · b4837cc
1 parent 94e850e
commit b4837cc
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 4 deletions.
diff --git a/custom_components/gtfs2/const.py b/custom_components/gtfs2/const.py
@@ -8,6 +8,7 @@
 
 DEFAULT_NAME = "GTFS Sensor2"
 DEFAULT_PATH = "gtfs2"
+DEFAULT_PATH_GEOJSON = "www"
 
 CONF_DATA = "data"
 CONF_DESTINATION = "destination"

diff --git a/custom_components/gtfs2/coordinator.py b/custom_components/gtfs2/coordinator.py
@@ -101,7 +101,11 @@ async def _async_update_data(self) -> dict[str, str]:
                     self._headers = None
                 self._headers = None
                 self.info = {}
-                self._route_id = self._data["next_departure"]["route_id"]
+                try: 
+                    self._route_id = self._data["next_departure"]["route_id"]
+                except Exception as ex:  # pylint: disable=broad-except
+                    _LOGGER.error("Error getting entity route_id for realtime data, for origin: %s with error: %s", data["origin"], ex)
+                    self._route_id = data["route"].split(": ")[0]
                 self._stop_id = data["origin"].split(": ")[0]
                 self._direction = data["direction"]
                 self._relative = False

diff --git a/custom_components/gtfs2/gtfs_rt_helper.py b/custom_components/gtfs2/gtfs_rt_helper.py
@@ -1,5 +1,7 @@
 import logging
 from datetime import datetime, timedelta
+import json
+import os
 
 import homeassistant.helpers.config_validation as cv
 import homeassistant.util.dt as dt_util
@@ -43,6 +45,8 @@
     DEFAULT_SERVICE,
     DEFAULT_ICON,
     DEFAULT_DIRECTION,
+    DEFAULT_PATH,
+    DEFAULT_PATH_GEOJSON,
 
     TIME_STR_FORMAT
 )
@@ -293,9 +297,12 @@ def get_rt_vehicle_positions(self):
         headers=self._headers,
         label="vehicle positions",
     )
+    #_LOGGER.error("GTFS RT feed entities: %s", feed_entities)
+    geojson_body = []
+    geojson_element = {"geometry": {"coordinates":[],"type": "Point"}, "properties": {"id": "", "title": "", "trip_id": "", "route_id": "", "direction_id": "", "vehicle_id": "", "vehicle_label": ""}, "type": "Feature"}
     for entity in feed_entities:
         vehicle = entity.vehicle
-
+        
         if not vehicle.trip.trip_id:
             # Vehicle is not in service
             continue
@@ -314,8 +321,40 @@ def get_rt_vehicle_positions(self):
             ],
             2,
         )
+        positions[vehicle.trip.trip_id] = vehicle.position        
+
+        #construct geojson only for configured rout/direction
+        if str(self._route_id) == str(vehicle.trip.route_id) and str(self._direction) == str(vehicle.trip.direction_id):
+            geojson_element = {"geometry": {"coordinates":[],"type": "Point"}, "properties": {"id": "", "title": "", "trip_id": "", "route_id": "", "direction_id": "", "vehicle_id": "", "vehicle_label": ""}, "type": "Feature"}
+            geojson_element["geometry"]["coordinates"] = []
+            geojson_element["geometry"]["coordinates"].append(vehicle.position.longitude)
+            geojson_element["geometry"]["coordinates"].append(vehicle.position.latitude)
+            geojson_element["properties"]["id"] = vehicle.trip.trip_id
+            geojson_element["properties"]["title"] = vehicle.trip.trip_id
+            geojson_element["properties"]["trip_id"] = vehicle.trip.trip_id
+            geojson_element["properties"]["route_id"] = vehicle.trip.route_id
+            geojson_element["properties"]["direction_id"] = vehicle.trip.direction_id
+            geojson_element["properties"]["vehicle_id"] = "tbd"
+            geojson_element["properties"]["vehicle_label"] = "tbd"
+            geojson_body.append(geojson_element)
+
+    self.geojson = {"features": geojson_body, "type": "FeatureCollection"}
+
 
-        positions[vehicle.trip.trip_id] = vehicle.position
-
+    #_LOGGER.error("GTFS RT Positions: %s", positions)
+    _LOGGER.error("GTFS RT geojson body: %s", json.dumps(self.geojson))
+    self._route_dir = self._route_id + "_" + self._direction
+    update_geojson(self)
     return positions
+
+
+def update_geojson(self):    
+    geojson_dir = self.hass.config.path(DEFAULT_PATH_GEOJSON)
+    os.makedirs(geojson_dir, exist_ok=True)
+    file = os.path.join(geojson_dir, self._route_dir + ".json")
+    #_LOGGER.error("gtfs geojson file: %s", file)
+    with open(file, "w") as outfile:
+        json.dump(self.geojson, outfile)
+
+