From cec2eca2350a1883b39b6a8645ed513bd8dc0da6 Mon Sep 17 00:00:00 2001
From: Klaus-Uwe Mitterer <git@klaus-uwe.me>
Date: Mon, 2 Oct 2017 21:02:07 +0200
Subject: [PATCH] Add Zugradar

---
 classes/__init__.py |  1 +
 main.py             | 16 ++++++++++++++++
 2 files changed, 17 insertions(+)

diff --git a/classes/__init__.py b/classes/__init__.py
index b29c676..e6226a3 100644
--- a/classes/__init__.py
+++ b/classes/__init__.py
@@ -1,3 +1,4 @@
 from classes.connection import Connection
 from classes.station import Station
 from classes.service import Service
+from classes.train import Train
diff --git a/main.py b/main.py
index 0a3f0c5..297cd20 100644
--- a/main.py
+++ b/main.py
@@ -5,6 +5,7 @@ import pytz
 import workers.conn
 import workers.val
 import workers.closest
+import workers.radar
 
 def application(env, re):
  if env["REQUEST_METHOD"] == "POST":
@@ -184,6 +185,21 @@ def application(env, re):
   yield output.encode()
   return
 
+ elif rtype.lower() in ["radar", "live"]:
+  trains = args["train"] if "train" in args else None
+  try:
+   output = workers.radar.worker(trains, json)
+  except Exception as e:
+   re("500 Internal Server Error", [])
+   yield "<h1>500 Internal Server Error</h1>".encode()
+   if "debug" in args:
+    yield e.encode()
+   return
+
+  re("200 OK", [("Content-Type", "application/json" if json else "text/xml")])
+  yield output.encode()
+  return
+
  else:
   re("400 Bad Request", [])
   yield "<h1>400 Bad Request</h1>".encode()