diff --git a/app.py b/app.py index 851dcd4..019f75d 100644 --- a/app.py +++ b/app.py @@ -3,4 +3,22 @@ from flask_restful import Resource, Api, reqparse import pandas as pd import ast app = Flask(__name__) -api = Api(app) \ No newline at end of file +api = Api(app) + +class Users(Resource): + def get(self): + data = pd.read_csv('users.csv') # read CSV + data = data.to_dict() # convert dataframe to dictionary + return {'data': data}, 200 # return data and 200 OK code + +class Locations(Resource): + def get(self): + data = pd.read_csv('locations.csv') # read CSV + data = data.to_dict() # convert dataframe to dictionary + return {'data': data}, 200 # return data and 200 OK code + +api.add_resource(Users, '/users') # '/users' is our entry point for Users +api.add_resource(Locations, '/locations') # and '/locations' is our entry point for Locations + +if __name__ == '__main__': + app.run(debug=True) # run our Flask app \ No newline at end of file