diff --git a/alembic/versions/cfab6c8243cb_add_environment_creator_role.py b/alembic/versions/cfab6c8243cb_add_environment_creator_role.py index 7c548831..1ab23aee 100644 --- a/alembic/versions/cfab6c8243cb_add_environment_creator_role.py +++ b/alembic/versions/cfab6c8243cb_add_environment_creator_role.py @@ -10,21 +10,51 @@ import sqlalchemy as sa from sqlalchemy.dialects import postgresql # revision identifiers, used by Alembic. -revision = 'cfab6c8243cb' # pragma: allowlist secret -down_revision = 'f03333c42bdb' +revision = "cfab6c8243cb" # pragma: allowlist secret +down_revision = "f03333c42bdb" # pragma: allowlist secret branch_labels = None depends_on = None def upgrade(): # ### commands auto generated by Alembic - please adjust! ### - op.add_column('environments', sa.Column('creator_id', postgresql.UUID(as_uuid=True), nullable=False)) - op.create_foreign_key("fk_users_id", 'environments', 'users', ['creator_id'], ['id']) + op.add_column( + "environments", + sa.Column("creator_id", postgresql.UUID(as_uuid=True), nullable=True), + ) + op.create_foreign_key( + "fk_users_id", "environments", "users", ["creator_id"], ["id"] + ) # ### end Alembic commands ### + # Default environment's creator id to the portfolio POC + conn = op.get_bind() + results = conn.execute( + """ + SELECT + DISTINCT ON (e.id) e.id as id, pr.user_id as user_id + FROM environments e + JOIN applications a ON e.application_id = a.id + JOIN portfolios p ON a.portfolio_id = p.id + JOIN portfolio_roles pr ON pr.portfolio_id = p.id + JOIN portfolio_roles_permission_sets prps ON prps.portfolio_role_id = pr.id + JOIN permission_sets ps ON prps.permission_set_id = ps.id + WHERE ps.name = 'portfolio_poc' + ORDER BY e.id; + """ + ).fetchall() + + for r in results: + conn.execute( + "UPDATE environments e SET creator_id = %s WHERE e.id = %s", + [r["user_id"], r["id"]], + ) + + op.alter_column("environments", "creator_id", nullable=False) + def downgrade(): # ### commands auto generated by Alembic - please adjust! ### - op.drop_constraint("fk_users_id", 'environments', type_='foreignkey') - op.drop_column('environments', 'creator_id') + op.drop_constraint("fk_users_id", "environments", type_="foreignkey") + op.drop_column("environments", "creator_id") # ### end Alembic commands ###