基于ID字段创建或更新Django模型

时间:2016-08-02 17:57:31

标签: django

我正在尝试使用相同的表单来创建或更新Django模型。不幸的是,如果标识符为update_or_create(),则其他答案中建议的None方法不起作用。目前,我正在检查id是否为None,获得ModelForm,然后根据需要在单独的代码流中创建或更新对象。

这是我的代码:

# When the Conquest already exists
if c_id is not None:
    # Check if Conquest is valid, else throw error
    conq = get_object_or_404(Conquest,id=c_id)

    # Verify ModelForm data based on existing instance
    form = ConquestForm(request.POST,instance=conq)
    if not form.is_valid():
        print form.errors
        return json_response(messages='Invalid form data')

    conq.title=form.cleaned_data['title']
    conq.description=form.cleaned_data['description']
    conq.start_time=form.cleaned_data['start_time']
    conq.end_time=form.cleaned_data['end_time']
    conq.save()
    return JsonResponse({'success':'true','message':'Conquest updated'},status=200)

else:
    # Verify ModelForm data based
    form = ConquestForm(request.POST)    
    if not form.is_valid():
        print form.errors
        return json_response(messages='Invalid form data')

    conq = Conquest.objects.create(
            title=form.cleaned_data['title'], 
            description=form.cleaned_data['description'],
            start_time=form.cleaned_data['start_time'],
            end_time=form.cleaned_data['end_time'])

    return JsonResponse({'success':'true','message':'Conquest created'},status=200)

如何避免代码重复:

  1. 使用/不指定form来创建instance对象

  2. 在id null

  3. 的情况下创建和更新单独的代码

    如果我没有指定instance=conq,我会收到完整性错误,因为title必须是唯一的。

1 个答案:

答案 0 :(得分:2)

首先需要了解一些事项。使用ModelForm时,不要手动创建/更新实例。当您致电form.save()时,您会创建/更新。

如果向表单的实例参数提供None,它实际上会将当前表单视为创建表单,否则会将其作为参数instance的对象的更新表单处理。

如果您使用表单自动处理这些内容,则您将无法使用IntegrityError,因为表单无效且错误位于form.errors,您可以轻松将其返回给客户端以进行通知

这是粗略的实现,它可能无法开箱即用,但至少它应足以指导您需要做的事情:

def func(request, c_id):
    if c_id:
        conq = get_object_or_404(Conquest, id=c_id)
    else:
        conq = None

    form = ConquestForm(request.POST or None, instance=conq)
    if form.is_valid():
        # this could either be a creation or an update
        conquest = form.save()
        return JsonResponse({'success': true})
    else:
        errors = form.errors
        return JsonReponse({'success':false, 'errors': errors})

关于ModelForm的Django doc。