Skip to content
Snippets Groups Projects
Commit ef25c506 authored by Bo-Chun Chen's avatar Bo-Chun Chen
Browse files

Update init state script with updated_by field

parent 1f237b33
No related branches found
No related tags found
2 merge requests!147Merge previous default branch feat-cod-rmq into main,!127Add updated_by field to user_state table
...@@ -4,6 +4,7 @@ import dataset ...@@ -4,6 +4,7 @@ import dataset
import sys import sys
import subprocess import subprocess
import rabbit_config as rcfg import rabbit_config as rcfg
import rc_util
from datetime import datetime from datetime import datetime
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
...@@ -15,6 +16,7 @@ args = parser.parse_args() ...@@ -15,6 +16,7 @@ args = parser.parse_args()
default_state = "ok" default_state = "ok"
today = datetime.now() today = datetime.now()
updated_by = rc_util.get_caller_info()
# Chunk size for insert into db # Chunk size for insert into db
size = 1000 size = 1000
...@@ -38,7 +40,12 @@ if len(users) > 50: ...@@ -38,7 +40,12 @@ if len(users) > 50:
while start < len(users): while start < len(users):
end = start + size if start + size < len(users) else len(users) end = start + size if start + size < len(users) else len(users)
data = [ 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] for user in users[start:end]
] ]
if args.dry_run: if args.dry_run:
...@@ -54,5 +61,10 @@ else: ...@@ -54,5 +61,10 @@ else:
print(f"Table insert user: {user}, state: {default_state}") print(f"Table insert user: {user}, state: {default_state}")
else: else:
table.insert( table.insert(
{"username": user, "state": default_state, "date": today} {
"username": user,
"state": default_state,
"date": today,
"updated_by": updated_by,
}
) )
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment