atst/alembic/versions/59973fa17ded_portfolio_state_machine_table.py
2020-01-14 16:53:15 -05:00

114 lines
3.8 KiB
Python

"""portfolio state machine table.
Revision ID: 59973fa17ded
Revises: 828d8c188dce
Create Date: 2020-01-08 10:37:32.924245
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
import sqlalchemy_json
# revision identifiers, used by Alembic.
revision = "59973fa17ded" # pragma: allowlist secret
down_revision = "828d8c188dce" # pragma: allowlist secret
branch_labels = None
depends_on = None
def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.create_table(
"portfolio_job_failures",
sa.Column("id", sa.Integer(), nullable=False),
sa.Column("task_id", sa.String(), nullable=False),
sa.Column("portfolio_id", postgresql.UUID(as_uuid=True), nullable=False),
sa.ForeignKeyConstraint(["portfolio_id"], ["portfolios.id"],),
sa.PrimaryKeyConstraint("id"),
)
op.create_table(
"portfolio_state_machines",
sa.Column(
"time_created",
sa.TIMESTAMP(timezone=True),
server_default=sa.text("now()"),
nullable=False,
),
sa.Column(
"time_updated",
sa.TIMESTAMP(timezone=True),
server_default=sa.text("now()"),
nullable=False,
),
sa.Column(
"deleted", sa.Boolean(), server_default=sa.text("false"), nullable=False
),
sa.Column(
"id",
postgresql.UUID(as_uuid=True),
server_default=sa.text("uuid_generate_v4()"),
nullable=False,
),
sa.Column("portfolio_id", postgresql.UUID(as_uuid=True), nullable=True),
sa.Column(
"state",
sa.Enum(
"UNSTARTED",
"STARTING",
"STARTED",
"COMPLETED",
"FAILED",
"TENANT_CREATED",
"TENANT_IN_PROGRESS",
"TENANT_FAILED",
"BILLING_PROFILE_CREATED",
"BILLING_PROFILE_IN_PROGRESS",
"BILLING_PROFILE_FAILED",
"ADMIN_SUBSCRIPTION_CREATED",
"ADMIN_SUBSCRIPTION_IN_PROGRESS",
"ADMIN_SUBSCRIPTION_FAILED",
name="fsmstates",
native_enum=False,
),
nullable=False,
),
sa.ForeignKeyConstraint(["portfolio_id"], ["portfolios.id"],),
sa.PrimaryKeyConstraint("id"),
)
op.add_column("portfolios", sa.Column("app_migration", sa.String(), nullable=True))
op.add_column(
"portfolios", sa.Column("complexity", sa.ARRAY(sa.String()), nullable=True)
)
op.add_column(
"portfolios", sa.Column("complexity_other", sa.String(), nullable=True)
)
op.add_column(
"portfolios",
sa.Column("csp_data", sqlalchemy_json.NestedMutableJson(), nullable=True),
)
op.add_column(
"portfolios", sa.Column("dev_team", sa.ARRAY(sa.String()), nullable=True)
)
op.add_column("portfolios", sa.Column("dev_team_other", sa.String(), nullable=True))
op.add_column("portfolios", sa.Column("native_apps", sa.String(), nullable=True))
op.add_column(
"portfolios", sa.Column("team_experience", sa.String(), nullable=True)
)
# ### end Alembic commands ###
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_column("portfolios", "team_experience")
op.drop_column("portfolios", "native_apps")
op.drop_column("portfolios", "dev_team_other")
op.drop_column("portfolios", "dev_team")
op.drop_column("portfolios", "csp_data")
op.drop_column("portfolios", "complexity_other")
op.drop_column("portfolios", "complexity")
op.drop_column("portfolios", "app_migration")
op.drop_table("portfolio_state_machines")
op.drop_table("portfolio_job_failures")
# ### end Alembic commands ###