Mercurial > bus-service / file diff

--- 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/<int:numero>')
 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