From ef25c506d427b1f7dd926e83cdc7e9d621a9a7d1 Mon Sep 17 00:00:00 2001
From: Bo-Chun Louis Chen <louistw@uab.edu>
Date: Mon, 16 May 2022 14:32:18 -0500
Subject: [PATCH] Update init state script with updated_by field

---
 init_user_state.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/init_user_state.py b/init_user_state.py
index c2b7628..8d5c608 100644
--- a/init_user_state.py
+++ b/init_user_state.py
@@ -4,6 +4,7 @@ import dataset
 import sys
 import subprocess
 import rabbit_config as rcfg
+import rc_util
 from datetime import datetime
 
 parser = argparse.ArgumentParser()
@@ -15,6 +16,7 @@ args = parser.parse_args()
 
 default_state = "ok"
 today = datetime.now()
+updated_by = rc_util.get_caller_info()
 
 # Chunk size for insert into db
 size = 1000
@@ -38,7 +40,12 @@ if len(users) > 50:
     while start < len(users):
         end = start + size if start + size < len(users) else len(users)
         data = [
-            dict(username=user, state=default_state, date=today)
+            dict(
+                username=user,
+                state=default_state,
+                date=today,
+                updated_by=updated_by,
+            )
             for user in users[start:end]
         ]
         if args.dry_run:
@@ -54,5 +61,10 @@ else:
             print(f"Table insert user: {user}, state: {default_state}")
         else:
             table.insert(
-                {"username": user, "state": default_state, "date": today}
+                {
+                    "username": user,
+                    "state": default_state,
+                    "date": today,
+                    "updated_by": updated_by,
+                }
             )
-- 
GitLab