SQLAlchemy 在 postgres 中添加/更新 jsonb 字段

colebod209 发表于 Dev

colebod209

我正在尝试使用 SQLAlchemy 更新 postgres 数据库中 jsonb 字段内的值。

一直在尝试使用func.jsonb_set，但我不太清楚如何实现它。

使用如下表 ( test)，我希望找到一种添加/编辑 json 数据的通用方法。

ID	数据	姓名
1	{“年龄”：44，“名字”：“巴里”，孩子”：[“baz”，“吉姆”]}	巴里
2	{“年龄”：47，“姓名”：“戴夫”，“孩子”：[“杰夫”，“简”]}	戴夫

以下适用于 postgres 的简单更新。

UPDATE "test" SET "data"=jsonb_set("data"::jsonb, '{age}', '45')
WHERE "data"::json->>'name'='dave';

我可以用来update更新这样的单个值：

testobj_res.update(
    {
        TestObj.data: cast(
            cast(TestObj.data, JSONB).concat(func.jsonb_build_object("age", 45)),
            JSON,
        )
    }
)

session.commit()

我希望能够传递例如多个字段的更新{"name": "barry", "age": 45, "height": 150}。

我尝试使用func.jsonb_set添加更复杂的 json 结构而不是('age', 45)

testobj_res.first().data = func.jsonb_set(
    TestObj.data.cast(JSONB),
    ("age", 45),
    cast(TestObj.data, JSONB))

session.commit()

但我得到：

sqlalchemy.exc.ProgrammingError: (psycopg2.errors.UndefinedFunction) function jsonb_set(jsonb, record, jsonb) does not exist
LINE 1: UPDATE public.test SET data=jsonb_set(CAST(public.test.data ...
                                    ^
HINT:  No function matches the given name and argument types. You might need to add explicit type casts.

完整示例代码：

import os
from sqlalchemy.dialects.postgresql import JSON, JSONB
from sqlalchemy import func, cast
import sqlalchemy as sa
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
import urllib
from dotenv import load_dotenv

load_dotenv()

user = urllib.parse.quote_plus(os.environ.get("DB_USER"))
passwd = urllib.parse.quote_plus(os.environ.get("DB_PW"))

DB_URL = "postgresql://{}:{}@{}:{}/{}".format(
    user,
    passwd,
    os.environ.get("DB_HOST"),
    os.environ.get("DB_PORT"),
    os.environ.get("DB_NAME"),
)

engine = sa.create_engine(DB_URL)
Session = sessionmaker(bind=engine, autoflush=True)

session = Session()

Base = declarative_base()


class TestObj(Base):
    __tablename__ = "test"
    __table_args__ = {"autoload_with": engine, "schema": "public"}


testobj_res = session.query(TestObj).filter(TestObj.name == "dave")

testobj_res.first().data = func.jsonb_set(
    TestObj.data.cast(JSONB),
    ("age", 45),
    cast(TestObj.data, JSONB))

session.commit()

蛇符

这段代码

testobj_res.first().data = func.jsonb_set(
    TestObj.data.cast(JSONB),
    ("age", 45),
    cast(TestObj.data, JSONB))

正在传递一个 JSONB 对象、记录和另一个 JSONB 对象，jsonb_set但该函数需要一个 JSONB 对象、一个 JSONPath 字符串和路径的新值 JSONB ^*。

解('age', 45)包元组并删除最终的 JSONB 将产生所需的结果。

with Session.begin() as s:
    testobj = s.scalar(sa.select(TestObj).limit(1))
    testobj.data = sa.func.jsonb_set(
        sa.cast(testobj.data, postgresql.JSONB),
        '{age}',
        sa.cast(42, postgresql.JSONB)
    )

^*还有一个可选的布尔 create-if-not-exists 参数，我们可以忽略它。

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。