"""扩展应用宝二维码密文字段容量"""

from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


revision: str = "20260812_0021"
down_revision: Union[str, None] = "20260812_0020"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def _is_mysql() -> bool:
    return op.get_bind().dialect.name in {"mysql", "mariadb"}


def upgrade() -> None:
    if _is_mysql():
        op.execute(
            "ALTER TABLE yyb_recharge_tasks MODIFY login_qr_data MEDIUMTEXT NULL"
        )
        op.execute(
            "ALTER TABLE yyb_recharge_tasks MODIFY payment_qr_data MEDIUMTEXT NULL"
        )


def downgrade() -> None:
    if _is_mysql():
        op.execute("ALTER TABLE yyb_recharge_tasks MODIFY login_qr_data TEXT NULL")
        op.execute("ALTER TABLE yyb_recharge_tasks MODIFY payment_qr_data TEXT NULL")