69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
from flask import Flask
|
|
from flask_restful import Resource, Api, reqparse
|
|
from datetime import date, timedelta
|
|
import configparser
|
|
import pandas as pd
|
|
import numpy as np
|
|
import ast
|
|
import kiss
|
|
import glob
|
|
app = Flask(__name__)
|
|
api = Api(app)
|
|
|
|
def read_config():
|
|
config = configparser.ConfigParser()
|
|
config.read('config.ini')
|
|
return config
|
|
|
|
def read_logs(log_folder):
|
|
# Read some log files
|
|
# Zulu time, so let's look at tomorrow, today, and yesterday.
|
|
# TODO Load new files into a database on a schedule?
|
|
today = date.today()
|
|
yesterday = today - timedelta(days = 1)
|
|
tomorrow = today + timedelta(days = 1)
|
|
file_list = glob.glob(log_folder+str(yesterday)+"*") + \
|
|
glob.glob(log_folder+str(today)+"*") + \
|
|
glob.glob(log_folder+str(tomorrow)+"*")
|
|
list_stacked = pd.DataFrame()
|
|
for file in file_list:
|
|
file1 = pd.read_csv(file)
|
|
list_stacked = pd.concat([list_stacked, file1])
|
|
# TODO Can we do this without numpy?
|
|
list_stacked.replace(np.nan, 0, inplace=True)
|
|
#print(list_stacked.head())
|
|
return(list_stacked)
|
|
|
|
class Users(Resource):
|
|
def get(self):
|
|
data = pd.read_csv('users.csv') # read CSV
|
|
data = data.to_dict(orient = 'records') # 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(orient = 'records') # convert dataframe to dictionary
|
|
return {'data': data}, 200 # return data and 200 OK code
|
|
|
|
class Packets(Resource):
|
|
def get(self):
|
|
data = read_logs(log_folder) # re-reads the log files every time
|
|
data = data.to_dict(orient = 'records') # convert dataframe to dictionary
|
|
#data = data.to_json(orient='records')
|
|
return {'data': data}, 200 # return data and 200 OK code
|
|
|
|
# Read config
|
|
config = read_config()
|
|
log_folder = config['Settings']['log_folder']
|
|
# Load logs first (just to check for errors before page loads)
|
|
data = read_logs(log_folder)
|
|
|
|
|
|
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
|