"""add_sc_all_tracks_partition

Revision ID: 07a38708c8ce
Revises: 7e4ca0d58ad7
Create Date: 2020-12-10 11:40:48.944492

"""

# revision identifiers, used by Alembic.
revision = '07a38708c8ce'
down_revision = '7e4ca0d58ad7'
branch_labels = None
depends_on = None

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

def upgrade():

    op.execute('''
        alter table sc_tracks add column lifecycle_stage text null;

        create table sc_tracks_archived (like sc_tracks including all);
        
        create table sc_tracks_all (like sc_tracks_archived including defaults) 
        partition by list (lifecycle_stage);
        
        alter table sc_tracks_all attach partition sc_tracks_archived for values in ('archived');        
        alter table sc_tracks_all attach partition sc_tracks default;        
    ''')



def downgrade():
    op.drop_table('sc_tracks_all')
    op.drop_column('sc_tracks', 'lifecycle_stage')
    # ### end Alembic commands ###