From 78c337a8ebddb366a0428f8c11ee91d0e066747d Mon Sep 17 00:00:00 2001
From: JuliusFreudenberger <jfgithub@coole-files.de>
Date: Tue, 15 Oct 2019 19:42:35 +0200
Subject: [PATCH] Added object for query; added option to filter by destination

Parsing and getting arguments now is much easier with object
if user additionally sends "to {destination}" the departures are
filtered to match only this destination (case-insensitive)
---
 bot.py | 59 ++++++++++++++++++++++++++++++++++++++++------------------
 1 file changed, 41 insertions(+), 18 deletions(-)

diff --git a/bot.py b/bot.py
index e71bdba..1f39039 100644
--- a/bot.py
+++ b/bot.py
@@ -12,6 +12,31 @@ dispatcher = updater.dispatcher
 logging.basicConfig(format='%(acstime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
 
 
+class Query:
+    input = ""
+    station_id = -1
+    departure_count = 4
+    line_filter = -1
+    destination = ""
+
+    def __init__(self, args):
+        last_index_string = len(args)
+        if args[-1].isdigit():
+            self.departure_count = int(args[-1])
+            last_index_string -= 1
+        request_tmp = ' '.join(args[0:last_index_string]).strip()
+        separator_index = request_tmp.find("to")
+        if separator_index != -1:
+            self.input = request_tmp[:separator_index - 1].strip()
+            self.destination = request_tmp[separator_index + 2:].strip()
+        else:
+            self.input = request_tmp
+        if args[0].isdigit():
+            self.station_id = args[0].strip()
+        else:
+            self.station_id = search_station(request_tmp[:separator_index].strip())
+
+
 def start(update):
     update.message.reply_text("Hello")
 
@@ -25,27 +50,28 @@ def search_station(query):
 
 
 def get_vvs_departures(update, context):
-    number_departures = 4
-    last_index_string = len(context.args)
     if len(context.args) == 0:
         update.message.reply_text("No argument specified")
         return
-    if context.args[-1].isdigit():
-        number_departures = context.args[-1]
-        last_index_string -= 1
-    if context.args[0].isdigit():
-        station_id = context.args[0]
-    else:
-        station_id = search_station(' '.join(context.args[0:last_index_string]))
-    request = requests.get("https://efa-api.asw.io/api/v1/station/" + station_id + "/departures")
+    query = Query(context.args)
+    if query.station_id == -1:
+        update.message.reply_text("No station matching this name found!")
+        return
+    request = requests.get("https://efa-api.asw.io/api/v1/station/" + query.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()[:int(number_departures)]:
-        update.message.reply_text(
-            "Line " + station['number'] + " to \"" + station['direction'] + "\" at "
-            + station['departureTime']['hour'].zfill(2) + ':' + station['departureTime']['minute'].zfill(2)
-            + " +" + str(station['delay']))
+    printed_departures = 0
+    for station in request.json():
+        if station['direction'].casefold().find(query.destination) != -1 or station['direction'].find(
+                query.destination) != -1:
+            update.message.reply_text(
+                "Line " + station['number'] + " to \"" + station['direction'] + "\" at "
+                + station['departureTime']['hour'].zfill(2) + ':' + station['departureTime']['minute'].zfill(2)
+                + " +" + str(station['delay']))
+            printed_departures += 1
+        if printed_departures >= query.departure_count:
+            break
 
 
 def inline_station_search(update, context):
@@ -77,6 +103,3 @@ updater.dispatcher.add_handler(CommandHandler('vvs', get_vvs_departures))
 
 updater.start_polling()
 updater.idle()
-    if query.station_id == -1:
-        update.message.reply_text("No station matching this name found!")
-        return