nextbus_predictions.py

import urllib
import re


URLS = {
    '10 Inbound Wisconsin & Madera': 'http://webservices.nextbus.com/service/publicXMLFeed?command=predictions&a=sf-muni&r=10&s=6966',  # noqa E501
    '48 Inbound Wisconsin & 25th': 'http://webservices.nextbus.com/service/publicXMLFeed?command=predictions&a=sf-muni&r=48&s=3513',  # noqa E501
}

SECONDS_MINUTES_RE = re.compile('seconds="(\d+)" minutes="(\d+)"')


def format_prediction(seconds, minutes):
    return '{minutes}m{seconds}s'.format(
        minutes=minutes,
        seconds=int(seconds) % 60,
    )


def request_predictions():
    predictions = {}
    for stop, url in URLS.items():
        connection = urllib.urlopen(url)
        raw = connection.read()
        connection.close()
        rs = SECONDS_MINUTES_RE.findall(raw)
        predictions[stop] = [format_prediction(*tuple) for tuple in rs]
    return predictions


print(request_predictions())