在 Django 中,如何立即将组的创建者添加到该组?



我正在Django中创建一个'社交'应用程序,用户可以在其中创建组(联盟(,其他人可以加入这些组。用户配置文件联盟通过成员模型连接。我希望这个小组的创建者立即成为它的成员。

我对 Django 缺乏经验,事实上,这是我的第一个项目。但我想这也许可以通过信号来解决?

我的用户配置文件模型:

class Profile(models.Model):
...
user = models.OneToOneField(User, on_delete=models.CASCADE)
alliances = models.ManyToManyField('Alliance', through='Membership')
...

我的联盟模式:

class Alliance(models.Model):
...
name = models.CharField(max_length=10, unique=True)
members = models.ManyToManyField('Profile', through='Membership')
...

我的会员模式:

class Membership(models.Model):
...
profile = models.ForeignKey('Profile', on_delete=models.CASCADE)
alliance = models.ForeignKey('Alliance', on_delete=models.CASCADE)
...

我认为可行的解决方案(使用信号(如下所示:

@receiver(post_save, sender=Alliance)
def create_membership(sender, instance, created, **kwargs):
if created:
Membership.objects.create(profile=???, alliance=instance)

其中"???"应该是创作者个人资料。我会很高兴得到任何帮助。

信号无法访问触发它的请求,因为首先并不是说触发器是HTTP请求。您可以尝试检查回溯,但这是一个黑客解决方案,可能会产生比它的价值更多的麻烦。

您需要在创建组的窗体视图中执行此操作。例如,在视图中,您可以添加以下内容:

from django.views.generic import CreateView
from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import HttpResponseRedirect
from app.models import Alliance, Membership
class AllianceCreateView(LoginRequiredMixin, CreateView):
model = Alliance
def form_valid(self, form):
self.object = form.save()
Membership.objects.create(
profile=self.request.user.profile,
alliance=self.object
)
return HttpResponseRedirect(self.get_success_url())

请注意,建模很奇怪。通常,您不会指定两个ManyToManyField。Django 将自动添加第二个。因此,您可以在Profile中指定:

class Profile(models.Model):
# ...
user = models.OneToOneField(User, on_delete=models.CASCADE)
alliances = models.ManyToManyField(
'Alliance',
through='Membership',
related_name='members'
)

最新更新