"""Add invoice payment tracking

Revision ID: 80e66447bbd9
Revises: 7f2e1c5ac8a7
Create Date: 2026-07-14

"""

from alembic import op
import sqlalchemy as sa


revision = "80e66447bbd9"
down_revision = "7f2e1c5ac8a7"
branch_labels = None
depends_on = None



def upgrade():

    with op.batch_alter_table("invoice") as batch_op:

        batch_op.add_column(
            sa.Column(
                "paid_amount",
                sa.Float(),
                nullable=False,
                server_default="0"
            )
        )


        batch_op.add_column(
            sa.Column(
                "due_amount",
                sa.Float(),
                nullable=False,
                server_default="0"
            )
        )


        batch_op.add_column(
            sa.Column(
                "payment_status",
                sa.String(length=20),
                nullable=False,
                server_default="Paid"
            )
        )


        batch_op.add_column(
            sa.Column(
                "payment_method",
                sa.String(length=30),
                nullable=False,
                server_default="Cash"
            )
        )



def downgrade():

    with op.batch_alter_table("invoice") as batch_op:

        batch_op.drop_column("payment_method")

        batch_op.drop_column("payment_status")

        batch_op.drop_column("due_amount")

        batch_op.drop_column("paid_amount")