"""create_revenue_raw_table

Revision ID: 37ba12ded81e
Revises: e8efbcb0183e
Create Date: 2021-08-16 07:43:30.921227

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '37ba12ded81e'
down_revision = 'e8efbcb0183e'
branch_labels = None
depends_on = None


def upgrade():
    op.create_table(
        "RawStreamRates",
        sa.Column('country', sa.String(), nullable=False),
        sa.Column('country_code', sa.String(), nullable=False, primary_key=True),
        sa.Column('cross_platform_blended_rate', sa.String(), nullable=False),
        sa.Column('spotify_blended_rate', sa.String(), nullable=False),
        sa.Column('apple_blended_rate', sa.String(), nullable=False),
        sa.Column('amazon_blended_rate', sa.String(), nullable=False),
        sa.Column('youtube_blended_rate', sa.String(), nullable=False)
    )
    op.add_column("PRSImportInfo", sa.Column('imported_streams_rate_date', sa.Date(), nullable=True))

    op.create_table(
        "StreamRate",
        sa.Column('id', sa.BIGINT(),
                  autoincrement=True,
                  nullable=False,
                  primary_key=True
        ),
        sa.Column('territory_id', sa.BigInteger(), nullable=False),

        sa.Column('cross_platform_blended_rate', sa.DECIMAL(), nullable=False),
        sa.Column('spotify_blended_rate', sa.DECIMAL(), nullable=False),
        sa.Column('apple_blended_rate', sa.DECIMAL(), nullable=False),
        sa.Column('amazon_blended_rate', sa.DECIMAL(), nullable=False),
        sa.Column('youtube_blended_rate', sa.DECIMAL(), nullable=False),
        sa.Column('imported_date', sa.Date(), nullable=False),
        sa.ForeignKeyConstraint(['territory_id'], ['Territory.id'], name=op.f('fk_StreamRate_territory_id_Territory'), ondelete='CASCADE'),
    )

def downgrade():
    op.drop_table('RawStreamRates')