Skip to content

Commit fa36c51

Browse files
committed
Force uppercase route_ids
Closes #12
1 parent 7711077 commit fa36c51

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

app.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
"""
1111

1212
from mtapi.mtapi import Mtapi
13-
from flask import Flask, request, jsonify, render_template, abort
13+
from flask import Flask, request, jsonify, render_template, abort, redirect
1414
from flask.json import JSONEncoder
1515
from datetime import datetime
1616
from functools import wraps, reduce
@@ -102,6 +102,10 @@ def by_location():
102102
@app.route('/by-route/<route>', methods=['GET'])
103103
@cross_origin
104104
def by_route(route):
105+
106+
if route.islower():
107+
return redirect(request.host_url + 'by-route/' + route.upper(), code=301)
108+
105109
try:
106110
data = mta.get_by_route(route)
107111
return _make_envelope(data)

mtapi/mtapi.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def _update(self):
144144
continue
145145

146146
direction = trip.direction[0]
147-
route_id = trip.route_id
147+
route_id = trip.route_id.upper()
148148

149149
for update in entity.trip_update.stop_time_update:
150150
trip_stop = TripStop(update)
@@ -193,6 +193,8 @@ def get_routes(self):
193193
return self._routes.keys()
194194

195195
def get_by_route(self, route):
196+
route = route.upper()
197+
196198
if self.is_expired():
197199
self._update()
198200

0 commit comments

Comments
 (0)