Django - 从数据库中的表单中保存数据

时间:2016-04-17 11:22:31

标签: python django django-models django-forms django-views

我刚刚完成了一个电子邮件注册表单的Django教程,我正在尝试调整代码的某些部分以满足我的需求(在注册表单中添加一个额外的字段)

根据教程,我创建了一个电子邮件注册模型表单:

class JoinForm(forms.ModelForm):
    class Meta:
        model = Join
        fields = ["email","wunschrad"]

我通过添加wunschrad字段

对其进行了一些修改

然后我更新了我的模型以包含wunschrad并将数据库与南方同步,所有工作都很好。低于我的模特:

class Join(models.Model):
    email = models.EmailField()
    wunschrad = models.CharField(max_length=120)
    friend = models.ForeignKey("self", related_name='referral', null=True, blank=True)
    ref_id = models.CharField(max_length=120, default='ABC', unique=True)
    ip_address = models.CharField(max_length=120, default='ABC')
    timestamp = models.DateTimeField(auto_now_add = True, auto_now = False)
    updated = models.DateTimeField(auto_now_add = False, auto_now = True)

    def __unicode__(self):
        return "%s" % (self.email)

    class Meta:
        unique_together = ("email", "ref_id",)

我遇到问题的方法是如何采用我的观点来反映这一变化。这是我做的(我的更改在评论中突出显示)

def home(request):
    try:
        join_id = request.session['join_id_ref']
        obj = Join.objects.get(id=join_id)
    except:
        obj = None

    form = JoinForm(request.POST or None)
    if form.is_valid():
        new_join = form.save(commit=False)
        email = form.cleaned_data['email']
        wunschrad = form.cleaned_data['wunschrad'] #I ADDED THIS
        new_join_old, created = Join.objects.get_or_create(email=email, wunschrad=wunschrad) # I ADDED WUNSCHRAD HERE
        if created:
            new_join_old.ref_id = get_ref_id()
            if not obj == None:
                new_join_old.friend = obj
            new_join_old.ip_address = get_ip(request)
            new_join_old.save()

        return HttpResponseRedirect("/%s" %(new_join_old.ref_id))


    context = {"form": form}
    template = "home.html"
    return render(request, template, context)

我的问题/问题如下:此代码段有效但如果用户使用相同的电子邮件地址注册并选择wunschrad的第一个选项,第二次选择第二个选项(澄清:wunschrad是一个drop-向下列表有两个选项)Django在数据库中保存了两个版本,如下所示:Django Admin Screenshot

有没有人知道如何更改代码以便每个用户只保存一次?

许多人提前感谢您的帮助!

1 个答案:

答案 0 :(得分:0)

所以你的问题是通过向get_or_create添加wunschrad,你告诉Django在数据库中查找具有这两个值而不仅仅是电子邮件的条目。将其更改为此应修复它:

new_join_old, created = Join.objects.get_or_create(email=email)
new_join_old.wunschrad = wunschrad
new_join_old.save()

如果电子邮件已经不在数据库中,这将确保它只创建一个新对象。

作为旁注,如果您希望您的电子邮件是唯一的,并且如果您尝试使用相同的电子邮件保存另一个条目,则会引发错误,将unique = True添加到models.EmailField(),如:

models.EmailField(unique=True)