"""Add youtube tables

Revision ID: 496f05aecdbe
Revises: ce2f9a32e798
Create Date: 2016-12-18 09:50:57.369899

"""

# revision identifiers, used by Alembic.
revision = '496f05aecdbe'
down_revision = 'ce2f9a32e798'
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('yt_channel_videos',
    sa.Column('video_ytid', sa.String(), nullable=False),
    sa.Column('channel_ytid', sa.String(), nullable=False),
    sa.PrimaryKeyConstraint('video_ytid', 'channel_ytid')
    )
    op.create_table('yt_channels',
    sa.Column('ytid', sa.String(), autoincrement=False, nullable=False),
    sa.Column('name', sa.String(), nullable=True),
    sa.Column('first_seen', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False),
    sa.Column('last_updated', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False),
    sa.PrimaryKeyConstraint('ytid')
    )
    op.create_table('yt_video_statistics',
    sa.Column('video_ytid', sa.String(), nullable=False),
    sa.Column('as_of', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False),
    sa.Column('comment_count', sa.Integer(), autoincrement=False, nullable=False),
    sa.Column('dislike_count', sa.Integer(), autoincrement=False, nullable=False),
    sa.Column('favorite_count', sa.Integer(), autoincrement=False, nullable=False),
    sa.Column('like_count', sa.Integer(), autoincrement=False, nullable=False),
    sa.Column('view_count', sa.Integer(), autoincrement=False, nullable=False),
    sa.PrimaryKeyConstraint('video_ytid', 'as_of')
    )
    op.create_table('yt_videos',
    sa.Column('ytid', sa.String(), autoincrement=False, nullable=False),
    sa.Column('title', sa.String(), nullable=False),
    sa.Column('published', sa.TIMESTAMP(timezone=True), nullable=False),
    sa.Column('first_seen', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False),
    sa.Column('last_updated', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False),
    sa.Column('data', postgresql.JSONB(), nullable=True),
    sa.Column('tags', postgresql.JSONB(), server_default=sa.text("'[]'::json"), nullable=False),
    sa.PrimaryKeyConstraint('ytid')
    )
    ### end Alembic commands ###


def downgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.drop_table('yt_videos')
    op.drop_table('yt_video_statistics')
    op.drop_table('yt_channels')
    op.drop_table('yt_channel_videos')
    ### end Alembic commands ###