我有一个模型字段,我正在尝试使用 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"))
请注意 '__'
功能以更新子项。