diff -r 22c22ff63e66 -r 48efa8ca14dd palvelu.py --- a/palvelu.py Mon Apr 17 22:56:39 2017 +0300 +++ b/palvelu.py Tue Apr 18 14:59:18 2017 +0300 @@ -33,18 +33,37 @@ @app.route('/ajovuoro/') def ajoreitti(numero): + from flask import request from bussit import ajot ajovuoro = ajot[numero] + suppea = request.args.get('suppea') is not None reitti = [] + suppea_reitti = [] + käydyt_alueet = set() for pysähdys in ajovuoro.reitti: aika = datetime.combine(date.today(), time()) + pysähdys.saapumisaika + muotoiltu_aika = muotoile_aika(aika, suhteellinen = False) reitti.append({ - 'aika': muotoile_aika(aika, suhteellinen = False), + 'aika': muotoiltu_aika, 'tunniste': pysähdys.pysäkki.tunniste, + 'alue': pysähdys.pysäkki.alue, 'nimi': pysähdys.pysäkki.nimi, }) - return render_template('ajovuoro.html', reitti = reitti, numero = numero, - linja = ajovuoro.linja.viite, selite = ajovuoro.kyltti) + alue = pysähdys.pysäkki.alue + if alue: + if alue not in käydyt_alueet: + suppea_reitti.append({ + 'aika': muotoiltu_aika, + 'alue': alue + }) + käydyt_alueet.add(alue) + sivu = suppea and 'ajovuoro-suppea.html' or 'ajovuoro.html' + return render_template(sivu, reitti = reitti, suppea_reitti = suppea_reitti, + numero = numero, linja = ajovuoro.linja.viite, selite = ajovuoro.kyltti) if __name__ == '__main__': - app.run(debug = True) \ No newline at end of file + from argparse import ArgumentParser + 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