Присоединение зарегистрированного пользователя к объекту после создания объекта

Question

Я пытаюсь, чтобы вошедший в систему пользователь заполнил форму для создания группы. При создании группы мне нужно, чтобы пользователь автоматически добавлялся в группу.

Для решения этой проблемы мы работаем с двумя моделями - User и Group.

Пользователь - модель по умолчанию, предоставленная Django.

Группа определена так:

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

И я пытаюсь провести рефакторинг:

@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)

Вот форма группы:

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'})

Цель состоит в том, чтобы использовать встроенный в Django CreateView. Реорганизованный вид выглядит так:

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

Мне еще предстоит реализовать логику add_user и add_admin для этого представления. Надеюсь, что я могу управлять этими методами в другом месте.

Я не уверен, куда идти отсюда. Должна ли эта логика обрабатываться пользователем или группой?

Если по Группе я должен использовать Менеджера?

Если пользователь, я должен создать пользовательскую модель пользователя?

Answer 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 <b>form_valid</b>(self, form):
        self.object = group = form.save()
        <b>group.all_users.add(self.request.user)</b>
        return HttpResponseRedirect(self.get_success_url())