xiaoxia-saas/alembic/versions/069_project_is_default_idempotent.py

"""Projects is_default + partial unique index for idempotent default project (Issue #1775)

Revision ID: 069_project_is_default
Revises: 068_user_profile_completed
Create Date: 2026-09-08

背景：
小程序端 getOrCreateDefaultProject 在重试/并发/前端重复调用下，
仅靠应用层"先查再插"不保证幂等，会给同一用户重复创建默认项目。

改动：
1. projects 表新增 is_default 布尔列（默认 false）
2. 部分唯一索引 uq_projects_owner_default：(owner_user_id) WHERE is_default = true
   —— 保证每个用户至多一个默认项目
3. 存量数据回填：把名为"默认项目"的存量项目按创建时间最早者标记为 is_default=true
   （只标记不删除；存量重复项目的清理另行确认后单独执行）

注意：部分唯一索引依赖 PostgreSQL，不支持 downgrade 到其他方言。
"""

import sqlalchemy as sa

from alembic import op

revision = "069_project_is_default"
down_revision = "068_user_profile_completed"
branch_labels = None
depends_on = None


def upgrade() -> None:
    # 1. 新增 is_default 列
    op.add_column(
        "projects",
        sa.Column(
            "is_default",
            sa.Boolean(),
            nullable=False,
            server_default=sa.text("false"),
        ),
    )

    # 2. 存量回填：每个拥有"默认项目"的用户，只把最早创建的那一个标记为默认。
    #    用 ROW_NUMBER() 取每组第一条；非"默认项目"命名的项目不标记（保守，不动用户自建项目）。
    op.execute("""
        UPDATE projects p
        SET is_default = true
        WHERE p.id IN (
            SELECT id FROM (
                SELECT id,
                       ROW_NUMBER() OVER (
                           PARTITION BY owner_user_id
                           ORDER BY created_at ASC, id ASC
                       ) AS rn
                FROM projects
                WHERE name = '默认项目'
            ) t
            WHERE t.rn = 1
        )
        """)

    # 3. 部分唯一索引：每用户至多一个默认项目（只约束 is_default = true 的行）
    op.execute("""
        CREATE UNIQUE INDEX uq_projects_owner_default
        ON projects (owner_user_id)
        WHERE is_default = true
        """)


def downgrade() -> None:
    op.execute("DROP INDEX IF EXISTS uq_projects_owner_default")
    op.drop_column("projects", "is_default")