#!/usr/bin/env python2

from httplib2 import Http
from urllib import urlencode
from BeautifulSoup import BeautifulSoup
import simplejson
import codecs

h = Http()

stations = [x.split(';')[1] for x in open('stations-nl-2015-08.csv', 'r').read().split('\n')[1:-1]]

storage = {}

o = codecs.open('stationsvoorzieningen-2015-08.csv', 'w', 'utf-8')

for station in stations:
	resp, content = h.request("http://www.ns.nl/reizigers/reisinformatie/stationsvoorzieningen/?r220_r1_r1_r1:station=%s"%(station), "GET")

	soup = BeautifulSoup(content)
	voorzieningen = soup.find('div', id="voorzieningen")

	if voorzieningen is not None:
		storage[station] = set([unicode(x.text) for x in voorzieningen.findAll('li')])

	o.write('%s\t%s\n' % (station, ';'.join(storage[station])))
