"""create-artist-team

Revision ID: 0c3d5bb8c11a
Revises: 140f649d4b5a
Create Date: 2021-07-09 10:12:10.061632

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

# revision identifiers, used by Alembic.
revision = '0c3d5bb8c11a'
down_revision = '4f880c1a4ead'
branch_labels = None
depends_on = None


def upgrade():
    op.create_table('ArtistTeam',
        sa.Column('id', sa.BigInteger(), server_default=sa.text('nextval(\'"ArtistTeam_id_seq"\'::regclass)'), autoincrement=True, nullable=False),
        sa.Column('artist_id', sa.Integer(), nullable=False),
        sa.Column('label_id', sa.Integer(), nullable=False),
        sa.PrimaryKeyConstraint('id', name='pk_ArtistTeam'),
        sa.UniqueConstraint('artist_id', 'label_id', name='uq_ArtistTeam_artist_id_label_id')
    )
    op.create_foreign_key(op.f('fk_ArtistTeam_artist_id_Artist'), 'ArtistTeam', 'Artist', ['artist_id'], ['id'],
                          ondelete='CASCADE')
    op.create_foreign_key(op.f('fk_ArtistTeam_label_id_Label'), 'ArtistTeam', 'Label', ['label_id'], ['id'],
                          ondelete='CASCADE')

    op.create_table('ArtistTeamUser',
        sa.Column('id', sa.BigInteger(),
            server_default=sa.text('nextval(\'"ArtistTeamUser_id_seq"\'::regclass)'),
            autoincrement=True,
            nullable=False
        ),
        sa.Column('artist_team_id', sa.BigInteger(), nullable=False),
        sa.Column('user_id', sa.Integer(), nullable=False),
        sa.Column('role', sa.Integer(), nullable=False),
        sa.PrimaryKeyConstraint('id', name='pk_ArtistTeamUser'),
        sa.UniqueConstraint('artist_team_id', 'user_id', name='uq_ArtistTeamUser_artist_team_id_user_id')
    )
    op.create_foreign_key(op.f('fk_ArtistTeamUser_user_id_User'), 'ArtistTeamUser', 'User', ['user_id'], ['id'],
                          ondelete='CASCADE')
    op.create_foreign_key(op.f('fk_ArtistTeamUser_artist_team_id_ArtistTeam'), 'ArtistTeamUser', 'ArtistTeam',
                          ['artist_team_id'], ['id'], ondelete='CASCADE')

    op.create_table('ArtistTeamUserCampaignTypeGroup',
        sa.Column('artist_team_user_id', sa.Integer(), nullable=False),
        sa.Column('campaign_type_group_id', sa.Integer(), nullable=False),
        sa.ForeignKeyConstraint(['campaign_type_group_id'], ['CampaignTypeGroup.id'], name=op.f('fk_ArtistTeamUserCampaignTypeGroup_campaign_type_group_id_CampaignTypeGroup')),
        sa.ForeignKeyConstraint(['artist_team_user_id'], ['ArtistTeamUser.id'], name=op.f('fk_ArtistTeamUserCampaignTypeGroup_artist_team_user_id_ArtistTeamUser'), ondelete='CASCADE'),
        sa.PrimaryKeyConstraint('artist_team_user_id', 'campaign_type_group_id', name=op.f('pk_ArtistTeamUserCampaignTypeGroup')),
    )

    bind = op.get_bind()
    bind.execute('''ALTER SEQUENCE "ArtistTeam_id_seq" RESTART WITH 1 ''')
    bind.execute('''ALTER SEQUENCE "ArtistTeamUser_id_seq" RESTART WITH 1 ''')


def downgrade():
    op.drop_table('ArtistTeamUserCampaignTypeGroup')
    op.drop_table('ArtistTeamUser')
    op.drop_table('ArtistTeam')