diff -r 48efa8ca14dd -r ca2df60215ff palvelu.py --- 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)