Tweak authtoken

1 files changed, 4 insertions, 2 deletions

diff --git a/rest_framework/authtoken/models.py b/rest_framework/authtoken/models.py
index fd47e6c7..f7e78ef3 100644
--- a/rest_framework/authtoken/models.py
+++ b/rest_framework/authtoken/models.py
@@ -9,8 +9,7 @@ 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)
+    user = models.OneToOneField('auth.User', related_name='api_key')
     created = models.DateTimeField(auto_now_add=True)
 
     def save(self, *args, **kwargs):
@@ -21,3 +20,6 @@ class Token(models.Model):
     def generate_key(self):
         unique = str(uuid.uuid4())
         return hmac.new(unique, digestmod=sha1).hexdigest()
+
+    def __unicode__(self):
+        return self.key