Sat, 10 Jun 2017 21:03:26 +0300
Refaktorioitu aikatauluhaku
#!/usr/bin/env python3 from flask import Flask, render_template, abort, send_from_directory from datetime import datetime, date, time, timedelta from os import path, listdir from configparser import ConfigParser import locale from misc import * from busroute import supista_reitti app = Flask(__name__) # Varmista ettei järjestelmän kieliasetukset sotke muotoiluja locale.setlocale(locale.LC_ALL, locale.getdefaultlocale()) # Lataa käännökset class Translator: def __init__(self): self.languages = {} def load_language(self, file_path): language_name = path.splitext(path.basename(file_path))[0] ini = ConfigParser() ini.read(path.join(file_path)) self.languages[language_name] = ini def __call__(self, name, section, language = None): language = language or language_for_page() try: return self.languages[language][section][name] except KeyError: return name tr = Translator() for file in listdir('tr'): tr.load_language(path.join('tr', file)) def language_for_page(): from flask import request if request.args.get('kääntämätön') is not None: return None else: for language_name in tr.languages: if request.args.get(language_name) is not None: return language_name else: return 'fi' @app.route('/pysäkki/<tunniste>') def pysäkkiaikataulu(tunniste): from buses import pysäkit from math import ceil aikataulu = [] try: pysäkki = pysäkit[tunniste] except KeyError: abort(404) for (aika, pysähdys) in pysäkki.schedule(100): kyltti = supista_reitti(pysähdys.ajo.suppea_reitti(pysähdys)) kyltti = [tr(paikka, 'paikat') for paikka in kyltti] kylttiesitys = ' - '.join(kyltti) if len(kylttiesitys) > 25: k = ceil(len(kyltti) / 2) kylttiesitys = ' - '.join(kyltti[:k]) + '\n' + ' - '.join(kyltti[k:]) aikataulu.append({ 'aika': muotoile_aika(aika), 'linja': pysähdys.ajo.linja.viite, 'kyltti': kylttiesitys, 'ajovuoro': pysähdys.ajo.nimi, 'yö': yö(aika), }) return render_template( 'pysäkki.html', aikataulu = aikataulu, viite = tunniste, nimi = tr(pysäkki.nimi, 'pysäkit'), linkki_karttaan = pysäkki.linkki_karttaan, alue = pysäkki.alue, sijainti = pysäkki.sijainti, ) @app.route('/ajovuoro/<numero>') def ajoreitti(numero): from flask import request from buses import ajot try: ajovuoro = ajot[numero] except KeyError: abort(404) suppea = request.args.get('suppea') is not None reitti = [] suppea_reitti = [] käydyt_alueet = set() for pysähdys in ajovuoro.reitti: aika = datetime.combine(tänään(), time()) + pysähdys.saapumisaika muotoiltu_aika = muotoile_aika(aika) reitti.append({ 'aika': muotoiltu_aika, 'tunniste': pysähdys.pysäkki.tunniste, 'alue': tr(pysähdys.pysäkki.alue or '', 'paikat'), 'nimi': tr(pysähdys.pysäkki.nimi, 'pysäkit'), }) 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) kyltti = supista_reitti([k['alue'] for k in suppea_reitti], kokonainen = True) 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 = ' - '.join(tr(paikka, 'paikat') for paikka in kyltti), yö = yö(datetime.combine(tänään(), time()) + ajovuoro.reitti[-1].saapumisaika), ) @app.route('/static/<path:path>') def static_file(path): return send_from_directory(path.join('static', path)) if __name__ == '__main__': from argparse import ArgumentParser parser = ArgumentParser() parser.add_argument('-p', '--port', type = int, default = 5000) parser.add_argument('-d', '--debug', action = 'store_true') args = parser.parse_args() app.run(debug = args.debug, port = args.port)