"""Quiz origin tracking for uploads and AI-generated tests.

Revision ID: k4f5a6b7c819
Revises: j3e4f5a6b708
"""
from alembic import op

revision = "k4f5a6b7c819"
down_revision = "j3e4f5a6b708"
branch_labels = None
depends_on = None


def upgrade():
    op.execute("ALTER TABLE quizzes ADD COLUMN IF NOT EXISTS origin VARCHAR(20) NOT NULL DEFAULT 'bank'")


def downgrade():
    op.execute("ALTER TABLE quizzes DROP COLUMN IF EXISTS origin")