sl.py

# -*- coding: UTF-8 -*-

import httplib
import json

class SlData:

	host = 'api.sl.se'
	stationUrl = '/api2/typeahead.json?key=XXX&searchstring='
	realtimeUrl = '/api2/realtimedeparturesV4.json?key=XXX&TimeWindow=60&siteid='


	@staticmethod
	def findStation(stationSearch):
		
		connection = httplib.HTTPConnection(SlData.host)
		connection.request('GET', SlData.stationUrl + stationSearch)
		response = connection.getresponse()
		stationData = json.loads(response.read(), 'utf-8')
		connection.close()

		if len(stationData['ResponseData']) == 0:
		    raise InputError('Hittade ingen hållplats')

		return stationData['ResponseData'][0]

	@staticmethod
	def nextBusesFromStation(station, direction):
		return SlData.nextBusesFromSiteId(station['SiteId'], direction)

	@staticmethod
	def nextBusesFromSiteId(siteId, direction):

		connection = httplib.HTTPConnection(SlData.host)
		connection.request('GET', SlData.realtimeUrl + siteId)
		response = connection.getresponse()
		realtimeData = json.loads(response.read(), 'utf-8')
		connection.close()

		return filter(lambda bus: bus['JourneyDirection'] == direction, realtimeData['ResponseData']['Buses'])