卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章34173本站已运行391

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

如何使用 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()。每次调用都会设置一个密钥。

因此,在 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"))

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

卓越飞翔博客
上一篇: go install 在 dockerfile 中需要花费很多时间
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏