我正在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'
)