"""Add fields to track tax transactions

Revision ID: 73171002f840
Revises: 8acb9ae6dfbe
Create Date: 2025-06-13 10:39:23.786580

"""

import sqlalchemy as sa
from alembic import op

# Polar Custom Imports

# revision identifiers, used by Alembic.
revision = "73171002f840"
down_revision = "8acb9ae6dfbe"
branch_labels: tuple[str] | None = None
depends_on: tuple[str] | None = None


def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column(
        "checkouts", sa.Column("tax_processor_id", sa.String(), nullable=True)
    )
    op.add_column(
        "orders", sa.Column("tax_transaction_processor_id", sa.String(), nullable=True)
    )
    op.add_column(
        "refunds", sa.Column("tax_transaction_processor_id", sa.String(), nullable=True)
    )
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column("refunds", "tax_transaction_processor_id")
    op.drop_column("orders", "tax_transaction_processor_id")
    op.drop_column("checkouts", "tax_processor_id")
    # ### end Alembic commands ###
