ffmpeg video cover add file

2026-06-02 14:28:26 +08:00
parent 06d583c9f7
commit a2d7a59182
16 changed files with 284 additions and 48 deletions
@@ -17,23 +17,101 @@ branch_labels: Union[str, Sequence[str], None] = None
 depends_on: Union[str, Sequence[str], None] = None


+TABLE_NAME = "credit_ratios"
+
+
+def _has_index(table_name: str, index_name: str) -> bool:
+    bind = op.get_bind()
+    inspector = sa.inspect(bind)
+    indexes = inspector.get_indexes(table_name)
+    return any(index.get("name") == index_name for index in indexes)
+
+
+def _has_foreign_key(table_name: str, fk_name: str) -> bool:
+    bind = op.get_bind()
+    inspector = sa.inspect(bind)
+    foreign_keys = inspector.get_foreign_keys(table_name)
+    return any(fk.get("name") == fk_name for fk in foreign_keys)
+
+
+def _create_index_if_not_exists(
+    index_name: str,
+    table_name: str,
+    columns: list[str],
+    unique: bool = False,
+) -> None:
+    if not _has_index(table_name, index_name):
+        op.create_index(index_name, table_name, columns, unique=unique)
+
+
+def _drop_index_if_exists(index_name: str, table_name: str) -> None:
+    if _has_index(table_name, index_name):
+        op.drop_index(index_name, table_name=table_name)
+
+
 def upgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_index(op.f('ix_credit_ratios_gen_type'), 'credit_ratios', ['gen_type'], unique=False)
-    op.create_index('ix_credit_ratios_gen_type_engine_resolution', 'credit_ratios', ['gen_type', 'model_config_id', 'resolution'], unique=False)
-    op.create_index('ix_credit_ratios_gen_type_resolution', 'credit_ratios', ['gen_type', 'resolution'], unique=False)
-    op.create_index(op.f('ix_credit_ratios_model_config_id'), 'credit_ratios', ['model_config_id'], unique=False)
-    op.create_index(op.f('ix_credit_ratios_resolution'), 'credit_ratios', ['resolution'], unique=False)
-    op.drop_constraint(op.f('credit_ratios_model_config_id_fkey'), 'credit_ratios', type_='foreignkey')
-    # ### end Alembic commands ###
+    # 索引存在才不重复创建，避免 DuplicateTable
+    _create_index_if_not_exists(
+        "ix_credit_ratios_gen_type",
+        TABLE_NAME,
+        ["gen_type"],
+        unique=False,
+    )
+
+    _create_index_if_not_exists(
+        "ix_credit_ratios_gen_type_engine_resolution",
+        TABLE_NAME,
+        ["gen_type", "model_config_id", "resolution"],
+        unique=False,
+    )
+
+    _create_index_if_not_exists(
+        "ix_credit_ratios_gen_type_resolution",
+        TABLE_NAME,
+        ["gen_type", "resolution"],
+        unique=False,
+    )
+
+    _create_index_if_not_exists(
+        "ix_credit_ratios_model_config_id",
+        TABLE_NAME,
+        ["model_config_id"],
+        unique=False,
+    )
+
+    _create_index_if_not_exists(
+        "ix_credit_ratios_resolution",
+        TABLE_NAME,
+        ["resolution"],
+        unique=False,
+    )
+
+    # 外键存在才删除，避免 UndefinedObject
+    if _has_foreign_key(TABLE_NAME, "credit_ratios_model_config_id_fkey"):
+        op.drop_constraint(
+            "credit_ratios_model_config_id_fkey",
+            TABLE_NAME,
+            type_="foreignkey",
+        )


 def downgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    op.create_foreign_key(op.f('credit_ratios_model_config_id_fkey'), 'credit_ratios', 'model_configs', ['model_config_id'], ['id'])
-    op.drop_index(op.f('ix_credit_ratios_resolution'), table_name='credit_ratios')
-    op.drop_index(op.f('ix_credit_ratios_model_config_id'), table_name='credit_ratios')
-    op.drop_index('ix_credit_ratios_gen_type_resolution', table_name='credit_ratios')
-    op.drop_index('ix_credit_ratios_gen_type_engine_resolution', table_name='credit_ratios')
-    op.drop_index(op.f('ix_credit_ratios_gen_type'), table_name='credit_ratios')
-    # ### end Alembic commands ###
+    # 注意：
+    # 如果 model_config_id 已经改成 ImageEngine / VideoEngine 的 id，
+    # 恢复到 model_configs 外键可能因为历史数据不匹配而失败。
+    # 当前业务是删除外键，所以 downgrade 这里只做尽量安全处理。
+
+    if not _has_foreign_key(TABLE_NAME, "credit_ratios_model_config_id_fkey"):
+        op.create_foreign_key(
+            "credit_ratios_model_config_id_fkey",
+            TABLE_NAME,
+            "model_configs",
+            ["model_config_id"],
+            ["id"],
+        )
+
+    _drop_index_if_exists("ix_credit_ratios_resolution", TABLE_NAME)
+    _drop_index_if_exists("ix_credit_ratios_model_config_id", TABLE_NAME)
+    _drop_index_if_exists("ix_credit_ratios_gen_type_resolution", TABLE_NAME)
+    _drop_index_if_exists("ix_credit_ratios_gen_type_engine_resolution", TABLE_NAME)
+    _drop_index_if_exists("ix_credit_ratios_gen_type", TABLE_NAME)