From 04a11ccdc04d8d9a4a0a0dbeacbc9b90f52208fb Mon Sep 17 00:00:00 2001
From: Bo-Chun Louis Chen <louistw@uab.edu>
Date: Fri, 13 May 2022 17:45:27 -0500
Subject: [PATCH] Add get_caller_info function in rc_util
---
rc_util.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/rc_util.py b/rc_util.py
index 9becf2f..fbb7c39 100644
--- a/rc_util.py
+++ b/rc_util.py
@@ -5,6 +5,7 @@ import signal
import logging
import argparse
import pika
+import pwd
import uuid
from rc_rmq import RCRMQ
import json
@@ -63,6 +64,7 @@ def add_account(username, queuename, email, full="", reason=""):
)
rc_rmq.disconnect()
+
def certify_account(username, queuename, state="ok", service="all"):
rc_rmq.publish_msg(
{
@@ -77,6 +79,7 @@ def certify_account(username, queuename, state="ok", service="all"):
)
rc_rmq.disconnect()
+
def worker(ch, method, properties, body):
msg = json.loads(body)
username = msg["username"]
@@ -260,3 +263,7 @@ def update_state(username, state, debug=False):
)
return result
+
+
+def get_caller_info():
+ return f"{pwd.getpwuid(os.getuid()).pw_name}@{os.uname().nodename}"
--
GitLab