"""2026070901_add_credits_ratio_增加上传图片积分规则

Revision ID: 2026070901
Revises: 6c1aaf036f43
Create Date: 2026-07-01 00:00:00.000000

该文件包含 2026-07-01 的数据库迁移内容：
1. 积分规则表增加传入视频计费字段（input_video_ratio, input_video_base_credits, input_video_per_second_credits）
"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


revision: str = '2026070901'
down_revision: Union[str, None] = '6c1aaf036f43'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # ========================================
    # 2026-07-01 - 积分规则表增加传入图片计费字段
    # ========================================
    op.add_column('credit_ratios', sa.Column('input_image_ratio', sa.Float(), server_default='1.0', nullable=False))
    op.add_column('credit_ratios', sa.Column('input_image_base_credits', sa.Float(), server_default='0.0', nullable=False))
    op.add_column('credit_ratios', sa.Column('input_image_per_image_credits', sa.Float(), server_default='0.5', nullable=False))


def downgrade() -> None:
    # ========================================
    # 2026-07-01 - 积分规则表增加传入图片计费字段（回滚）
    # ========================================
    op.drop_column('credit_ratios', 'input_image_per_image_credits')
    op.drop_column('credit_ratios', 'input_image_base_credits')
    op.drop_column('credit_ratios', 'input_image_ratio')