This commit is contained in:
Ilya Moroz 2017-05-23 02:38:50 +00:00 committed by GitHub
commit 74b314ea8e

View File

@ -31,7 +31,13 @@ class Token(models.Model):
def save(self, *args, **kwargs):
if not self.key:
self.key = self.generate_key()
while True:
generated_key = self.generate_key()
try:
self.objects.get(key=generated_key)
except:
self.key = generated_key
break
return super(Token, self).save(*args, **kwargs)
def generate_key(self):