A small helper script for quickly parsing agency GTFS-RT feeds for debugging

gtfs_rt_proto_parser.py

"""
A small helper script for quickly parsing agency GTFS-RT feeds for debugging.
Usage:
    $ service_venv python3 parse_gtfs_rt_feed.py --help
"""
import argparse

import requests
from google.transit import gtfs_realtime_pb2


def main(args):
    feed = gtfs_realtime_pb2.FeedMessage()
    response = requests.get(args.feed_url)
    feed.ParseFromString(response.content)
    print(feed)


def parse_cmd_line():
    parser = argparse.ArgumentParser(description='Print parsed GTFS-RT protobuf updates from agency feeds')

    parser.add_argument(
        'feed_url',
        action='store',
        metavar='FEED_URL',
        help='The agency GTFS-RT feed url to query',
    )

    return parser.parse_args()


if __name__ == '__main__':
    args = parse_cmd_line()
    main(args)