Move invite uuid generation into the model

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']
 

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