102 lines
3.0 KiB
Python
102 lines
3.0 KiB
Python
import os
|
|
from datetime import datetime
|
|
|
|
import pandas as pd
|
|
from flask_apscheduler import APScheduler
|
|
from dotenv import load_dotenv
|
|
from flask import Flask, jsonify, render_template, request
|
|
|
|
from mta_manager import MTA, Feed, Route
|
|
|
|
load_dotenv()
|
|
|
|
app = Flask(__name__)
|
|
app.secret_key = "SuperSecretDontEvenTryToGuessMeGGEZNoRe"
|
|
app._static_folder = os.path.abspath("templates/static/")
|
|
|
|
scheduler = APScheduler()
|
|
scheduler.init_app(app)
|
|
|
|
|
|
stops = pd.read_csv("stops.txt")
|
|
start_time = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
|
|
|
|
ROUTES = [Route.A, Route.C, Route.E, Route.N1, Route.N2, Route.N3]
|
|
STATION_STOP_IDs = ["127S", "127N", "A27N", "A27S"]
|
|
|
|
|
|
def link_to_station(data) -> {}:
|
|
linked_data = {}
|
|
for key, value in data.items():
|
|
stop_name = stops.loc[stops["stop_id"] == key]
|
|
stop_name = stop_name["stop_name"].values[0]
|
|
if stop_name not in linked_data:
|
|
linked_data[stop_name] = {}
|
|
if "N" in key:
|
|
linked_data[stop_name]["North"] = value
|
|
elif "S" in key:
|
|
linked_data[stop_name]["South"] = value
|
|
return linked_data
|
|
|
|
|
|
@app.route("/", methods=["GET"])
|
|
def index():
|
|
# TODO: Shove this into a sqlite database
|
|
station_names = sorted(list(set(stops["stop_name"].to_list())))
|
|
return render_template(
|
|
"layouts/index.html",
|
|
station_names=station_names,
|
|
station_1="42 St-Port Authority Bus Terminal",
|
|
station_2="Times Sq-42 St"
|
|
)
|
|
|
|
|
|
@app.route("/start_time", methods=["GET"])
|
|
def get_start_time():
|
|
return start_time
|
|
|
|
|
|
@app.route("/mta_data", methods=["POST"])
|
|
async def get_mta_data():
|
|
if len(mtaController.trains) == 0:
|
|
_ = update_trains()
|
|
arrival_by_station_and_route = {}
|
|
for stop_id in STATION_STOP_IDs:
|
|
arrival_by_station_and_route[stop_id] = {}
|
|
for route in ROUTES:
|
|
arrival_tiems = mtaController.get_arrival_times(route, stop_id)
|
|
if len(arrival_tiems) > 0:
|
|
arrival_by_station_and_route[stop_id][route.value] = arrival_tiems
|
|
return arrival_by_station_and_route
|
|
|
|
|
|
@app.route("/get_stop_id", methods=["POST"])
|
|
def get_stop_id():
|
|
stop_name = request.json["stop_name"]
|
|
stops.loc[stops["stop_name"] == stop_name]
|
|
return jsonify({"station_changed": True})
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
api_key = os.getenv('MTA_API_KEY', '')
|
|
|
|
old_data = None
|
|
last_updated = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
|
|
|
|
mtaController = MTA(
|
|
api_key,
|
|
feeds=[Feed.ACE, Feed.N1234567]
|
|
)
|
|
def update_trains():
|
|
app.logger.debug("UPDATING TRAINS")
|
|
mtaController.get_incoming_trains()
|
|
|
|
scheduler.add_job("train_updater", func=update_trains, trigger="interval", seconds=10)
|
|
scheduler.start()
|
|
|
|
debug = os.getenv("DEBUG", 'False').lower() in ('true', '1', 't')
|
|
app.run(host="localhost", debug=True, port=5000, use_reloader=False)
|
|
|
|
print("Exiting Main Thread")
|