changed Pandas to read from a csv with the current structure

Author: Ajsalemo

app.py

@@ -3,10 +3,9 @@
 import pandas as pd
 from flask import Flask, jsonify
 
-from data import data
-
 app = Flask(__name__)
-number_stations = data.number_stations
+number_stations = pd.read_csv("./data/number_stations.csv")
+
 
 @app.route('/')
 def index():
@@ -20,7 +19,7 @@ def index():
 def station_names():
     st_df = pd.DataFrame(number_stations)
     # Filter the DataFrame columns to retrieve all stations under 'Number Station name'
-    st_name = st_df['Number Station name']
+    st_name = st_df['Number']
     # If the DataFrame returns a response, convert the Series to a JSON string
     f = st_name.to_json()
     # Then convert the JSON string to a dict to properly format it when sending the response to the client
@@ -29,12 +28,10 @@ def station_names():
 
 @app.route('/filter_station_names/<name>')
 def filer_station_names(name):
-    print(name)
     if name is None:
         return jsonify({ 'error': 'Station name cannot be blank' })
-    fsn_df = pd.DataFrame(number_stations)
     # Pass the parameter from the URL and check if it matches against a known station name
-    f_name = fsn_df[fsn_df['Number Station name'] == name]
+    f_name = number_stations[number_stations['Name'] == name]
     # If the DataFrame returns empty, send an error back to the client
     if f_name.empty:
         return jsonify({ 'error': 'That station does not exist'})

data/number_stations.csv

@@ -0,0 +1,2 @@
+ID,Name,Country,Active Counterparts,Inactive counterparts
+E06,English Man 00000,Russia,"S06, M14, F01, F06","E17, G06, V06, V23"