From 8885d094311f3d733da560bce3df31038c3beb5b Mon Sep 17 00:00:00 2001 From: Prafullsh Date: Wed, 13 Nov 2024 20:34:46 +0530 Subject: [PATCH 1/7] add new field run number --- .../bindings/intersection.proto | 9 ++++ .../bindings/intersection_pb2.py | 52 ++++++++++++++++++- .../factories/factories.py | 8 ++- .../translators/cta_subway.py | 4 +- 4 files changed, 70 insertions(+), 3 deletions(-) diff --git a/gtfs_realtime_translators/bindings/intersection.proto b/gtfs_realtime_translators/bindings/intersection.proto index e2cd94a..590ce1f 100644 --- a/gtfs_realtime_translators/bindings/intersection.proto +++ b/gtfs_realtime_translators/bindings/intersection.proto @@ -29,3 +29,12 @@ extend transit_realtime.TripUpdate { extend transit_realtime.TripUpdate.StopTimeUpdate { optional IntersectionStopTimeUpdate intersection_stop_time_update = 1987; } + + +message IntersectionVehicleDescriptor { + optional uint32 run_number = 1; +} + +extend transit_realtime.VehicleDescriptor { + optional IntersectionVehicleDescriptor intersection_vehicle_descriptor = 1988; +} \ No newline at end of file diff --git a/gtfs_realtime_translators/bindings/intersection_pb2.py b/gtfs_realtime_translators/bindings/intersection_pb2.py index 3c38512..6b14047 100644 --- a/gtfs_realtime_translators/bindings/intersection_pb2.py +++ b/gtfs_realtime_translators/bindings/intersection_pb2.py @@ -21,7 +21,7 @@ package='', syntax='proto2', serialized_options=None, - serialized_pb=_b('\n\x12intersection.proto\x1a\x13gtfs-realtime.proto\"\xfe\x01\n\x16IntersectionTripUpdate\x12\x10\n\x08headsign\x18\x01 \x01(\t\x12\x18\n\x10route_short_name\x18\x02 \x01(\t\x12\x17\n\x0froute_long_name\x18\x03 \x01(\t\x12\x13\n\x0broute_color\x18\x04 \x01(\t\x12\x18\n\x10route_text_color\x18\x05 \x01(\t\x12\x10\n\x08\x62lock_id\x18\x06 \x01(\t\x12\x17\n\x0f\x61gency_timezone\x18\x07 \x01(\t\x12\x15\n\rcustom_status\x18\x08 \x01(\t\x12\x1a\n\x12scheduled_interval\x18\t \x01(\x05\x12\x12\n\nroute_icon\x18\n \x01(\t\"\xce\x01\n\x1aIntersectionStopTimeUpdate\x12\r\n\x05track\x18\x01 \x01(\t\x12\x45\n\x11scheduled_arrival\x18\x02 \x01(\x0b\x32*.transit_realtime.TripUpdate.StopTimeEvent\x12G\n\x13scheduled_departure\x18\x03 \x01(\x0b\x32*.transit_realtime.TripUpdate.StopTimeEvent\x12\x11\n\tstop_name\x18\x04 \x01(\t:X\n\x18intersection_trip_update\x12\x1c.transit_realtime.TripUpdate\x18\xc3\x0f \x01(\x0b\x32\x17.IntersectionTripUpdate:p\n\x1dintersection_stop_time_update\x12+.transit_realtime.TripUpdate.StopTimeUpdate\x18\xc3\x0f \x01(\x0b\x32\x1b.IntersectionStopTimeUpdate') + serialized_pb=_b('\n\x12intersection.proto\x1a\x13gtfs-realtime.proto\"\xfe\x01\n\x16IntersectionTripUpdate\x12\x10\n\x08headsign\x18\x01 \x01(\t\x12\x18\n\x10route_short_name\x18\x02 \x01(\t\x12\x17\n\x0froute_long_name\x18\x03 \x01(\t\x12\x13\n\x0broute_color\x18\x04 \x01(\t\x12\x18\n\x10route_text_color\x18\x05 \x01(\t\x12\x10\n\x08\x62lock_id\x18\x06 \x01(\t\x12\x17\n\x0f\x61gency_timezone\x18\x07 \x01(\t\x12\x15\n\rcustom_status\x18\x08 \x01(\t\x12\x1a\n\x12scheduled_interval\x18\t \x01(\x05\x12\x12\n\nroute_icon\x18\n \x01(\t\"\xce\x01\n\x1aIntersectionStopTimeUpdate\x12\r\n\x05track\x18\x01 \x01(\t\x12\x45\n\x11scheduled_arrival\x18\x02 \x01(\x0b\x32*.transit_realtime.TripUpdate.StopTimeEvent\x12G\n\x13scheduled_departure\x18\x03 \x01(\x0b\x32*.transit_realtime.TripUpdate.StopTimeEvent\x12\x11\n\tstop_name\x18\x04 \x01(\t\"3\n\x1dIntersectionVehicleDescriptor\x12\x12\n\nrun_number\x18\x01 \x01(\r:X\n\x18intersection_trip_update\x12\x1c.transit_realtime.TripUpdate\x18\xc3\x0f \x01(\x0b\x32\x17.IntersectionTripUpdate:p\n\x1dintersection_stop_time_update\x12+.transit_realtime.TripUpdate.StopTimeUpdate\x18\xc3\x0f \x01(\x0b\x32\x1b.IntersectionStopTimeUpdate:m\n\x1fintersection_vehicle_descriptor\x12#.transit_realtime.VehicleDescriptor\x18\xc4\x0f \x01(\x0b\x32\x1e.IntersectionVehicleDescriptor') , dependencies=[gtfs__realtime__pb2.DESCRIPTOR,]) @@ -42,6 +42,14 @@ message_type=None, enum_type=None, containing_type=None, is_extension=True, extension_scope=None, serialized_options=None, file=DESCRIPTOR) +INTERSECTION_VEHICLE_DESCRIPTOR_FIELD_NUMBER = 1988 +intersection_vehicle_descriptor = _descriptor.FieldDescriptor( + name='intersection_vehicle_descriptor', full_name='intersection_vehicle_descriptor', index=2, + number=1988, type=11, cpp_type=10, label=1, + has_default_value=False, default_value=None, + message_type=None, enum_type=None, containing_type=None, + is_extension=True, extension_scope=None, + serialized_options=None, file=DESCRIPTOR) _INTERSECTIONTRIPUPDATE = _descriptor.Descriptor( @@ -189,12 +197,45 @@ serialized_end=507, ) + +_INTERSECTIONVEHICLEDESCRIPTOR = _descriptor.Descriptor( + name='IntersectionVehicleDescriptor', + full_name='IntersectionVehicleDescriptor', + filename=None, + file=DESCRIPTOR, + containing_type=None, + fields=[ + _descriptor.FieldDescriptor( + name='run_number', full_name='IntersectionVehicleDescriptor.run_number', index=0, + number=1, type=13, cpp_type=3, label=1, + has_default_value=False, default_value=0, + message_type=None, enum_type=None, containing_type=None, + is_extension=False, extension_scope=None, + serialized_options=None, file=DESCRIPTOR), + ], + extensions=[ + ], + nested_types=[], + enum_types=[ + ], + serialized_options=None, + is_extendable=False, + syntax='proto2', + extension_ranges=[], + oneofs=[ + ], + serialized_start=509, + serialized_end=560, +) + _INTERSECTIONSTOPTIMEUPDATE.fields_by_name['scheduled_arrival'].message_type = gtfs__realtime__pb2._TRIPUPDATE_STOPTIMEEVENT _INTERSECTIONSTOPTIMEUPDATE.fields_by_name['scheduled_departure'].message_type = gtfs__realtime__pb2._TRIPUPDATE_STOPTIMEEVENT DESCRIPTOR.message_types_by_name['IntersectionTripUpdate'] = _INTERSECTIONTRIPUPDATE DESCRIPTOR.message_types_by_name['IntersectionStopTimeUpdate'] = _INTERSECTIONSTOPTIMEUPDATE +DESCRIPTOR.message_types_by_name['IntersectionVehicleDescriptor'] = _INTERSECTIONVEHICLEDESCRIPTOR DESCRIPTOR.extensions_by_name['intersection_trip_update'] = intersection_trip_update DESCRIPTOR.extensions_by_name['intersection_stop_time_update'] = intersection_stop_time_update +DESCRIPTOR.extensions_by_name['intersection_vehicle_descriptor'] = intersection_vehicle_descriptor _sym_db.RegisterFileDescriptor(DESCRIPTOR) IntersectionTripUpdate = _reflection.GeneratedProtocolMessageType('IntersectionTripUpdate', (_message.Message,), dict( @@ -211,9 +252,18 @@ )) _sym_db.RegisterMessage(IntersectionStopTimeUpdate) +IntersectionVehicleDescriptor = _reflection.GeneratedProtocolMessageType('IntersectionVehicleDescriptor', (_message.Message,), dict( + DESCRIPTOR = _INTERSECTIONVEHICLEDESCRIPTOR, + __module__ = 'intersection_pb2' + # @@protoc_insertion_point(class_scope:IntersectionVehicleDescriptor) + )) +_sym_db.RegisterMessage(IntersectionVehicleDescriptor) + intersection_trip_update.message_type = _INTERSECTIONTRIPUPDATE gtfs__realtime__pb2.TripUpdate.RegisterExtension(intersection_trip_update) intersection_stop_time_update.message_type = _INTERSECTIONSTOPTIMEUPDATE gtfs__realtime__pb2.TripUpdate.StopTimeUpdate.RegisterExtension(intersection_stop_time_update) +intersection_vehicle_descriptor.message_type = _INTERSECTIONVEHICLEDESCRIPTOR +gtfs__realtime__pb2.VehicleDescriptor.RegisterExtension(intersection_vehicle_descriptor) # @@protoc_insertion_point(module_scope) diff --git a/gtfs_realtime_translators/factories/factories.py b/gtfs_realtime_translators/factories/factories.py index fe6ba62..72ebb3c 100644 --- a/gtfs_realtime_translators/factories/factories.py +++ b/gtfs_realtime_translators/factories/factories.py @@ -62,6 +62,7 @@ def create(*args, **kwargs): custom_status = kwargs.get('custom_status', None) scheduled_interval = kwargs.get('scheduled_interval', None) route_icon = kwargs.get('route_icon', None) + run_number = kwargs.get('run_number', None) trip_descriptor = gtfs_realtime.TripDescriptor(trip_id=trip_id, route_id=route_id, @@ -70,6 +71,8 @@ def create(*args, **kwargs): stop_time_update = gtfs_realtime.TripUpdate.StopTimeUpdate(arrival=arrival, departure=departure, stop_id=stop_id) + + vehicle_descriptor = gtfs_realtime.TripUpdate.VehicleDescriptor(run_number=run_number) if track: stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].track = track @@ -79,9 +82,12 @@ def create(*args, **kwargs): stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].scheduled_departure.time = scheduled_departure if stop_name: stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].stop_name = stop_name + if run_number: + vehicle_descriptor.Extensions[intersection_gtfs_realtime.intersection_vehicle_descriptor].run_number = run_number trip_update = gtfs_realtime.TripUpdate(trip=trip_descriptor, - stop_time_update=[stop_time_update]) + stop_time_update=[stop_time_update], + run_number=run_number) if headsign: trip_update.Extensions[intersection_gtfs_realtime.intersection_trip_update].headsign = headsign diff --git a/gtfs_realtime_translators/translators/cta_subway.py b/gtfs_realtime_translators/translators/cta_subway.py index 9a2eaf1..da2c400 100644 --- a/gtfs_realtime_translators/translators/cta_subway.py +++ b/gtfs_realtime_translators/translators/cta_subway.py @@ -53,6 +53,7 @@ def __make_trip_update(cls, _id, prediction): prediction['schInt']) route_icon = cls.__get_route_icon(prediction['flags'], headsign) + run_number = prediction['rn'] return TripUpdate.create(entity_id=entity_id, route_id=route_id, @@ -63,7 +64,8 @@ def __make_trip_update(cls, _id, prediction): custom_status=custom_status, agency_timezone=cls.TIMEZONE, scheduled_interval=scheduled_interval, - route_icon = route_icon) + route_icon=route_icon, + run_number=run_number) @classmethod def __get_custom_status(cls, arrival_time, prediction_time): From 6d1c3a9cd5b2d9d3547d9caec0313fa95915bcfd Mon Sep 17 00:00:00 2001 From: Prafullsh Date: Thu, 14 Nov 2024 00:12:28 +0530 Subject: [PATCH 2/7] new line --- gtfs_realtime_translators/bindings/intersection.proto | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gtfs_realtime_translators/bindings/intersection.proto b/gtfs_realtime_translators/bindings/intersection.proto index 590ce1f..db2d32f 100644 --- a/gtfs_realtime_translators/bindings/intersection.proto +++ b/gtfs_realtime_translators/bindings/intersection.proto @@ -37,4 +37,4 @@ message IntersectionVehicleDescriptor { extend transit_realtime.VehicleDescriptor { optional IntersectionVehicleDescriptor intersection_vehicle_descriptor = 1988; -} \ No newline at end of file +} From 971a63731ed1ea152baa0c1021e1c920228c5667 Mon Sep 17 00:00:00 2001 From: Prafullsh Date: Thu, 14 Nov 2024 00:36:26 +0530 Subject: [PATCH 3/7] update descriptor --- gtfs_realtime_translators/factories/factories.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gtfs_realtime_translators/factories/factories.py b/gtfs_realtime_translators/factories/factories.py index 72ebb3c..85f55e4 100644 --- a/gtfs_realtime_translators/factories/factories.py +++ b/gtfs_realtime_translators/factories/factories.py @@ -72,7 +72,7 @@ def create(*args, **kwargs): departure=departure, stop_id=stop_id) - vehicle_descriptor = gtfs_realtime.TripUpdate.VehicleDescriptor(run_number=run_number) + vehicle_descriptor = gtfs_realtime.VehicleDescriptor(run_number=run_number) if track: stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].track = track From 4d29f4121dbc5124d1e95de687776c4b29928171 Mon Sep 17 00:00:00 2001 From: Prafullsh Date: Thu, 14 Nov 2024 20:02:42 +0530 Subject: [PATCH 4/7] change vehicle descriptor details --- gtfs_realtime_translators/bindings/intersection.proto | 2 +- gtfs_realtime_translators/bindings/intersection_pb2.py | 6 +++--- gtfs_realtime_translators/factories/factories.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/gtfs_realtime_translators/bindings/intersection.proto b/gtfs_realtime_translators/bindings/intersection.proto index db2d32f..a078b6b 100644 --- a/gtfs_realtime_translators/bindings/intersection.proto +++ b/gtfs_realtime_translators/bindings/intersection.proto @@ -36,5 +36,5 @@ message IntersectionVehicleDescriptor { } extend transit_realtime.VehicleDescriptor { - optional IntersectionVehicleDescriptor intersection_vehicle_descriptor = 1988; + optional IntersectionVehicleDescriptor intersection_vehicle_descriptor = 1987; } diff --git a/gtfs_realtime_translators/bindings/intersection_pb2.py b/gtfs_realtime_translators/bindings/intersection_pb2.py index 6b14047..0fbddcd 100644 --- a/gtfs_realtime_translators/bindings/intersection_pb2.py +++ b/gtfs_realtime_translators/bindings/intersection_pb2.py @@ -21,7 +21,7 @@ package='', syntax='proto2', serialized_options=None, - serialized_pb=_b('\n\x12intersection.proto\x1a\x13gtfs-realtime.proto\"\xfe\x01\n\x16IntersectionTripUpdate\x12\x10\n\x08headsign\x18\x01 \x01(\t\x12\x18\n\x10route_short_name\x18\x02 \x01(\t\x12\x17\n\x0froute_long_name\x18\x03 \x01(\t\x12\x13\n\x0broute_color\x18\x04 \x01(\t\x12\x18\n\x10route_text_color\x18\x05 \x01(\t\x12\x10\n\x08\x62lock_id\x18\x06 \x01(\t\x12\x17\n\x0f\x61gency_timezone\x18\x07 \x01(\t\x12\x15\n\rcustom_status\x18\x08 \x01(\t\x12\x1a\n\x12scheduled_interval\x18\t \x01(\x05\x12\x12\n\nroute_icon\x18\n \x01(\t\"\xce\x01\n\x1aIntersectionStopTimeUpdate\x12\r\n\x05track\x18\x01 \x01(\t\x12\x45\n\x11scheduled_arrival\x18\x02 \x01(\x0b\x32*.transit_realtime.TripUpdate.StopTimeEvent\x12G\n\x13scheduled_departure\x18\x03 \x01(\x0b\x32*.transit_realtime.TripUpdate.StopTimeEvent\x12\x11\n\tstop_name\x18\x04 \x01(\t\"3\n\x1dIntersectionVehicleDescriptor\x12\x12\n\nrun_number\x18\x01 \x01(\r:X\n\x18intersection_trip_update\x12\x1c.transit_realtime.TripUpdate\x18\xc3\x0f \x01(\x0b\x32\x17.IntersectionTripUpdate:p\n\x1dintersection_stop_time_update\x12+.transit_realtime.TripUpdate.StopTimeUpdate\x18\xc3\x0f \x01(\x0b\x32\x1b.IntersectionStopTimeUpdate:m\n\x1fintersection_vehicle_descriptor\x12#.transit_realtime.VehicleDescriptor\x18\xc4\x0f \x01(\x0b\x32\x1e.IntersectionVehicleDescriptor') + serialized_pb=_b('\n\x12intersection.proto\x1a\x13gtfs-realtime.proto\"\xfe\x01\n\x16IntersectionTripUpdate\x12\x10\n\x08headsign\x18\x01 \x01(\t\x12\x18\n\x10route_short_name\x18\x02 \x01(\t\x12\x17\n\x0froute_long_name\x18\x03 \x01(\t\x12\x13\n\x0broute_color\x18\x04 \x01(\t\x12\x18\n\x10route_text_color\x18\x05 \x01(\t\x12\x10\n\x08\x62lock_id\x18\x06 \x01(\t\x12\x17\n\x0f\x61gency_timezone\x18\x07 \x01(\t\x12\x15\n\rcustom_status\x18\x08 \x01(\t\x12\x1a\n\x12scheduled_interval\x18\t \x01(\x05\x12\x12\n\nroute_icon\x18\n \x01(\t\"\xce\x01\n\x1aIntersectionStopTimeUpdate\x12\r\n\x05track\x18\x01 \x01(\t\x12\x45\n\x11scheduled_arrival\x18\x02 \x01(\x0b\x32*.transit_realtime.TripUpdate.StopTimeEvent\x12G\n\x13scheduled_departure\x18\x03 \x01(\x0b\x32*.transit_realtime.TripUpdate.StopTimeEvent\x12\x11\n\tstop_name\x18\x04 \x01(\t\"3\n\x1dIntersectionVehicleDescriptor\x12\x12\n\nrun_number\x18\x01 \x01(\r:X\n\x18intersection_trip_update\x12\x1c.transit_realtime.TripUpdate\x18\xc3\x0f \x01(\x0b\x32\x17.IntersectionTripUpdate:p\n\x1dintersection_stop_time_update\x12+.transit_realtime.TripUpdate.StopTimeUpdate\x18\xc3\x0f \x01(\x0b\x32\x1b.IntersectionStopTimeUpdate:m\n\x1fintersection_vehicle_descriptor\x12#.transit_realtime.VehicleDescriptor\x18\xc3\x0f \x01(\x0b\x32\x1e.IntersectionVehicleDescriptor') , dependencies=[gtfs__realtime__pb2.DESCRIPTOR,]) @@ -42,10 +42,10 @@ message_type=None, enum_type=None, containing_type=None, is_extension=True, extension_scope=None, serialized_options=None, file=DESCRIPTOR) -INTERSECTION_VEHICLE_DESCRIPTOR_FIELD_NUMBER = 1988 +INTERSECTION_VEHICLE_DESCRIPTOR_FIELD_NUMBER = 1987 intersection_vehicle_descriptor = _descriptor.FieldDescriptor( name='intersection_vehicle_descriptor', full_name='intersection_vehicle_descriptor', index=2, - number=1988, type=11, cpp_type=10, label=1, + number=1987, type=11, cpp_type=10, label=1, has_default_value=False, default_value=None, message_type=None, enum_type=None, containing_type=None, is_extension=True, extension_scope=None, diff --git a/gtfs_realtime_translators/factories/factories.py b/gtfs_realtime_translators/factories/factories.py index 85f55e4..600edfa 100644 --- a/gtfs_realtime_translators/factories/factories.py +++ b/gtfs_realtime_translators/factories/factories.py @@ -72,7 +72,7 @@ def create(*args, **kwargs): departure=departure, stop_id=stop_id) - vehicle_descriptor = gtfs_realtime.VehicleDescriptor(run_number=run_number) + vehicle_descriptor = gtfs_realtime.VehicleDescriptor(id=run_number) if track: stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].track = track @@ -87,7 +87,7 @@ def create(*args, **kwargs): trip_update = gtfs_realtime.TripUpdate(trip=trip_descriptor, stop_time_update=[stop_time_update], - run_number=run_number) + vehicle=run_number) if headsign: trip_update.Extensions[intersection_gtfs_realtime.intersection_trip_update].headsign = headsign From 123e4d3e8e2d3a6cc1c673444412bd1c0514a583 Mon Sep 17 00:00:00 2001 From: Prafullsh Date: Fri, 15 Nov 2024 19:01:06 +0530 Subject: [PATCH 5/7] update value for trip update --- gtfs_realtime_translators/factories/factories.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gtfs_realtime_translators/factories/factories.py b/gtfs_realtime_translators/factories/factories.py index 600edfa..2432fad 100644 --- a/gtfs_realtime_translators/factories/factories.py +++ b/gtfs_realtime_translators/factories/factories.py @@ -82,12 +82,12 @@ def create(*args, **kwargs): stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].scheduled_departure.time = scheduled_departure if stop_name: stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].stop_name = stop_name - if run_number: - vehicle_descriptor.Extensions[intersection_gtfs_realtime.intersection_vehicle_descriptor].run_number = run_number + # if run_number: + # vehicle_descriptor.Extensions[intersection_gtfs_realtime.intersection_vehicle_descriptor].run_number = run_number trip_update = gtfs_realtime.TripUpdate(trip=trip_descriptor, stop_time_update=[stop_time_update], - vehicle=run_number) + vehicle=vehicle_descriptor) if headsign: trip_update.Extensions[intersection_gtfs_realtime.intersection_trip_update].headsign = headsign From 1fef0d5611ffb55bdc23dcf7aa9a24d5a9fd9fc0 Mon Sep 17 00:00:00 2001 From: Prafullsh Date: Tue, 19 Nov 2024 02:53:26 +0530 Subject: [PATCH 6/7] updated value --- gtfs_realtime_translators/factories/factories.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gtfs_realtime_translators/factories/factories.py b/gtfs_realtime_translators/factories/factories.py index 2432fad..b251f68 100644 --- a/gtfs_realtime_translators/factories/factories.py +++ b/gtfs_realtime_translators/factories/factories.py @@ -72,7 +72,7 @@ def create(*args, **kwargs): departure=departure, stop_id=stop_id) - vehicle_descriptor = gtfs_realtime.VehicleDescriptor(id=run_number) + vehicle_descriptor = gtfs_realtime.VehicleDescriptor(run_number=run_number) if track: stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].track = track From 9530fb5dd978e70d3f93f40432825abe4c1d9836 Mon Sep 17 00:00:00 2001 From: Prafullsh Date: Tue, 19 Nov 2024 02:56:47 +0530 Subject: [PATCH 7/7] add run number check --- gtfs_realtime_translators/factories/factories.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gtfs_realtime_translators/factories/factories.py b/gtfs_realtime_translators/factories/factories.py index b251f68..27eea28 100644 --- a/gtfs_realtime_translators/factories/factories.py +++ b/gtfs_realtime_translators/factories/factories.py @@ -72,7 +72,7 @@ def create(*args, **kwargs): departure=departure, stop_id=stop_id) - vehicle_descriptor = gtfs_realtime.VehicleDescriptor(run_number=run_number) + vehicle_descriptor = gtfs_realtime.VehicleDescriptor() if track: stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].track = track @@ -82,8 +82,8 @@ def create(*args, **kwargs): stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].scheduled_departure.time = scheduled_departure if stop_name: stop_time_update.Extensions[intersection_gtfs_realtime.intersection_stop_time_update].stop_name = stop_name - # if run_number: - # vehicle_descriptor.Extensions[intersection_gtfs_realtime.intersection_vehicle_descriptor].run_number = run_number + if run_number: + vehicle_descriptor.Extensions[intersection_gtfs_realtime.intersection_vehicle_descriptor].run_number = run_number trip_update = gtfs_realtime.TripUpdate(trip=trip_descriptor, stop_time_update=[stop_time_update],