#!/usr/bin/python

import xbmc, xbmcgui
import urllib,urllib2 , re
from string import split

try: Emulating = xbmcgui.Emulating
except: Emulating = False

ACTION_PREVIOUS_MENU = 10
ACTION_SELECT_ITEM = 7
ACTION_PARENT_DIR = 9

class MyClass(xbmcgui.Window):
  def __init__(self):
    if Emulating: xbmcgui.Window.__init__(self)
    self.addControl(xbmcgui.ControlImage(0,0,720,480, "background.png"))
    # Make all the buttons
    self.button0 = xbmcgui.ControlButton(70, 150, 120, 30, "Albertplatz")
    self.addControl(self.button0)
    self.button1 = xbmcgui.ControlButton(70, 200, 120, 30, "Bischofsweg")
    self.addControl(self.button1)
    self.button2 = xbmcgui.ControlButton(70, 250, 120, 30, "Bischofsplatz")
    self.addControl(self.button2)
    self.button3 = xbmcgui.ControlButton(70, 300, 120, 30, "Rudolf-Leonhard-Straße")
    self.addControl(self.button3)
    self.button4 = xbmcgui.ControlButton(70, 350, 120, 30, "Sachsenallee")
    self.addControl(self.button4)
    self.setFocus(self.button0)
    self.button0.controlDown(self.button1)
    self.button1.controlUp(self.button0)
    self.button1.controlDown(self.button2)
    self.button2.controlUp(self.button1)
    self.button2.controlDown(self.button3)
    self.button3.controlUp(self.button2)
    self.button3.controlDown(self.button4)
    self.button4.controlUp(self.button3)
    
    self.depList1 = xbmcgui.ControlList(220,135,260,350)
    self.addControl(self.depList1)
    self.depList2 = xbmcgui.ControlList(260,135,410,350)
    self.addControl(self.depList2)
    self.depList3 = xbmcgui.ControlList(410,135,420,350)
    self.addControl(self.depList3)

  def onAction(self, action):
   if action == ACTION_PREVIOUS_MENU:
     self.close()
   if action == ACTION_PARENT_DIR:
     self.removeControl(self.strAction)
  def onControl(self, control):
    if control == self.button0:
      self.loadFeed("Albertplatz")
    if control == self.button1:
      self.loadFeed("Bischofsweg")
    if control == self.button2:
      self.loadFeed("Bischofsplatz")
    if control == self.button3:
      self.loadFeed("Rudolf-Leonhard-Strasse")
    if control == self.button4:
      self.loadFeed("Sachsenallee")
  def loadFeed(self,haltestelle):
    self.depList1.reset()
    self.depList2.reset()
    self.depList3.reset()
    Base_URL = "http://widgets.vvo-online.de/abfahrtsmonitor/Abfahrten.do?ort=Dresden&hst="+haltestelle+"&vz=5&lim=8"
    WebSock = urllib.urlopen(Base_URL)
    WebHTML = WebSock.read()
    WebSock.close()
    s=[x[1:-1].split('","') for x in WebHTML.replace('"', '"')[2:-2].split('],[')]
    offset=210
    self.depList1.addItem("Linie")
    self.depList2.addItem("Richtung")
    self.depList3.addItem("in Min")
    for y in s:
     self.depList1.addItem(y[0])
     self.depList2.addItem(y[1].replace('ü','ü').replace('ß','ß'))
     self.depList3.addItem(y[2])
mydisplay = MyClass()
mydisplay.doModal()
del mydisplay