"""user classification/certification

Revision ID: 0007_user_classification
Revises: 0006_user_source_fields
Create Date: 2026-08-25
"""
from __future__ import annotations

from alembic import op
import sqlalchemy as sa

revision = "0007_user_classification"
down_revision = "0006_user_source_fields"
branch_labels = None
depends_on = None


def _add(table, col):
    conn = op.get_bind()
    cols = {r[1] for r in conn.execute(sa.text(f"PRAGMA table_info({table})"))}
    if col.name not in cols:
        op.add_column(table, col)


def upgrade() -> None:
    _add("users", sa.Column("certification_status", sa.String, server_default="uncertified"))
    _add("users", sa.Column("certification_time", sa.String, server_default=""))
    _add("users", sa.Column("affiliation", sa.String, server_default="independent"))
    _add("users", sa.Column("park_id", sa.String, server_default=""))
    _add("users", sa.Column("park_name", sa.String, server_default=""))
    _add("users", sa.Column("account_type", sa.String, server_default="opc_default"))


def downgrade() -> None:
    pass