#!/usr/bin/python3

import database as db

import json

from flask import Flask, request

from flask_socketio import SocketIO

from flask_socketio import send, emit

from flask_cors import CORS, cross_origin

import os

import sys

import logging

PATH = None

pathToExeFromCurDir = sys.argv[0]

current_directory = os.getcwd()

if(pathToExeFromCurDir.startswith("/")):

PATH = pathToExeFromCurDir

elif(pathToExeFromCurDir.startswith(".")):

PATH = current_directory + pathToExeFromCurDir[1:]

else:

PATH = current_directory + "/" + pathToExeFromCurDir

PATH = PATH.rsplit("/", 1)[0] + "/"

logging.basicConfig(

filename=PATH+'application.log',

format='%(asctime)s.%(msecs)-3d:%(filename)s:%(funcName)s:%(levelname)s:%(lineno)d:%(message)s',

datefmt='%Y-%m-%d %H:%M:%S',

level=logging.DEBUG

)

app = Flask(__name__)

cors = CORS(app)

app.config['SECRET_KEY'] = 'secret!'

socketio = SocketIO(app, cors_allowed_origins="*")

print("Connecting to the database")

process = db.main()

process.setConnection()

print("Connection successful")

@socketio.on('connect')

def handle_my_custom_event():

send("Hello from flask server")

@socketio.on('query')

def process_query(query):

logging.warning("No result sent back from the database")

@socketio.on('validate_user')

def validate_user(credentials):

emit('response', str(json.dumps({"result": False})))

@socketio.on('message')

def handle_message(message):

logging.info(f"Message from {request.remote_addr}:{message}")

print(f"Message from {request.remote_addr}:{message}")

@socketio.on_error_default # handles all namespaces without an explicit error handler

def default_error_handler(e):

print("ERROR ENCOUNTERED IN SERVER:", e)

print("args:" + str(request.event))

@socketio.on('disconnect')

def handle_disconnect():

print(f"{request.remote_addr} disconnected.")

if __name__ == '__main__':

socketio.run(app, host='0.0.0.0')

else:

exit(0)