"""Add share_token to bill_briefs

Revision ID: 0016
Revises: 0015
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

revision = "0016"
down_revision = "0015"
branch_labels = None
depends_on = None


def upgrade():
    op.add_column(
        "bill_briefs",
        sa.Column(
            "share_token",
            postgresql.UUID(as_uuid=False),
            nullable=True,
            server_default=sa.text("gen_random_uuid()"),
        ),
    )
    op.create_unique_constraint("uq_brief_share_token", "bill_briefs", ["share_token"])
    op.create_index("ix_brief_share_token", "bill_briefs", ["share_token"])


def downgrade():
    op.drop_index("ix_brief_share_token", "bill_briefs")
    op.drop_constraint("uq_brief_share_token", "bill_briefs")
    op.drop_column("bill_briefs", "share_token")