Skip to content

Commit

Permalink
Handle data in the migration
Browse files Browse the repository at this point in the history
  • Loading branch information
mayaCostantini committed Feb 3, 2022
1 parent 8f8365e commit b82b8c9
Showing 1 changed file with 8 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,17 +18,25 @@

def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
connection = op.get_bind()
result = connection.execute("SELECT project_url FROM python_package_metadata_project_url").fetchall()
op.add_column("python_package_metadata_project_url", sa.Column("url", sa.Text(), nullable=True))
op.add_column("python_package_metadata_project_url", sa.Column("label", sa.Text(), nullable=True))
projects_source_url = [{"label": r[0].strip(), "url": r[1].strip()} for r in result.split(",")]
op.bulk_insert("python_package_metadata_project_url", projects_source_url)
op.drop_column("python_package_metadata_project_url", "project_url")
# ### end Alembic commands ###


def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
connection = op.get_bind()
op.add_column(
"python_package_metadata_project_url", sa.Column("project_url", sa.TEXT(), autoincrement=False, nullable=True)
)
result = connection.execute("SELECT label, url FROM python_package_metadata_project_url").fetchall()
op.drop_column("python_package_metadata_project_url", "label")
op.drop_column("python_package_metadata_project_url", "url")
concat_label_url = [{"project_url": r.join(",")} for r in result]
op.bulk_insert("python_package_metadata_project_url", concat_label_url)
# ### end Alembic commands ###

0 comments on commit b82b8c9

Please sign in to comment.