Question

我正在尝试创建一个简单的CSV导入表单，导入到我的模型GroupMembers中。 GroupMembers模型具有链接到我的Groups模型的多对多字段。您的想法是，您可以上传特定组的CSV成员资格文件;您可以从与CSV上载相同的页面上的下拉列表中选择一个组。成员可以是多个组的成员，因此如果该成员已存在，则应该只添加新的组关系。

我的代码部分工作 - 它添加了组成员 - 但我遇到了与组的多对多链接的问题。这是我到目前为止的代码

models.py（group）

class Group (models.Model):
    initials = models.CharField(max_length = 20)
    initials_lower = models.CharField(max_length = 20)
    full_name = models.CharField('Full name', max_length = 100)

models.py（groupmember） - 这是csv添加到

的地方

from django.db import models
from datetime import datetime

class GroupMember (models.Model):
    name = models.CharField('Name', max_length = 30, unique = True)
    pub_date = models.DateTimeField('Date published', auto_now_add = True)
    groups = models.ManyToManyField('groups.Group', blank = True)

    def __unicode__(self):
        return self.name

forms.py

from django import forms
import csv
from myproject.groupmembers.models import GroupMember
from myproject.groups.models import Group


class GroupImportForm (forms.Form):

    csv_file = forms.FileField()
    group_name = forms.ModelChoiceField(queryset=Group.objects.all())

    def save(self):
        records = csv.reader(self.cleaned_data["csv_file"])
        for line in records:
            input_data = GroupMember()
            input_data.handle = line[0]
            input_data.groups.create(initials=self.cleaned_data["group_name"]) <-- this is where the problem is
            input_data.save()

views.py（部分）

def batch_add (request):
    if request.method == 'POST':
        form = GroupImportForm(request.POST, request.FILES)
    if form.is_valid():
        form.save()
    else:
        form = GroupImportForm()

    return render_to_response(
        'members/batch_add.html',
        {
        'form': form,
        },
    context_instance=RequestContext(request)
    )

我上传时遇到的错误是'initials_lower'不存在，这意味着它正在尝试创建一个新的Group记录，我不希望它这样做。我只是希望它使用从select中选择的预先存在的组，并将其添加到多组到多个字段的组中。

希望有意义，任何帮助表示赞赏

Answer 1

使用get_or_create仅在目标尚不存在的情况下创建对象：

input_data.groups.get_or_create(initials=self.cleaned_data["group_name"])

CSV导入到具有多对多字段的模型

1 个答案: