1

video-gen-api/alembic/versions/2026070901_add_credits_ratio_增加上传图片积分规则.py

@@ -0,0 +1,37 @@
+"""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')