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 = """\n""" output += station.xml(services=True) return output