陈世睿
060cb74a84
队友这条迁移只在 PG 测过:server_default 用了 PG 专属 ::jsonb(SQLite 报 unrecognized token), 且 op.alter_column 改可空/类型 SQLite 原生不支持。改成:platforms default 分方言(PG 仍 ::jsonb、 SQLite 用纯 JSON 字面量)+ 所有 alter_column 包进 op.batch_alter_table(SQLite 重建表、PG 等价直接 ALTER)。 线上 PG 行为不变,本地 SQLite dev 能正常 upgrade。 Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
76 lines
3.4 KiB
Python
76 lines
3.4 KiB
Python
"""cps v2: 活动 payload(淘口令/京东链接) + 群多平台/sid 可空 + 点击 event_type
|
|
|
|
Revision ID: cps_v2_platforms
|
|
Revises: cps_link_tables
|
|
Create Date: 2026-06-17 11:00:00.000000
|
|
|
|
接入淘宝/京东:
|
|
- cps_activity.payload 淘宝整段淘口令文本 / 京东推广链接(美团仍用 act_id)
|
|
- cps_group.platforms 该群发哪些平台(多选);sid 改可空(纯淘宝/京东无 sid)
|
|
- cps_link.sid 可空 + target_url 加长(淘口令较长)
|
|
- cps_click.event_type visit / copy(淘宝落地页"复制口令")
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
|
|
revision: str = "cps_v2_platforms"
|
|
down_revision: Union[str, Sequence[str], None] = "cps_link_tables"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
# ⚠️ SQLite 不支持 ALTER COLUMN(改可空/类型),必须用 op.batch_alter_table(重建表)模拟;
|
|
# env.py 的 render_as_batch 只管 autogenerate 渲染、不会让"手写的 op.alter_column"在运行时自动 batch。
|
|
# batch_alter_table 在 Postgres 上等价于直接 ALTER(recreate=auto 不重建),两端通用。
|
|
def upgrade() -> None:
|
|
# 活动:淘宝淘口令 / 京东链接(纯 add column,SQLite 直接支持)
|
|
op.add_column("cps_activity", sa.Column("payload", sa.Text(), nullable=True))
|
|
|
|
# 群:多平台(现有群都是美团,server_default 回填)+ sid 可空。
|
|
# server_default 分方言:Postgres 用 ::jsonb 显式转;SQLite 不认 ::jsonb(报 unrecognized token ":"),
|
|
# 用纯 JSON 字符串字面量即可(JSON 列在 SQLite 实际是 TEXT)。
|
|
platforms_default = (
|
|
"'[\"meituan\"]'::jsonb"
|
|
if op.get_bind().dialect.name == "postgresql"
|
|
else "'[\"meituan\"]'"
|
|
)
|
|
with op.batch_alter_table("cps_group") as batch_op:
|
|
batch_op.add_column(
|
|
sa.Column(
|
|
"platforms",
|
|
sa.JSON().with_variant(postgresql.JSONB(), "postgresql"),
|
|
nullable=False,
|
|
server_default=sa.text(platforms_default),
|
|
)
|
|
)
|
|
batch_op.alter_column("sid", existing_type=sa.String(64), nullable=True)
|
|
|
|
# link:sid 可空 + target 加长
|
|
with op.batch_alter_table("cps_link") as batch_op:
|
|
batch_op.alter_column("sid", existing_type=sa.String(64), nullable=True)
|
|
batch_op.alter_column("target_url", existing_type=sa.String(1024), type_=sa.String(2048))
|
|
|
|
# click:sid 可空 + 事件类型
|
|
with op.batch_alter_table("cps_click") as batch_op:
|
|
batch_op.alter_column("sid", existing_type=sa.String(64), nullable=True)
|
|
batch_op.add_column(
|
|
sa.Column("event_type", sa.String(16), nullable=False, server_default="visit")
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
with op.batch_alter_table("cps_click") as batch_op:
|
|
batch_op.drop_column("event_type")
|
|
batch_op.alter_column("sid", existing_type=sa.String(64), nullable=False)
|
|
with op.batch_alter_table("cps_link") as batch_op:
|
|
batch_op.alter_column("target_url", existing_type=sa.String(2048), type_=sa.String(1024))
|
|
batch_op.alter_column("sid", existing_type=sa.String(64), nullable=False)
|
|
with op.batch_alter_table("cps_group") as batch_op:
|
|
batch_op.alter_column("sid", existing_type=sa.String(64), nullable=False)
|
|
batch_op.drop_column("platforms")
|
|
op.drop_column("cps_activity", "payload")
|