encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-25 22:54:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

removed the possibility of conflict with the key generation

Browse Source
This commit is contained in:
Ilya Moroz 2015-11-02 11:23:35 +02:00
parent 119a073299
commit b449a7ba68
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
rest_framework/authtoken/models.py
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):