"""sc_track_regressions

Revision ID: ba7486130780
Revises: e4fe6d1602fc
Create Date: 2019-12-10 20:00:57.426738

"""

# revision identifiers, used by Alembic.
revision = 'ba7486130780'
down_revision = 'e4fe6d1602fc'
branch_labels = None
depends_on = None

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

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.create_table('sc_track_regressions',
    sa.Column('track_scid', sa.Integer(), nullable=False),
    sa.Column('num_days', sa.Integer(), nullable=True),
    sa.Column('total_plays', sa.Integer(), nullable=True),
    sa.Column('max_daily_plays', sa.Integer(), nullable=True),
    sa.Column('min_daily_plays', sa.Integer(), nullable=True),
    sa.Column('avg_daily_plays', sa.Float(), nullable=True),
    sa.Column('plays_yesterday', sa.Integer(), nullable=True),
    sa.Column('idates', postgresql.ARRAY(sa.Integer()), nullable=True),
    sa.Column('daily_plays', postgresql.ARRAY(sa.Integer()), nullable=True),
    sa.Column('slope15', sa.Float(), nullable=True),
    sa.Column('rsq15', sa.Float(), nullable=True),
    sa.Column('yint15', sa.Float(), nullable=True),
    sa.Column('slope22', sa.Float(), nullable=True),
    sa.Column('rsq22', sa.Float(), nullable=True),
    sa.Column('yint22', sa.Float(), nullable=True),
    sa.Column('slope29', sa.Float(), nullable=True),
    sa.Column('rsq29', sa.Float(), nullable=True),
    sa.Column('yint29', sa.Float(), nullable=True),
    sa.ForeignKeyConstraint(['track_scid'], ['sc_tracks.scid'], name='fk_sc_track_regressions'),
    sa.PrimaryKeyConstraint('track_scid')
    )
    op.create_index('ix_sc_track_regressions_15', 'sc_track_regressions', ['num_days', 'slope15', 'rsq15', 'plays_yesterday'], unique=False)
    op.create_index('ix_sc_track_regressions_22', 'sc_track_regressions', ['num_days', 'slope22', 'rsq22', 'plays_yesterday'], unique=False)
    op.create_index('ix_sc_track_regressions_29', 'sc_track_regressions', ['num_days', 'slope29', 'rsq29', 'plays_yesterday'], unique=False)


def downgrade():
    op.drop_index('ix_sc_track_regressions_29', table_name='sc_track_regressions')
    op.drop_index('ix_sc_track_regressions_22', table_name='sc_track_regressions')
    op.drop_index('ix_sc_track_regressions_15', table_name='sc_track_regressions')
    op.drop_table('sc_track_regressions')