创建对象后将登录的用户附加到对象

时间:2019-06-29 13:37:17

标签: python django django-models refactoring

我正在尝试让一个已登录的用户填写一个表格来创建一个组。创建网上论坛时,我需要将用户自动添加到网上论坛。

对于此问题,我们正在使用两个模型-用户和组。

用户是Django提供的默认模型。

组的定义如下:

renderItem={() =>this.renderItem(library)}

我要重构的视图是:

class Group(models.Model):
    name = models.CharField(max_length=255, unique=True)
    admins = models.ManyToManyField(User, default=1, related_name='user_username')
    all_users = models.ManyToManyField(User, default=1)

    def __str__(self):
        return self.name

    def get_absolute_url(self):
        return reverse('home')

    def get_admins(self):
        return ", ".join([u.username for u in self.admins.all()])

    def add_admin(self, user):
        self.admins.add(user)

    def get_all_users(self):
        return ", ".join([u.username for u in self.all_users.all()])

    def add_user(self, user):
        self.all_users.add(user)
        self.save()

    def is_admin(self, user):
        if user in self.admins.all():
            return True
        else:
            return False

这是GroupForm:

@login_required
def user_generated_group(request):
    if request.method == 'POST':
        form = GroupForm(request.POST)
        user = request.user
        if form.is_valid():
            group = Group.objects.create(name=form.cleaned_data['name'])
            group.add_admin(user)
            group.add_user(user)
            group.save()
        return HttpResponseRedirect(reverse('home'))
    else:
        form = GroupForm()
        context = {
            'form': form,
            'type': 'group',
            'sidebar': Sidebar(request),
        }
        return render(request, 'form.html', context)

目标是利用Django的内置CreateView。到目前为止,重构后的视图看起来像:

class GroupForm(forms.ModelForm):
    class Meta:
        model = Group
        fields = '__all__'
        exclude = ['all_users', 'admins', ]

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields['name'].widget.attrs.update({'class': 'input'})

我还没有实现此视图的add_user和add_admin逻辑。希望我可以在其他地方管理这些方法。

我不确定从这里去哪里。这种逻辑应该由用户还是由组来处理?

如果是按小组,我应该使用经理吗?

如果由用户使用,我应该创建自定义用户模型吗?

1 个答案:

答案 0 :(得分:1)

您可以覆盖form_valid [Django-doc]方法,以将self.request.user添加到组中,例如:

from django.http import HttpResponseRedirect

class CreateGroup(LoginRequiredMixin, CreateView):
    model = Group
    form_class = GroupForm
    template_name = 'form.html'

    def form_valid(self, form):
        self.object = group = form.save()
        group.all_users.add(self.request.user)
        return HttpResponseRedirect(self.get_success_url())