"""Add artist_spyid column

Revision ID: 97c38df88248
Revises: fb047f44b1cd
Create Date: 2017-01-07 16:31:12.165152

"""

# revision identifiers, used by Alembic.
revision = '97c38df88248'
down_revision = 'fb047f44b1cd'
branch_labels = None
depends_on = None

from alembic import op
import sqlalchemy as sa

from sqlalchemy.sql import text


def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.add_column('spy_tracks', sa.Column('primary_artist_spyid', sa.String(), nullable=True))

    conn = op.get_bind()
    # conn.execute(text("""insert into
    #                          field_tags
    #                          (id, field_id, code, description)
    #                      values
    #                          (1, 'zasz', :code , :description')"""), **t)
    conn.execute(text("""
        update spy_tracks
        set primary_artist_spyid = spy_tracks.data->'artists'->0->>'id'
        """))

    ### end Alembic commands ###


def downgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('spy_tracks', 'primary_artist_spyid')
    ### end Alembic commands ###