"""credito/debito de imposto (imposto_id em gasto/insumo + cenario_imposto)

Revision ID: b3171abdaafc
Revises: 898afd3b600b
Create Date: 2026-09-18 01:30:00.000000

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = 'b3171abdaafc'
down_revision: Union[str, None] = '898afd3b600b'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    for tabela in ('ambiente_gasto', 'maquina_gasto', 'funcionario_gasto', 'insumo'):
        op.add_column(tabela, sa.Column('imposto_id', sa.Integer(), nullable=True))
        op.create_foreign_key(f'{tabela}_imposto_id_fkey', tabela, 'imposto', ['imposto_id'], ['id'])

    op.create_table(
        'cenario_imposto',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('cenario_id', sa.Integer(), nullable=False),
        sa.Column('imposto_id', sa.Integer(), nullable=False),
        sa.Column('aliquota_pct', sa.Numeric(precision=6, scale=4), nullable=False),
        sa.ForeignKeyConstraint(['cenario_id'], ['cenario_precificacao.id'], ),
        sa.ForeignKeyConstraint(['imposto_id'], ['imposto.id'], ),
        sa.PrimaryKeyConstraint('id'),
        sa.UniqueConstraint('cenario_id', 'imposto_id', name='uq_cenario_imposto'),
    )


def downgrade() -> None:
    op.drop_table('cenario_imposto')

    for tabela in ('ambiente_gasto', 'maquina_gasto', 'funcionario_gasto', 'insumo'):
        op.drop_constraint(f'{tabela}_imposto_id_fkey', tabela, type_='foreignkey')
        op.drop_column(tabela, 'imposto_id')