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