Skip to content
Snippets Groups Projects

Draft: Add created column

Open Bo-Chun Chen requested to merge louistw/rabbitmq_agents:feat-created-column into main
Files
2
@@ -35,7 +35,7 @@ def upgrade():
@@ -35,7 +35,7 @@ def upgrade():
for row in reader:
for row in reader:
if row[0] in users:
if row[0] in users:
users[row[0]]["created"] = fromisoformat(row[1][:-3])
users[row[0]]["created"] = fromisoformat(row[1][:-3])
users[row[0]]["origin"] = row[2]
users[row[0]]["created_source"] = row[2]
# Add copy created column from last_update if they are not in csv file
# Add copy created column from last_update if they are not in csv file
for username in users:
for username in users:
@@ -43,13 +43,13 @@ def upgrade():
@@ -43,13 +43,13 @@ def upgrade():
users[username]["created"] = fromisoformat(
users[username]["created"] = fromisoformat(
users[username]["last_update"]
users[username]["last_update"]
)
)
users[username]["origin"] = "last_update"
users[username]["created_source"] = "last_update"
# No need to update this column back to the db
# No need to update this column back to the db
del users[username]["last_update"]
del users[username]["last_update"]
# Create new columns
# Create new columns
table.create_column("created", db.types.datetime)
table.create_column("created", db.types.datetime)
table.create_column("origin", db.types.text)
table.create_column("created_source", db.types.text)
# Update database in chunk
# Update database in chunk
table.update_many(list(users.values()), ["username"], ensure=True)
table.update_many(list(users.values()), ["username"], ensure=True)
Loading