diff --git a/alembic/versions/cfab6c8243cb_add_environment_creator_role.py b/alembic/versions/cfab6c8243cb_add_environment_creator_role.py new file mode 100644 index 00000000..33a7a2a0 --- /dev/null +++ b/alembic/versions/cfab6c8243cb_add_environment_creator_role.py @@ -0,0 +1,30 @@ +"""add Environment creator_role + +Revision ID: cfab6c8243cb +Revises: 502e79c55d2d +Create Date: 2019-09-10 11:21:43.252592 + +""" +from alembic import op +import sqlalchemy as sa +from sqlalchemy.dialects import postgresql + +# revision identifiers, used by Alembic. +revision = 'cfab6c8243cb' # pragma: allowlist secret +down_revision = '502e79c55d2d' +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.add_column('environments', sa.Column('creator_role_id', postgresql.UUID(as_uuid=True), nullable=False)) + op.create_foreign_key("fk_application_roles_id", 'environments', 'application_roles', ['creator_role_id'], ['id']) + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.drop_constraint("fk_application_roles_id", 'environments', type_='foreignkey') + op.drop_column('environments', 'creator_role_id') + # ### end Alembic commands ### diff --git a/atst/models/environment.py b/atst/models/environment.py index d85550c6..fde36934 100644 --- a/atst/models/environment.py +++ b/atst/models/environment.py @@ -18,6 +18,9 @@ class Environment( application_id = Column(ForeignKey("applications.id"), nullable=False) application = relationship("Application") + creator_role_id = Column(ForeignKey("application_roles.id"), nullable=False) + creator = relationship("ApplicationRole") + cloud_id = Column(String) root_user_info = Column(JSONB) baseline_info = Column(JSONB)