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