"""Add smart counter for quota consumption

Revision ID: d770f939dc13
Revises: 61026795ceb6
Create Date: 2026-01-26 17:11:25.402297

"""
from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'd770f939dc13'
down_revision = '61026795ceb6'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    with op.batch_alter_table('users', schema=None) as batch_op:
        batch_op.add_column(sa.Column('audits_created_current_month', sa.Integer(), server_default='0', nullable=False))
        batch_op.add_column(sa.Column('last_quota_reset_date', sa.DateTime(), nullable=True))

    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    with op.batch_alter_table('users', schema=None) as batch_op:
        batch_op.drop_column('last_quota_reset_date')
        batch_op.drop_column('audits_created_current_month')

    # ### end Alembic commands ###