From 47da751b1b1a196a2a64ba78a7c7a7f281d561c0 Mon Sep 17 00:00:00 2001
From: RunasSudo <runassudo@yingtongli.me>
Date: Tue, 1 Jan 2019 15:26:02 +1100
Subject: [PATCH] Fail gracefully if invalid query

---
 virtual_metro/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/virtual_metro/__init__.py b/virtual_metro/__init__.py
index bfc9bfb..c2ed591 100755
--- a/virtual_metro/__init__.py
+++ b/virtual_metro/__init__.py
@@ -112,6 +112,10 @@ def latest():
 	departures = do_request('/v3/departures/route_type/{}/stop/{}'.format(ROUTE_TYPE, flask.request.args['stop_id']), {'platform_numbers': flask.request.args['plat_id'], 'max_results': '5', 'expand': 'all'})
 	departures['departures'].sort(key=lambda x: x['scheduled_departure_utc'])
 	
+	if len(departures['departures']) == 0:
+		# Invalid stop ID, platform ID, no departures, etc.
+		return flask.jsonify(result)
+	
 	result['stop_name'] = departures['stops'][flask.request.args['stop_id']]['stop_name'].replace(' Station', '')
 	
 	# Next train