from flask import Flask
from flask_restful import Resource, Api, reqparse
import pandas as pd
import ast
import kiss
import glob
app = Flask(__name__)
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

# Read some log files
list_stacked = pd.DataFrame()
file_list = glob.glob("logs/*.log")
print(file_list)
for file in file_list:
    file1 = pd.read_csv(file)
    list_stacked = pd.concat([list_stacked, file1])

# TODO get rid of NaN in JSON data? Either blank them or wrap in quotes.
# https://jsoneditoronline.org
# SyntaxError: JSON.parse: unexpected character at line 8818 column 20 of the JSON data
#             "104": NaN,

# TODO do I need to rearrange the data to a different format? I want all the
# data for one packet (one row) together.

class Packets(Resource):
    def get(self):
        data = list_stacked
        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
api.add_resource(Packets, '/packets')  # and '/locations' is our entry point for Locations

if __name__ == '__main__':
    app.run(debug=True)  # run our Flask app