NerdyRoom™

$ ./dvb-trips.py Helmholzstrasse Saxoniastrasse
Zeit der Abfrage: 22:17
ab 22:08 an 22:35 dauer 00:27 umsteigen 0
 > ab          Fußweg           Helmholtzstraße
 > ab 22:25    Stadtbus 63      Dresden Westendstraße
 > ab          Fußweg           Dresden Tharandter Straße
ab 22:19 an 22:48 dauer 00:29 umsteigen 1
 > ab          Fußweg           Helmholtzstraße
 > ab 22:31    Straßenbahn 3    Dresden Plauen Nöthnitzer Straße
 > ab 22:42    Straßenbahn 7    Dresden Hauptbahnhof
ab 22:31 an 22:53 dauer 00:22 umsteigen 1
 > ab          Fußweg           Helmholtzstraße
 > ab 22:39    Regionalbus 360  Dresden Mommsenstraße
 > ab 22:42    Stadtbus 61      Dresden Technische Universität (Fritz-Foerster-Platz)
 > ab          Fußweg           Dresden Tharandter Straße
ab 22:37 an 23:04 dauer 00:27 umsteigen 0
 > ab          Fußweg           Helmholtzstraße
 > ab 22:54    Stadtbus 63      Dresden Westendstraße
 > ab          Fußweg           Dresden Tharandter Straße

#! /usr/bin/env python
# coding=utf-8
 
import sys
import datetime
import time
import xml.dom.minidom
import urllib
 
def getTrips(dep_start, dep_target, dep_datetime):
    url_template = 'http://www.dvb.de/de/Fahrplan/Verbindungsauskunft/direkt.do?vaform[starttypeswitch_stop]=1&vaform[zieltypeswitch_stop]=1&vaform[startort]=Dresden&vaform[startname]=%s&vaform[zielort]=Dresden&vaform[zielname]=%s&vaform[zeittyp]=dep&vaform[datum]=%s&vaform[zeit]=%s'
 
    now_url = url_template % (dep_start, dep_target, dep_datetime[0], dep_datetime[1])
 
    string = urllib.urlopen(now_url).read().replace('=voe&uk_id=', '=voe&uk_id=')
    dom = xml.dom.minidom.parseString(string)
 
    def getText(nodelist, recurse=False):
        rc = []
        for node in nodelist:
            if node.nodeType == node.TEXT_NODE:
                rc.append(node.data)
            elif recurse:
                rc.append(getText(node.childNodes, recurse))
        return ''.join(rc)
 
    overview_table = None
    detail_from = None
 
    for h2 in dom.getElementsByTagName('h2'):
        text = getText(h2.childNodes)
        if text[:15] == 'Verbindungen - ' and ord(text[15]) == 220 and text[16:] == 'bersicht':
            overview_table = h2.nextSibling
            while overview_table.nodeType != overview_table.ELEMENT_NODE:
                overview_table = overview_table.nextSibling
            if overview_table.nodeName != 'table':
                overview_table = None
        elif text == 'Verbindungen - Detailansicht':
            detail_from = h2.nextSibling
            while detail_from.nodeType != detail_from.ELEMENT_NODE:
                detail_from = detail_from.nextSibling
            if detail_from.nodeName != 'form':
                detail_from = None
 
    if not overview_table or not detail_from:
        sys.exit(1)
 
    for tbody in overview_table.childNodes:
        if tbody.nodeType == tbody.ELEMENT_NODE and tbody.tagName == 'tbody':
            break
 
    trips = []
    for tr in tbody.childNodes:
        if tr.nodeType != tr.ELEMENT_NODE or tr.tagName != 'tr':
            continue
        tds = []
        for td in tr.childNodes:
            if td.nodeType != tr.ELEMENT_NODE or td.tagName != 'td':
                continue
            tds.append(td)
        if len(tds) != 6:
            continue
        #print tds
        trip = []
        for td in tds:
            #if td.firstChild.nodeType == td.ELEMENT_NODE and td.firstChild.tagName == 'a':
            #    td = td.firstChild
            trip.append(getText(td.childNodes, True))
        trips.append(trip)
 
    trip_number = 0
    for tbody in detail_from.getElementsByTagName('tbody'):
 
        route = []
        for tr in tbody.childNodes:
            if tr.nodeType != tr.ELEMENT_NODE or tr.tagName != 'tr':
                continue
 
            section = []
            for td in tr.childNodes:
                if td.nodeType != tr.ELEMENT_NODE or td.tagName != 'td':
                    continue
                section.append(td)
 
            if getText(section[1].childNodes) != 'ab':
                continue
            route.append((
                getText(section[0].childNodes, True),
                getText(section[4].firstChild.childNodes, True),
                getText(section[2].childNodes),
            ))
 
        if len(route):
            #print route
            trips[trip_number].append(route)
        trip_number += 1
 
    return trips
 
def getDatetimeArg(dt):
    return ('%s.%s.%s' % (dt.day, dt.month, dt.year),
            '%s.%s' % (dt.hour, dt.minute))
 
if __name__ == '__main__':
    if len(sys.argv) != 3:
        print 'Usage %s [start-haltestelle] [end-haltestelle]' % sys.argv[0]
        sys.exit(0)
 
    try:
        while True:
            dep_dt = datetime.datetime.now()
            print 'Zeit der Abfrage:', dep_dt.strftime('%H:%M')
            trips = getTrips(sys.argv[1], sys.argv[2], getDatetimeArg(dep_dt))
            time_to_sleep = None
            for trip in trips:
                start = time.strptime(trip[1], '%H:%M')
                next_dt = datetime.datetime.combine(dep_dt.date(), datetime.time(start.tm_hour, start.tm_min))
                print 'ab %s an %s dauer %s umsteigen %s' % (trip[1], trip[2], trip[3], trip[4])
                for step in trip[6]:
                    print ' > ab %-8s %-16s %s' % step
 
            time.sleep(60)
            print
 
    except KeyboardInterrupt:
        sys.exit(0)

$ ln -s /usr/bin/ssh-argv ~/bin/$HOST
$ $HOST

Host $FQDN $ALIAS
  HostName $FQDN