无法通过SQLAlchemy语句修改PostgreSQL数据库Data行，请求协助

原代码存在的核心问题 #

1. archived字段处理逻辑错误：当custom_columns中不存在archived字段时，models.Data.custom_columns['archived']会返回null，此时用||与新构建的JSON对象合并会得到null，导致jsonb_set将archived设为null，而非预期的包含归档键值的对象。
2. 冗余的coalesce包裹：jsonb_set即使操作路径不存在，也会返回原JSONB对象，不会返回null，因此外层的coalesce完全多余。
3. 未启用路径创建：第一个jsonb_set未指定第三个参数为True，默认情况下，当archived路径不存在时，不会创建该字段，直接返回原对象。
4. 冗余类型转换：重复对custom_columns进行JSONB类型转换，增加代码复杂度。

修正后的SQLAlchemy代码 #

from sqlalchemy import func, cast, update
from sqlalchemy.dialects.postgresql import JSONB

stmt = (
    update(models.Data)
    .values(
        custom_columns=(
            func.jsonb_set(
                # 第一步：处理归档字段，不存在则创建并添加键值，存在则追加
                func.jsonb_set(
                    cast(models.Data.custom_columns, JSONB),
                    '{archived}',
                    # 确保archived字段不存在时用空对象，避免合并后为null
                    func.coalesce(
                        cast(models.Data.custom_columns['archived'], JSONB),
                        func.jsonb_build_object()
                    ).op('||')(
                        func.jsonb_build_object(
                            old_key,
                            cast(models.Data.custom_columns['fields'][old_key], JSONB)
                        )
                    ),
                    True  # 允许创建不存在的archived路径
                ),
                # 第二步：从fields中删除指定key
                '{fields}',
                cast(models.Data.custom_columns, JSONB)['fields'].op('-')(old_key)
            )
        ),
        updated_at=func.now()
    )
    .where(
        cast(models.Data.custom_columns, JSONB)['fields'].has_key(old_key)
    )
    .where(models.Data.project_id == project_id)
)

关键改进说明 #

• 修复归档字段初始化：用func.coalesce(..., func.jsonb_build_object())保证archived字段不存在时，以空JSON对象为基础合并归档键值，避免null导致的合并失败。
• 启用路径创建：第一个jsonb_set的第三个参数设为True，让PostgreSQL自动创建不存在的archived字段。
• 简化逻辑：移除多余的coalesce包裹，减少冗余类型转换，代码更简洁易读。

验证建议 #

可以先在PostgreSQL中执行对应的原生SQL验证逻辑是否正确，再同步调整SQLAlchemy代码：

UPDATE data
SET custom_columns = jsonb_set(
    jsonb_set(
        custom_columns::jsonb,
        '{archived}',
        coalesce(custom_columns->'archived', '{}'::jsonb) || jsonb_build_object('old_key', custom_columns->'fields'->'old_key'),
        true
    ),
    '{fields}',
    custom_columns->'fields' - 'old_key'
), updated_at = now()
WHERE custom_columns::jsonb->'fields' ? 'old_key' AND project_id = 'your_project_id';

内容的提问来源于stack exchange，提问作者Muhammad Khalid