Newer
Older
# run.py
import os
Mitchell Moore
committed
import time
from app import create_app
Mitchell Moore
committed
from flask_socketio import SocketIO
Mitchell Moore
committed
from flask import flash
config_name = os.getenv('FLASK_CONFIG')
app = create_app(config_name)
Mitchell Moore
committed
app.config['SECRET_KEY'] = 'vnkdjnfjknfl1232#'
Mitchell Moore
committed
socketio = SocketIO(app)
Mitchell Moore
committed
def messageReceived(methods=['GET', 'POST']):
print('message was received!!!')
def check_dir(user, interval):
"""
:param user: (string) username to check for in DB.
:param interval: (int) Frequency to check in seconds.
:return: (boolean) if account has been registered.
"""
seconds = 0
while seconds < 600:
querystring = "_" + user + ".done"
for filename in os.listdir("flat_db/"):
if filename.endswith(querystring):
return True
time.sleep(interval)
seconds = seconds + interval
return False
@socketio.on('user connect')
def handle_my_custom_event(json, methods=['GET', 'POST']):
username = json["user"]
print(time.strftime("%m-%d-%Y_%H:%M:%S") + '\tUser ' + username + ' connected.')
Mitchell Moore
committed
@socketio.on('user data')
def ingest_data(json, methods=['GET', 'POST']):
print (time.strftime("%m-%d-%Y_%H:%M:%S") + '\tQueue request received: ' + str(json))
Mitchell Moore
committed
try:
fullname = json["fullname"]
reason = json["reason"]
username = json["username"]
time_stamp = time.strftime("%m-%d-%Y_%H:%M:%S")
directory = "flat_db/"
complete_file_name = os.path.join(directory, time_stamp + "_" + username + ".txt")
if not os.path.exists(directory):
os.makedirs(directory)
file = open(complete_file_name, "w") # create time stamped file to be queued
file.write(fullname + "\n")
file.write(reason)
file.close()
print (time.strftime("%m-%d-%Y_%H:%M:%S") + '\tUser ' + username + ' added to queue')
Mitchell Moore
committed
socketio.emit("creating account")
except Exception as e:
print(time.strftime("%m-%d-%Y_%H:%M:%S") + "\tError in directory creation: ", e)
Mitchell Moore
committed
socketio.emit("Account creation failed")
@socketio.on("validate creation")
def creation_confirmation(json, methods=['GET', 'POST']):
username = json["username"]
if check_dir(username, 10):
print (time.strftime("%m-%d-%Y_%H:%M:%S") + '\tAccount successfully created for ' + username)
Mitchell Moore
committed
socketio.emit("Account created")
else:
socketio.emit("Account creation failed")
if __name__ == '__main__':
Mitchell Moore
committed
# app.run()
socketio.run(app)