Rsclub2_2/Hafas-Depature-Board-DB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Stupid Stuff

Browse Source
This commit is contained in:
Philipp Wagner 2023-10-09 17:30:29 +02:00
parent c851849851
commit b6e5d798ff
1 changed files with 111 additions and 111 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

222
Hafas_Main_web.py
View File

@ -1,111 +1,111 @@
import datetime import datetime
from typing import List from typing import List
from flask import Flask, render_template, request, make_response from flask import Flask, render_template, request, make_response
import json import json
# Part 1: Importieren der erforderlichen Module und Erstellen des Hafas-Clients # Part 1: Importieren der erforderlichen Module und Erstellen des Hafas-Clients
from pyhafas import HafasClient from pyhafas import HafasClient
from pyhafas.profile import DBProfile from pyhafas.profile import DBProfile
from pyhafas.types.fptf import Leg from pyhafas.types.fptf import Leg
app = Flask(__name__) app = Flask(__name__)
client = HafasClient(DBProfile()) client = HafasClient(DBProfile())
# Part 2: Suchen Sie nach der Haltestelle basierend auf der Benutzereingabe # Part 2: Suchen Sie nach der Haltestelle basierend auf der Benutzereingabe
def get_departures(station_name): def get_departures(station_name):
locations = client.locations(station_name) locations = client.locations(station_name)
if locations: if locations:
best_found_location = locations[0] best_found_location = locations[0]
return best_found_location return best_found_location
else: else:
return None return None
def station_board_leg_to_dict(leg): def station_board_leg_to_dict(leg):
delay_in_seconds = 0 delay_in_seconds = 0
if leg.delay is not None: if leg.delay is not None:
delay_in_seconds = leg.delay.total_seconds() // 60 delay_in_seconds = leg.delay.total_seconds() // 60
return { return {
"name": leg.name, "name": leg.name,
"direction": leg.direction, "direction": leg.direction,
"dateTime": leg.dateTime.strftime("%Y-%m-%d %H:%M:%S"), "dateTime": leg.dateTime.strftime("%Y-%m-%d %H:%M:%S"),
"delay": delay_in_seconds, "delay": delay_in_seconds,
"platform": leg.platform, "platform": leg.platform,
} }
# Part 3: Rufen Sie Abfahrtsinformationen für die gefundene Haltestelle ab # Part 3: Rufen Sie Abfahrtsinformationen für die gefundene Haltestelle ab
def get_departure_info(station_name): def get_departure_info(station_name):
best_found_location = get_departures(station_name) best_found_location = get_departures(station_name)
if best_found_location: if best_found_location:
departures: List[Leg] = client.departures( departures: List[Leg] = client.departures(
station=best_found_location, station=best_found_location,
date=datetime.datetime.now(), date=datetime.datetime.now(),
max_trips=15, max_trips=15,
products={ products={
'long_distance_express': True, 'long_distance_express': True,
'regional_express': True, 'regional_express': True,
'regional': True, 'regional': True,
'suburban': True, 'suburban': True,
'bus': True, 'bus': True,
'ferry': True, 'ferry': True,
'subway': True, 'subway': True,
'tram': True, 'tram': True,
'taxi': True 'taxi': True
} }
) )
return departures return departures
else: else:
return [] return []
@app.route('/', methods=['GET', 'POST']) @app.route('/', methods=['GET', 'POST'])
def index(): def index():
departures = [] departures = []
station_name = "" station_name = ""
# Get the station name from the cookie, if it exists # Get the station name from the cookie, if it exists
if request.cookies.get('station_name'): if request.cookies.get('station_name'):
station_name = request.cookies.get('station_name') station_name = request.cookies.get('station_name')
if request.method == 'POST': if request.method == 'POST':
station_name = request.form['station_name'] station_name = request.form['station_name']
departures = get_departure_info(station_name) departures = get_departure_info(station_name)
# Set the station name cookie # Set the station name cookie
response = make_response(render_template('index.html', departures=departures, station_name=station_name)) response = make_response(render_template('index.html', departures=departures, station_name=station_name))
response.set_cookie('station_name', station_name) response.set_cookie('station_name', station_name)
return response return response
return render_template('index.html', departures=departures, station_name=station_name) return render_template('index.html', departures=departures, station_name=station_name)
@app.route('/departures', methods=['GET']) @app.route('/departures', methods=['GET'])
def departures(): def departures():
""" """
Gibt die aktuellen Abfahrtsinformationen zurück. Gibt die aktuellen Abfahrtsinformationen zurück.
Returns: Returns:
JSON-Objekt mit den Abfahrtsinformationen. JSON-Objekt mit den Abfahrtsinformationen.
""" """
station_name=request.cookies.get('station_name') station_name=request.cookies.get('station_name')
departures = client.departures( departures = client.departures(
station=get_departures(station_name), station=get_departures(station_name),
date=datetime.datetime.now(), date=datetime.datetime.now(),
max_trips=15, max_trips=15,
products={ products={
'long_distance_express': True, 'long_distance_express': True,
'regional_express': True, 'regional_express': True,
'regional': True, 'regional': True,
'suburban': True, 'suburban': True,
'bus': True, 'bus': True,
'ferry': True, 'ferry': True,
'subway': True, 'subway': True,
'tram': True, 'tram': True,
'taxi': True 'taxi': True
} }
) )
# Konvertiere die Abfahrtsinformationen in ein JSON-Objekt # Konvertiere die Abfahrtsinformationen in ein JSON-Objekt
departures_json = json.dumps([station_board_leg_to_dict(leg) for leg in departures]) departures_json = json.dumps([station_board_leg_to_dict(leg) for leg in departures])
return departures_json return departures_json
if __name__ == '__main__': if __name__ == '__main__':
app.run(debug=True) app.run(host="0.0.0.0", debug=False)