如何使用 Django 的 ORM 使用 .update() 方法更新 JsonField 中多个键的值？

我有一个模型字段，我正在尝试使用 django 的 .update() 方法进行更新。该字段是一个 jsonfield。我通常会在这个领域保存一本字典。例如{"test": "是", "prod": "否"}

这是模型：

class question(models.model):
    # {"test": "yes", "prod": "no"}
    extra_data = models.jsonfield(default=dict, null=true, blank=true)

我可以使用此查询更新字典内的键（顺便说一句，效果很好）：

Question.filter(id="123").update(meta_data=Func(
                        F("extra_data"),
                        Value(["test"]),
                        Value("no", JSONField()),
                        function="jsonb_set",
                    ))

现在的问题是，有没有办法可以使用 .update() 一次更新多个密钥（在我自己的情况下是两个），如上面的查询所示？

我想使用 .update() 方法而不是 .save() 来实现此目的，这样我就可以避免调用 post_save 信号函数。

正确答案

免责声明：它看起来会很难看。

我过去曾这样做过，尽管是使用纯 sql，而不是使用 django。这个想法是递归调用jsonb_set()。每次调用都会设置一个密钥。

Kacha

KaCha是一款革命性的AI写真工具，用AI技术将照片变成杰作！

下载

因此，在 sql 中，要设置键 {"test":"yes","prod":"no"} 你可以这样做：

update question set meta_data = jsonb_set(jsonb_set(extra_data, '{test}', '"yes"'::jsonb), '{prod}', '"no"'::jsonb)
where id = 123;

请注意，jsonb_set有两个嵌套调用，最里面的使用meta_data，最外面的接收最里面的结果。

所以 django 的等价物看起来像这样（注意，未经测试）：

question.filter(id="123").update(
    meta_data=func(
        func(
            f("extra_data"),
            value(["test"]),
            value("yes", jsonfield()),
            function="jsonb_set",
        ),
        value(["prod"]),
        value("no", jsonfield()),
        function="jsonb_set",
    ))

或者您可以创建一个递归函数，为您返回混乱的情况，一次设置一项：

def recursive_jsonb_set(original, **kwargs):
    if kwargs:
        key, value = kwargs.popitem()
        return Func(
            recursive_jsonb_set(original, **kwargs),
            Value(key.split('__')),
            Value(value, JSONField()),
            function="jsonb_set")
    else:
        return original

Question.filter(id="123").update(
    meta_data=recursive_jsonb_set(F("extra_data"), test="yes", prod="no", other__status="done"))

请注意 '__' 功能以更新子项。