From fdf9392d38746ff2a2ece88d079a35f241d44c49 Mon Sep 17 00:00:00 2001
From: RunasSudo <runassudo@yingtongli.me>
Date: Tue, 1 Jan 2019 19:00:21 +1100
Subject: [PATCH] Fix weird Flemington Racecourse/South Kensington bugs in PTV
 API

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

diff --git a/virtual_metro/__init__.py b/virtual_metro/__init__.py
index e35dd6b..4179d90 100755
--- a/virtual_metro/__init__.py
+++ b/virtual_metro/__init__.py
@@ -39,6 +39,15 @@ timezone = pytz.timezone('Australia/Melbourne')
 def index():
 	return flask.render_template('index.html')
 
+def stop_to_name(stop, route_id):
+	name = stop['stop_name']
+	if name.endswith(' Station'):
+		name = name[:-8]
+	if name == 'Flemington Racecourse' and route_id and route_id != 16 and route_id != 17 and route_id != 1482:
+		# PTV bug??
+		name = 'South Kensington'
+	return name
+
 route_stops = {} # Cache lookup
 def parse_departure(departure, departures, timenow):
 	result = {}