From ccc95777166700b7da60ee9d8d3d4d193da13336 Mon Sep 17 00:00:00 2001
From: JuliusFreudenberger <jfgithub@coole-files.de>
Date: Fri, 11 Oct 2019 22:40:28 +0200
Subject: [PATCH] Added first functional code

ability to insert a station_id via inline message; ability to display
departures for specified station_name or station_id in direct chat
---
 bot.py | 74 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 bot.py

diff --git a/bot.py b/bot.py
new file mode 100644
index 0000000..84c3bae
--- /dev/null
+++ b/bot.py
@@ -0,0 +1,74 @@
+import logging
+import requests
+from telegram import InlineQueryResultArticle, InputTextMessageContent
+from telegram.ext import Updater, CommandHandler, InlineQueryHandler
+
+token_file = open("token", "r")
+updater = Updater(token=token_file.read(), use_context=True)
+token_file.close()
+
+dispatcher = updater.dispatcher
+
+logging.basicConfig(format='%(acstime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
+
+
+def start(update):
+    update.message.reply_text("Hello")
+
+
+def search_station(query):
+    request = requests.get("https://efa-api.asw.io/api/v1/station/?search=" + query)
+    if request.status_code != 200:
+        return -1
+    else:
+        return request.json()[0]['stationId']
+
+
+def get_vvs_departures(update, context):
+    if len(context.args) == 0:
+        update.message.reply_text("No argument specified")
+        return
+    if context.args[0].isdigit():
+        station_id = context.args[0]
+    else:
+        station_id = search_station(' '.join(context.args))
+    request = requests.get("https://efa-api.asw.io/api/v1/station/" + station_id + "/departures")
+    if request.status_code != 200:
+        update.message.reply_text("Error with server communication")
+    update.message.reply_text("Next departures:")
+    for station in request.json()[:4]:
+        update.message.reply_text(
+            "Line " + station['number'] + " to \"" + station['direction'] + "\" at "
+            + station['departureTime']['hour'].zfill(2) + ':' + station['departureTime']['minute'].zfill(2)
+            + " +" + str(station['delay']))
+
+
+def inline_station_search(update, context):
+    query = update.inline_query.query
+    if len(query) < 5 or not query:
+        return
+    results = list()
+    for station in get_stations_by_search_query(query):
+        results.append(
+            InlineQueryResultArticle(
+                id=station['stationId'],
+                title=station['name'],
+                input_message_content=InputTextMessageContent("/vvs " + station['stationId'])
+            )
+        )
+    context.bot.answer_inline_query(update.inline_query.id, results)
+
+
+def get_stations_by_search_query(name):
+    request = requests.get("https://efa-api.asw.io/api/v1/station/?search=" + name)
+    return request.json()
+
+
+inline_station_search_handler = InlineQueryHandler(inline_station_search)
+
+dispatcher.add_handler(inline_station_search_handler)
+
+updater.dispatcher.add_handler(CommandHandler('vvs', get_vvs_departures))
+
+updater.start_polling()
+updater.idle()