Adding reqs

server.py

@@ -3,6 +3,7 @@ import threading
 from flask import Flask, jsonify, render_template, request
 from mta_manager import MTA
 from pprint import pprint
+import pandas as pd
 
 app = Flask(__name__)
 app.secret_key = "SuperSecretDontEvenTryToGuessMeGGEZNoRe"
@@ -11,10 +12,15 @@ app._static_folder = os.path.abspath("templates/static/")
 
 subway_data = {}
 
+stops = pd.read_csv("stops.txt")
+
+
 @app.route("/", methods=["GET"])
 def index():
-    title = "Create the input image"
-    return render_template("layouts/index.html", title=title)
+    # TODO: Shove this into a sqlite database
+    station_names = sorted(list(set(stops["stop_name"].to_list())))
+    print(station_names)
+    return render_template("layouts/index.html", station_names=station_names)
 
 
 @app.route("/mta_data", methods=["POST"])
@@ -24,41 +30,56 @@ def get_mta_data():
         subway_data
     )
 
+
 @app.route("/stops", methods=["GET"])
 def get_routes():
     return jsonify()
 
+@app.route("/get_stop_id/<stop_name>", methods=["GET"])
+def get_stop_id(stop_name):
+    print(stop_name)
+    rows = stops.loc[stops["stop_name"] == stop_name]
+    print(rows)
+    return str(rows)
+
+
+
+def ack():
+    print('message was received!')
+
+
+class threadWrapper(threading.Thread):
+    def __init__(self, run, controller):
+        threading.Thread.__init__(self)
+        self.run = run
+        self.controller = controller
+
+    def run(self):
+        self.run()
+
+
+async def mta_callback(routes):
+    global subway_data
+    # TODO: Do away with this and throw it into websockets
+    subway_data = mtaController.convert_routes_to_station_first(routes)
+
+
+def start_mta():
+    mtaController.add_callback(mta_callback)
+    mtaController.start_updates()
+
+
 if __name__ == "__main__":
     api_key = os.getenv('MTA_API_KEY', '')
     mtaController = MTA(
+        # TODO: Update to only work with station names - need to be able to transfer the station names to train lines -
+        #  maybe with polling the station ids to see what train lines come up?
         api_key,
         ["A", "C", "E", "1", "2", "3"],
         ["127S", "127N", "A27N", "A27S"]
     )
-
-
-    async def mta_callback(routes):
-        global subway_data
-        # TODO: Do away with this and throw it into websockets
-        subway_data = mtaController.convert_routes_to_station_first(routes)
-
-
-    class threadWrapper(threading.Thread):
-        def __init__(self, run):
-            threading.Thread.__init__(self)
-            self.run = run
-
-        def run(self):
-            self.run()
-
-
-    def start_mta():
-        mtaController.add_callback(mta_callback)
-        mtaController.start_updates()
-
-
     threadLock = threading.Lock()
-    threads = [threadWrapper(start_mta)]
+    threads = [threadWrapper(start_mta, mtaController)]
 
     for t in threads:
         t.start()