Mercurial > bus-service / file diff

--- a/palvelu.py	Tue Apr 18 14:59:18 2017 +0300
+++ b/palvelu.py	Sun Apr 23 11:01:51 2017 +0300
@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-from flask import Flask, render_template
+from flask import Flask, render_template, abort
 from datetime import datetime, date, time, timedelta
 import locale
 
@@ -21,7 +21,10 @@
 def pysäkkiaikataulu(tunniste):
 	from bussit import pysäkit
 	aikataulu = []
-	pysäkki = pysäkit[tunniste]
+	try:
+		pysäkki = pysäkit[tunniste]
+	except KeyError:
+		abort(404)
 	for (aika, pysähdys) in pysäkki.aikataulu(100):
 		aikataulu.append({
 			'aika': muotoile_aika(aika),
@@ -35,7 +38,10 @@
 def ajoreitti(numero):
 	from flask import request
 	from bussit import ajot
-	ajovuoro = ajot[numero]
+	try:
+		ajovuoro = ajot[numero]
+	except KeyError:
+		abort(404)
 	suppea = request.args.get('suppea') is not None
 	reitti = []
 	suppea_reitti = []
@@ -66,4 +72,4 @@
 	parser = ArgumentParser()
 	parser.add_argument('-p', '--port', type = int, default = 5000)
 	args = parser.parse_args()
-	app.run(debug = True, port = args.port)
\ No newline at end of file
+	app.run(debug = True, port = args.port)