From 7cf58dbda563d1c8a6cd3b118e0fc80dc6865e77 Mon Sep 17 00:00:00 2001
From: Mitchell Moore <mmoo97@uab.edu>
Date: Wed, 8 Jul 2020 09:47:44 -0500
Subject: [PATCH] create test harness
---
run.py | 5 +++--
tasks.py | 46 +++++++++++++++++++++++-----------------------
2 files changed, 26 insertions(+), 25 deletions(-)
diff --git a/run.py b/run.py
index ff6b025..83fb57c 100644
--- a/run.py
+++ b/run.py
@@ -16,7 +16,7 @@ monkey.patch_all(subprocess=True)
config_name = os.getenv('FLASK_CONFIG')
app = create_app(config_name)
app.config['SECRET_KEY'] = vars.key
-socketio = SocketIO(app, message_queue= vars.message_queue)
+socketio = SocketIO(app)
@@ -36,7 +36,8 @@ def request_account(json, methods=['GET', 'POST']):
room = str(session['uid'])
print("Room: {}".format(room))
try:
- tasks.celery_create_account.delay(json['username'], json['fullname'], json['reason'], session=room)
+ print("hi")
+ # tasks.celery_create_account.delay(json['username'], json['fullname'], json['reason'], session=room)
except Exception as e:
print(time.strftime("%m-%d-%Y_%H:%M:%S") + "\tError in account creation: ", e)
socketio.emit("Account creation failed", room)
diff --git a/tasks.py b/tasks.py
index 7deec3a..c45f23e 100644
--- a/tasks.py
+++ b/tasks.py
@@ -4,26 +4,26 @@ from flask_socketio import SocketIO
import subprocess
import vars
-from gevent import monkey
-monkey.patch_all(subprocess=True)
-
-broker_url = vars.broker_url
-celery = Celery('flask_user_reg', broker=broker_url)
-
-socketio = SocketIO(message_queue=vars.message_queue)
-
-
-def send_msg(event, room):
- print("Post '{}' to room '{}'".format(event,room))
- socketio.emit(event, room=room)
-
-
-@celery.task
-def celery_create_account(username, fullname, reason, session):
- room = session
- print(time.strftime("%m-%d-%Y_%H:%M:%S") + '\tUser ' + username + ' added to queue')
- send_msg('creating account', room)
- print(username)
- subprocess.call(["/opt/rabbitmq_agents/flask_producer.py", "ohpc_account_create", username])
- print(time.strftime("%m-%d-%Y_%H:%M:%S") + '\tAccount successfully created for ' + username)
- send_msg('account ready', room)
+# from gevent import monkey
+# monkey.patch_all(subprocess=True)
+#
+# broker_url = vars.broker_url
+# celery = Celery('flask_user_reg', broker=broker_url)
+#
+# socketio = SocketIO(message_queue=vars.message_queue)
+#
+#
+# def send_msg(event, room):
+# print("Post '{}' to room '{}'".format(event,room))
+# socketio.emit(event, room=room)
+#
+#
+# @celery.task
+# def celery_create_account(username, fullname, reason, session):
+# room = session
+# print(time.strftime("%m-%d-%Y_%H:%M:%S") + '\tUser ' + username + ' added to queue')
+# send_msg('creating account', room)
+# print(username)
+# subprocess.call(["/opt/rabbitmq_agents/flask_producer.py", "ohpc_account_create", username])
+# print(time.strftime("%m-%d-%Y_%H:%M:%S") + '\tAccount successfully created for ' + username)
+# send_msg('account ready', room)
--
GitLab