from bs4 import BeautifulSoup
import datetime
import pytz
import threading
import queue
import sys

import workers.val
from classes import *

def getStation(name):
 return list(workers.val.validateName(name))[0]

def getService(sid, url, dtime, q = None, eq = None):
 try:
  zuppa = BeautifulSoup(HTTPClient().get(url).text, "html5lib")

  name = zuppa.findAll("div", { "class": "block" })[0].text.strip().replace("(Zug-Nr. ", " - ").replace(")", "")
  ddate = zuppa.findAll("div", { "class": "block" })[1].text.strip()

  table = zuppa.find("table", { "class": "resultTable" })
  rows = table.findAll("tr")[1:]

  for row in rows:
   if len(row.findAll("td")) > 6 and row.findAll("td")[4].text.strip() == dtime:
    depst = getStation(row.findAll("td")[1].text.strip())
    currdep = row.findAll("td")[5].text.strip()
    currdep = dtime if currdep == "pünktlich" else currdep or None
    deppf = row.findAll("td")[-1].text.strip() or None

  dest = getStation(rows[-1].findAll("td")[1].text.strip())
  atime = rows[-1].findAll("td")[2].text.strip()
  curarr = rows[-1].findAll("td")[3].text.strip()
  curarr = atime if curarr == "pünktlich" else curarr or None
  arrpf = rows[-1].findAll("td")[-1].text.strip()

  deptime = datetime.datetime.strptime("%s %s" % (ddate, dtime), "%d.%m.%Y %H:%M")
  arrtime = datetime.datetime.strptime("%s %s" % (ddate, atime), "%d.%m.%Y %H:%M")

  if arrtime < deptime:
   arrtime += datetime.timedelta(days = 1)

  if q:
   q.put((sid, Service(name, depst, deptime, dest, arrtime, dest, deppf, currdep, arrpf, curarr)))
  return q

 except:
  if eq:
   eq.put(sys.exc_info())
  raise

def daRequest(station, count = 3, time = datetime.datetime.now(), mode = False, details = False):
 outdate = datetime.datetime.strftime(time, "%d.%m.%Y")
 outtime = datetime.datetime.strftime(time, "%H:%M")

 url = "http://fahrplan.oebb.at/bin/stboard.exe/dn?input=%s&boardType=%s&time=%s&productsFilter=1111111111111111&dateBegin=%s&dateEnd=&selectDate=&maxJourneys=%i&start=yes&dirInput=&sqView=2" % (station.extid if station.extid else station.name, "arr" if mode else "dep", outtime, outdate, count)

 source = HTTPClient().get(url).text

 if "traininfo.exe/dn/" not in source:
  raise ValueError("No services found.")

 juha = BeautifulSoup(source, "html5lib")
 
 services = []

 table = juha.find("table", {"class": "resultTable"})

 for row in table.findAll("tr")[1:-1]:
  if not len(row.findAll("td")) < 4:
   services += [(row.findAll("a")[0].get("href"), row.findAll("td")[0].text.strip())]
 
 threads = []
 eq = queue.Queue()
 q = queue.PriorityQueue()

 for i in range(len(services)):
  t = threading.Thread(target=getService, args=(i, services[i][0], services[i][1], q, eq), daemon = True)
  t.start()
  threads += [t]

 for t in threads:
  t.join()

 if not eq.empty():
  exc = eq.get()
  raise exc[1].with_traceback(exc[2])

 while not q.empty():
  station.addService(q.get()[1])

 return station

def worker(station, count = 30, time = datetime.datetime.now(pytz.timezone("Europe/Vienna")), mode = False, details = False, json = False):
 station = daRequest(getStation(station), count, time, mode, details)

 if json:
  output = station.json(services = True)
 else:
  output = """<?xml version="1.0" encoding="UTF-8"?>\n"""
  output += station.xml(services = True)

 return output