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

revision = '004'
down_revision = '003'
branch_labels = None
depends_on = None

def upgrade():
    op.add_column('tax_documents', sa.Column('supersedes', JSONB, nullable=True), schema='data_center')
    op.add_column('tax_documents', sa.Column('references', JSONB, nullable=True), schema='data_center')

def downgrade():
    op.drop_column('tax_documents', 'references', schema='data_center')
    op.drop_column('tax_documents', 'supersedes', schema='data_center')