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

Rename columns for date and source

parent 6a23c120
No related branches found
No related tags found
No related merge requests found
......@@ -34,22 +34,22 @@ def upgrade():
next(reader)
for row in reader:
if row[0] in users:
users[row[0]]["created"] = fromisoformat(row[1][:-3])
users[row[0]]["origin"] = row[2]
users[row[0]]["date"] = fromisoformat(row[1])
users[row[0]]["source"] = row[2]
# Add copy created column from last_update if they are not in csv file
for username in users:
if "created" not in users[username]:
users[username]["created"] = fromisoformat(
users[username]["date"] = fromisoformat(
users[username]["last_update"]
)
users[username]["origin"] = "last_update"
users[username]["source"] = "account-web"
# No need to update this column back to the db
del users[username]["last_update"]
# Create new columns
table.create_column("created", db.types.datetime)
table.create_column("origin", db.types.text)
table.create_column("date", db.types.datetime)
table.create_column("source", db.types.text)
# Update database in chunk
table.update_many(list(users.values()), ["username"], ensure=True)
......
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