"""将斗鱼供应商直充默认账号模板切换为昵称。

Revision ID: 20260814_0028
Revises: 20260813_0027
Create Date: 2026-08-14
"""

from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


revision: str = "20260814_0028"
down_revision: Union[str, None] = "20260813_0027"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    """只迁移历史默认值，保留运营人员明确配置的模板名。"""
    bind = op.get_bind()
    inspector = sa.inspect(bind)
    if not inspector.has_table("douyu_config"):
        return
    columns = {column["name"] for column in inspector.get_columns("douyu_config")}
    if "gold_api_account_template_name" in columns:
        op.execute(
            sa.text(
                "UPDATE douyu_config SET gold_api_account_template_name = :new "
                "WHERE gold_api_account_template_name IS NULL "
                "OR TRIM(gold_api_account_template_name) = '' "
                "OR gold_api_account_template_name = :old"
            ).bindparams(new="斗鱼昵称", old="斗鱼UID")
        )


def downgrade() -> None:
    """回滚时只回退本迁移设置的默认昵称。"""
    bind = op.get_bind()
    inspector = sa.inspect(bind)
    if not inspector.has_table("douyu_config"):
        return
    columns = {column["name"] for column in inspector.get_columns("douyu_config")}
    if "gold_api_account_template_name" in columns:
        op.execute(
            sa.text(
                "UPDATE douyu_config SET gold_api_account_template_name = :old "
                "WHERE gold_api_account_template_name = :new"
            ).bindparams(new="斗鱼昵称", old="斗鱼UID")
        )