diff --git a/pinax/invitations/migrations/0001_initial.py b/pinax/invitations/migrations/0001_initial.py index 8755d48..a1143d8 100644 --- a/pinax/invitations/migrations/0001_initial.py +++ b/pinax/invitations/migrations/0001_initial.py @@ -21,7 +21,7 @@ class Migration(migrations.Migration): ('invites_sent', models.IntegerField(default=0)), ('invites_allocated', models.IntegerField(default=0)), ('invites_accepted', models.IntegerField(default=0)), - ('user', models.OneToOneField(to=settings.AUTH_USER_MODEL)), + ('user', models.OneToOneField(to=settings.AUTH_USER_MODEL, on_delete=models.CASCADE)), ], options={ }, @@ -34,9 +34,9 @@ class Migration(migrations.Migration): ('message', models.TextField(null=True)), ('sent', models.DateTimeField(default=django.utils.timezone.now)), ('status', models.IntegerField(choices=[(1, b'Sent'), (2, b'Accepted'), (3, b'Joined Independently')])), - ('from_user', models.ForeignKey(related_name='invites_sent', to=settings.AUTH_USER_MODEL)), - ('signup_code', models.OneToOneField(to='account.SignupCode')), - ('to_user', models.ForeignKey(related_name='invites_received', to=settings.AUTH_USER_MODEL, null=True)), + ('from_user', models.ForeignKey(related_name='invites_sent', to=settings.AUTH_USER_MODEL, on_delete=models.CASCADE)), + ('signup_code', models.OneToOneField(to='account.SignupCode', on_delete=models.CASCADE)), + ('to_user', models.ForeignKey(related_name='invites_received', to=settings.AUTH_USER_MODEL, on_delete=models.CASCADE, null=True)), ], options={ }, diff --git a/pinax/invitations/models.py b/pinax/invitations/models.py index a31bf86..e065539 100644 --- a/pinax/invitations/models.py +++ b/pinax/invitations/models.py @@ -25,16 +25,21 @@ class JoinInvitation(models.Model): (STATUS_JOINED_INDEPENDENTLY, "Joined Independently") ] - from_user = models.ForeignKey(settings.AUTH_USER_MODEL, related_name="invites_sent") + from_user = models.ForeignKey( + settings.AUTH_USER_MODEL, + on_delete=models.CASCADE, + related_name="invites_sent", + ) to_user = models.ForeignKey( settings.AUTH_USER_MODEL, + on_delete=models.CASCADE, null=True, related_name="invites_received" ) message = models.TextField(null=True) sent = models.DateTimeField(default=timezone.now) status = models.IntegerField(choices=INVITE_STATUS_CHOICES) - signup_code = models.OneToOneField(SignupCode) + signup_code = models.OneToOneField(SignupCode, on_delete=models.CASCADE) def to_user_email(self): return self.signup_code.email @@ -92,7 +97,7 @@ def send_invite(*args, **kwargs): class InvitationStat(models.Model): - user = models.OneToOneField(settings.AUTH_USER_MODEL) + user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE) invites_sent = models.IntegerField(default=0) invites_allocated = models.IntegerField( default=settings.PINAX_INVITATIONS_DEFAULT_INVITE_ALLOCATION diff --git a/pinax/invitations/urls.py b/pinax/invitations/urls.py index 5b9400c..14439ea 100644 --- a/pinax/invitations/urls.py +++ b/pinax/invitations/urls.py @@ -9,6 +9,7 @@ TopOffUserView ) +app_name = "pinax_invitations" urlpatterns = [ url(r"^invite/$", InviteView.as_view(), name="invite"),