优化斗鱼充值

2026-08-13 21:43:27 +08:00
parent 902e669614
commit 0bf69c1240
19 changed files with 450 additions and 266 deletions
@@ -0,0 +1,44 @@
+"""持久化斗鱼供应商直充外部订单号
+
+Revision ID: 20260813_0027
+Revises: 20260813_0026
+Create Date: 2026-08-13
+"""
+
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+revision: str = "20260813_0027"
+down_revision: Union[str, None] = "20260813_0026"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    bind = op.get_bind()
+    inspector = sa.inspect(bind)
+    if not inspector.has_table("douyu_tasks"):
+        return
+    columns = {column["name"] for column in inspector.get_columns("douyu_tasks")}
+    if "supplier_out_order_id" not in columns:
+        op.add_column("douyu_tasks", sa.Column("supplier_out_order_id", sa.String(length=64), nullable=True))
+        op.create_index(
+            "ix_douyu_tasks_supplier_out_order_id",
+            "douyu_tasks",
+            ["supplier_out_order_id"],
+            unique=True,
+        )
+
+
+def downgrade() -> None:
+    bind = op.get_bind()
+    inspector = sa.inspect(bind)
+    if not inspector.has_table("douyu_tasks"):
+        return
+    columns = {column["name"] for column in inspector.get_columns("douyu_tasks")}
+    if "supplier_out_order_id" in columns:
+        op.drop_index("ix_douyu_tasks_supplier_out_order_id", table_name="douyu_tasks")
+        op.drop_column("douyu_tasks", "supplier_out_order_id")