source: misc/verkeersinformatie@ 422

#!/usr/bin/env python

#

# File informatie van NU.nl op de CLI

# 

# Licence: BSDLike - http://rickvanderzwet.nl/LICENSE

# Rick van der Zwet <info@rickvanderzwet.nl>

from bs4 import BeautifulSoup

import re

import sys

import urllib2



rit = { 'werk' : ['N11', 'A44', 'A4', ] }



def usage():

  print "Gebruik: %s [rit|weg] <naam> [<naam2> ...]" % sys.argv[0]

  exit(128)



# Zoek uit welke wegen we moeten zoeken

wegen_filter = []

if len(sys.argv) > 1:

  if len(sys.argv) == 2:

    usage()

  elif sys.argv[1] == 'rit':

    for naam in sys.argv[2:]:

      if not rit.has_key(naam):

        print "Geen rit '{0}' bekend (opties: {1})".format(naam, rit.keys())

        exit(128)

      wegen_filter.extend(rit[naam])

  elif sys.argv[1] == 'weg':

    wegen_filter = sys.argv[2:]

  else:

    usage()

wegen_filter = sorted(set(wegen_filter))



# Vind de wegen bij NU.nl

FILE_URL = 'http://www.nu.nl/verkeer/index.html'

soup = BeautifulSoup(urllib2.urlopen(FILE_URL))

middle = soup.find('div', { 'class' : 'page_content' }) 

file_len = 0

for f in middle.find_all('p', { 'style' : 'clear: both' }):

  text = re.sub('[ ]+',' ', f.get_text().strip().replace('\n', '-'))

  if len(list(f.children)) != 8:

    print "# {0}\n".format(text)

    continue

  file_len += 1

  if wegen_filter and not any(['file: {0} '.format(weg.lower()) in text.lower() for weg in wegen_filter]):

    continue

  print text



print ""

if wegen_filter: print "# Gefiltered op wegen: {0}".format(", ".join(wegen_filter))

print "# Data van: {0}".format(FILE_URL)

print "# Totaal {0} files".format(file_len)