diff --git a/init_user_state.py b/init_user_state.py index c2b7628ecbeaa0db389201018147d8d8e3e8412b..8d5c6084e9b18067f590f513591ddbdfa0d47a24 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, + } )