"""kworb_youtube_trending

Revision ID: 94440b575488
Revises: a54c59983d1a
Create Date: 2021-02-12 15:43:29.552855

"""

# revision identifiers, used by Alembic.
revision = '94440b575488'
down_revision = 'a54c59983d1a'
branch_labels = None
depends_on = None

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

def upgrade():
    op.execute('''
        create table kworb_youtube_trending_charts
        (
            country_code varchar(2) not null,
            as_of date not null,
            position integer not null,
            title varchar not null,
            position_change integer,
            ytid varchar
            
        ) partition by range(as_of);

        create table kworb_youtube_trending_charts_default partition of kworb_youtube_trending_charts default; 

        create unique index on kworb_youtube_trending_charts_default (as_of, country_code, position);

            ''')


def downgrade():
    op.execute('drop table kworb_youtube_trending_charts')