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

Create a new table insead of adding columns

parent 5e27a5eb
No related branches found
No related tags found
No related merge requests found
...@@ -19,7 +19,6 @@ def upgrade(): ...@@ -19,7 +19,6 @@ def upgrade():
csv_filename = "./migrations/account-registration-history.csv" csv_filename = "./migrations/account-registration-history.csv"
db = dataset.connect(f"sqlite:///{db_path}/{db_name}") db = dataset.connect(f"sqlite:///{db_path}/{db_name}")
table = db["users"]
# Get all username and last_update # Get all username and last_update
usernames = db.query("SELECT username, last_update FROM users") usernames = db.query("SELECT username, last_update FROM users")
...@@ -47,12 +46,13 @@ def upgrade(): ...@@ -47,12 +46,13 @@ def upgrade():
"source": row[2], "source": row[2],
} }
# Create new columns data_types = dict(
table.create_column("date", db.types.datetime) username=db.types.text, date=db.types.datetime, source=db.types.text
table.create_column("source", db.types.text) )
# Update database in chunk # Insert into database in chunks
table.update_many(list(users.values()), ["username"], ensure=True) table = db["user_created"]
table.insert_many(list(users.values()), ensure=True, types=data_types)
if __name__ == "__main__": if __name__ == "__main__":
......
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