xiaoxia-saas/alembic/versions/093_viral_video_pricing_points_float.py

"""viral_video 动态积分定价 + 积分字段从 Integer 改为 Float (#2151)

Revision ID: 093
Revises: 092_viral_video_heartbeat
Create Date: 2026-10-02
"""

import sqlalchemy as sa

from alembic import op

revision = "093"
down_revision = "092_viral_video_heartbeat"
branch_labels = None
depends_on = None


def upgrade() -> None:
    conn = op.get_bind()
    inspector = sa.inspect(conn)

    # 1) points_accounts 三列 Integer -> Float
    pa_cols = {c["name"]: c for c in inspector.get_columns("points_accounts")}
    for col in ("balance", "total_earned", "total_spent"):
        if col in pa_cols:
            op.alter_column(
                "points_accounts",
                col,
                existing_type=sa.Integer(),
                type_=sa.Float(),
                existing_nullable=False,
            )

    # 2) points_transactions amount/balance_after Integer -> Float
    pt_cols = {c["name"]: c for c in inspector.get_columns("points_transactions")}
    for col in ("amount", "balance_after"):
        if col in pt_cols:
            op.alter_column(
                "points_transactions",
                col,
                existing_type=sa.Integer(),
                type_=sa.Float(),
                existing_nullable=False,
            )

    # 3) users.points_balance Integer -> Float
    user_cols = {c["name"]: c for c in inspector.get_columns("users")}
    if "points_balance" in user_cols:
        op.alter_column(
            "users",
            "points_balance",
            existing_type=sa.Integer(),
            type_=sa.Float(),
            existing_nullable=False,
        )

    # 4) viral_video_jobs.credits_cost Integer -> Float
    vv_cols = {c["name"]: c for c in inspector.get_columns("viral_video_jobs")}
    if "credits_cost" in vv_cols:
        op.alter_column(
            "viral_video_jobs",
            "credits_cost",
            existing_type=sa.Integer(),
            type_=sa.Float(),
            existing_nullable=False,
        )

    # 5) viral_video_jobs 新增列
    if "video_resolution" not in vv_cols:
        op.add_column(
            "viral_video_jobs",
            sa.Column("video_resolution", sa.String(20), nullable=False, server_default="720p"),
        )
    if "credits_prepaid" not in vv_cols:
        op.add_column(
            "viral_video_jobs",
            sa.Column("credits_prepaid", sa.Float(), nullable=False, server_default="0"),
        )
    if "credits_transaction_id" not in vv_cols:
        op.add_column(
            "viral_video_jobs",
            sa.Column("credits_transaction_id", sa.String(36), nullable=False, server_default=""),
        )


def downgrade() -> None:
    pass