diff options
Diffstat (limited to 'djangorestframework/authtoken/models.py')
| -rw-r--r-- | djangorestframework/authtoken/models.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/djangorestframework/authtoken/models.py b/djangorestframework/authtoken/models.py new file mode 100644 index 00000000..fd47e6c7 --- /dev/null +++ b/djangorestframework/authtoken/models.py @@ -0,0 +1,23 @@ +import uuid +import hmac +from hashlib import sha1 +from django.db import models + + +class Token(models.Model): + """ + The default authorization token model. + """ + key = models.CharField(max_length=40, primary_key=True) + user = models.ForeignKey('auth.User') + revoked = models.BooleanField(default=False) + created = models.DateTimeField(auto_now_add=True) + + def save(self, *args, **kwargs): + if not self.key: + self.key = self.generate_key() + return super(Token, self).save(*args, **kwargs) + + def generate_key(self): + unique = str(uuid.uuid4()) + return hmac.new(unique, digestmod=sha1).hexdigest() |
