From 0b5b0c54590932706595e248e4b8ccd8bf261af0 Mon Sep 17 00:00:00 2001
From: Bo-Chun Louis Chen <louistw@uab.edu>
Date: Mon, 12 Aug 2024 16:30:01 -0500
Subject: [PATCH] feat: add user to default groups
---
prod_rmq_agents/task_manager.py | 15 +++++++++++++--
1 file changed, 13 insertions(+), 2 deletions(-)
diff --git a/prod_rmq_agents/task_manager.py b/prod_rmq_agents/task_manager.py
index 3b82f83..210329e 100644
--- a/prod_rmq_agents/task_manager.py
+++ b/prod_rmq_agents/task_manager.py
@@ -266,13 +266,25 @@ def task_manager(ch, method, properties, body):
# Send trigger message
rc_rmq.publish_msg({"routing_key": routing_key, "msg": message})
+ if task_name == "create_account" and success:
+ rc_rmq.publish_msg(
+ {
+ "routing_key": "group_member.add",
+ "msg": {
+ "groups": {"add": rcfg.default_groups},
+ "username": username,
+ "host": msg.get("host"),
+ "updated_by": msg.get("updated_by"),
+ "interface": msg.get("interface"),
+ },
+ }
+ )
logger.debug(f"Trigger message '{routing_key}' sent")
logger.debug("Previous level messages acknowledged")
# Send report to admin
if completed or terminated:
-
notify_admin(username, current)
update_db(username, {"reported": True})
@@ -295,7 +307,6 @@ def timeout_handler(signum, frame):
delta = current_time - tracking[user]["last_update"]
if delta.seconds > timeout:
-
rc_rmq.publish_msg(
{
"routing_key": "complete." + user,
--
GitLab