79 lines
3.7 KiB
Python
79 lines
3.7 KiB
Python
"""new task order table
|
|
|
|
Revision ID: 6172ac7b8b26
|
|
Revises: 1c1394e496a7
|
|
Create Date: 2018-12-19 10:00:51.373083
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '6172ac7b8b26'
|
|
down_revision = '1c1394e496a7'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table('task_orders',
|
|
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('id', postgresql.UUID(as_uuid=True), server_default=sa.text('uuid_generate_v4()'), nullable=False),
|
|
sa.Column('workspace_id', postgresql.UUID(as_uuid=True), nullable=True),
|
|
sa.Column('user_id', postgresql.UUID(as_uuid=True), nullable=True),
|
|
sa.Column('scope', sa.String(), nullable=True),
|
|
sa.Column('defense_component', sa.String(), nullable=True),
|
|
sa.Column('app_migration', sa.String(), nullable=True),
|
|
sa.Column('native_apps', sa.String(), nullable=True),
|
|
sa.Column('complexity', sa.ARRAY(sa.String()), nullable=True),
|
|
sa.Column('complexity_other', sa.String(), nullable=True),
|
|
sa.Column('dev_team', sa.ARRAY(sa.String()), nullable=True),
|
|
sa.Column('dev_team_other', sa.String(), nullable=True),
|
|
sa.Column('team_experience', sa.String(), nullable=True),
|
|
sa.Column('start_date', sa.Date(), nullable=True),
|
|
sa.Column('end_date', sa.Date(), nullable=True),
|
|
sa.Column('clin_01', sa.Numeric(scale=2), nullable=True),
|
|
sa.Column('clin_02', sa.Numeric(scale=2), nullable=True),
|
|
sa.Column('clin_03', sa.Numeric(scale=2), nullable=True),
|
|
sa.Column('clin_04', sa.Numeric(scale=2), nullable=True),
|
|
sa.Column('ko_first_name', sa.String(), nullable=True),
|
|
sa.Column('ko_last_name', sa.String(), nullable=True),
|
|
sa.Column('ko_email', sa.String(), nullable=True),
|
|
sa.Column('ko_dod_id', sa.String(), nullable=True),
|
|
sa.Column('cor_first_name', sa.String(), nullable=True),
|
|
sa.Column('cor_last_name', sa.String(), nullable=True),
|
|
sa.Column('cor_email', sa.String(), nullable=True),
|
|
sa.Column('cor_dod_id', sa.String(), nullable=True),
|
|
sa.Column('so_first_name', sa.String(), nullable=True),
|
|
sa.Column('so_last_name', sa.String(), nullable=True),
|
|
sa.Column('so_email', sa.String(), nullable=True),
|
|
sa.Column('so_dod_id', sa.String(), nullable=True),
|
|
sa.Column('number', sa.String(), nullable=True),
|
|
sa.Column('loa', sa.ARRAY(sa.String()), nullable=True),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.ForeignKeyConstraint(['workspace_id'], ['workspaces.id'], ),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('number')
|
|
)
|
|
op.alter_column('workspaces', 'request_id',
|
|
existing_type=postgresql.UUID(),
|
|
nullable=True)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade():
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
conn = op.get_bind()
|
|
conn.execute("DELETE FROM workspace_roles wr USING workspaces w WHERE w.id=wr.workspace_id AND w.request_id IS NULL")
|
|
conn.execute("DELETE FROM audit_events ae USING workspaces w WHERE w.id=ae.workspace_id AND w.request_id IS NULL")
|
|
conn.execute("DELETE FROM task_orders tasks USING workspaces w WHERE w.id=tasks.workspace_id AND w.request_id IS NULL")
|
|
conn.execute("DELETE FROM workspaces WHERE request_id IS NULL")
|
|
op.alter_column('workspaces', 'request_id',
|
|
existing_type=postgresql.UUID(),
|
|
nullable=False)
|
|
op.drop_table('task_orders')
|
|
# ### end Alembic commands ###
|