diff --git a/profiles/models.py b/profiles/models.py index 74dea47..f55c7a1 100644 --- a/profiles/models.py +++ b/profiles/models.py @@ -1,6 +1,7 @@ from django.db import models from django.contrib.auth.models import User from django.db.models.signals import post_save +import shortuuid class Invite(models.Model): code = models.CharField(max_length=30) @@ -11,6 +12,11 @@ class Invite(models.Model): def __unicode__(self): return self.code + def save(self, *args, **kwargs): + if not self.code: + self.code = shortuuid.uuid()[:6].upper() + super(Invite, self).save(*args, **kwargs) + class Meta: ordering = ['deleted'] diff --git a/profiles/views.py b/profiles/views.py index c63f6d7..b95976e 100644 --- a/profiles/views.py +++ b/profiles/views.py @@ -10,7 +10,6 @@ from django.contrib.auth import authenticate, login from django.core.exceptions import ObjectDoesNotExist import forms import models -import shortuuid from minecraft.forms import ProfileForm @login_required @@ -39,7 +38,6 @@ def invites(request): def createInvite(request): invite = models.Invite() invite.creator = request.user - invite.code = shortuuid.uuid()[:6].upper() invite.save() return HttpResponseRedirect(reverse('profiles.views.invites'))