Things

Hafas_Main_web.py

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