Merge pull request #13 from HSLdevcom/filter-other-types

Check that the entity is a vehicle position entity
HSLdevcom · Jan 9, 2020 · f34de57 · f34de57
2 parents 14c182e + cbbb5d6
commit f34de57
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gtfsrthttp2mqtt.py b/gtfsrthttp2mqtt.py
@@ -76,6 +76,8 @@ def doGTFSRTPolling(self):
         try:
             feedmsg.ParseFromString(r.content)
             for entity in feedmsg.entity:
+                if not entity.HasField('vehicle'):
+                    continue
                 # Don't send message if route_id is missing from entity
                 # HasField() function does not work as expected, therefore we need check it like this
                 if "route_id" not in map(lambda x: x.name, entity.vehicle.trip.DESCRIPTOR.fields):