"""Schlagwörter/Regeln erweitern: Quelle + Korrektur-Zähler + „behalten".

Manuelle Schlagwörter je Kategorie und die lernende Korrektur-Erkennung nutzen
dieselbe Tabelle ``categorization_rules`` wie die automatisch gelernten Regeln.

Revision ID: 0006
Revises: 0005
Create Date: 2026-08-19
"""
from alembic import op
import sqlalchemy as sa

revision = "0006"
down_revision = "0005"
branch_labels = None
depends_on = None


def upgrade() -> None:
    with op.batch_alter_table("categorization_rules") as batch:
        batch.add_column(sa.Column("source", sa.String(length=32), nullable=False, server_default="learned"))
        batch.add_column(sa.Column("applied_count", sa.Integer(), nullable=False, server_default=sa.text("0")))
        batch.add_column(sa.Column("overridden_count", sa.Integer(), nullable=False, server_default=sa.text("0")))
        batch.add_column(sa.Column("confirmed", sa.Boolean(), nullable=False, server_default=sa.text("0")))


def downgrade() -> None:
    with op.batch_alter_table("categorization_rules") as batch:
        batch.drop_column("confirmed")
        batch.drop_column("overridden_count")
        batch.drop_column("applied_count")
        batch.drop_column("source")