How to use custom token model in Django Rest Framework

Question

I'd like to use Django Rest Framework auth but I want to have more than one token for one user. In order to do that I need to implement my own Token model, I found this in Token authentication class:

class TokenAuthentication(BaseAuthentication):
    """
    Simple token based authentication.
    ...
    """

    model = Token
    """
    A custom token model may be used, but must have the following properties.

    * key -- The string identifying the token
    * user -- The user to which the token belongs
    """

But I don't have an idea how I can specify this model. Should I subclass TokenAuthentication?

radtek · Accepted Answer · 2018-07-24 19:27:54Z

17

Define you own authentication method in settings.py:

REST_FRAMEWORK = {
    'DEFAULT_AUTHENTICATION_CLASSES': (
        'my_project.my_app.authentication.MyOwnTokenAuthentication',
     ),
}

In authentication.py:

from rest_framework.authentication import TokenAuthentication
from my_project.my_app.models.token import MyOwnToken

class MyOwnTokenAuthentication(TokenAuthentication):
    model = MyOwnToken

In models.py:

import binascii
import os

from django.db import models
from django.utils.translation import ugettext_lazy as _
from my_project.companies.models import Company


class MyOwnToken(models.Model):
    """
    The default authorization token model.
    """
    key = models.CharField(_("Key"), max_length=40, primary_key=True)

    company = models.OneToOneField(
        Company, related_name='auth_token',
        on_delete=models.CASCADE, verbose_name="Company"
    )
    created = models.DateTimeField(_("Created"), auto_now_add=True)

    class Meta:
        verbose_name = _("Token")
        verbose_name_plural = _("Tokens")

    def save(self, *args, **kwargs):
        if not self.key:
            self.key = self.generate_key()
        return super(MyOwnToken, self).save(*args, **kwargs)

    def generate_key(self):
        return binascii.hexlify(os.urandom(20)).decode()

    def __str__(self):
        return self.key

edited Jul 24, 2018 at 19:27

radtek

36.6k13 gold badges149 silver badges114 bronze badges

answered Dec 19, 2016 at 23:49

mullerivan

2,0601 gold badge15 silver badges15 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

Raju Singh Over a year ago

but, how can I call "MyOwnTokenAuthentication" from view?

Souza Over a year ago

@RajuSingh you don't need to add anything to views, after you define DEFAULT_AUTHENTICATION_CLASSES, you're defining MyOwnTokenAuthentication as middleware function. Then, on your APIViews, you can access the authenticated user with "self.request.user" . You can also define authentication per view basis: authentication_classes = [MyOwnTokenAuthentication]

Michal Šrůtek Over a year ago

Getting 'MyOwnToken' object has no attribute 'user' when trying to use this approach. It seems to me that it still has to be connected with a user, is that correct?

Jamie · Accepted Answer · 2014-12-29 21:34:25Z

9

What that message is saying is that the model Token can be swapped out with any other model, as long as it has properties key and user. That way, if, for instance, you want a more complicated way to generate token keys, you can define your own model.

So if you want a custom Token model, you should do the following:

Subclass the Token model from rest_framework.authtoken.models. Add whatever custom behavior you want here, but make sure it has a key property and user property.
Subclass the TokenAuthentication class from rest_framework.authentication. Set its model property to whatever your new Token model is.
Make sure to reference your new authentication class in whatever views you want.

answered Dec 29, 2014 at 21:34

Jamie

2,3111 gold badge22 silver badges39 bronze badges

1 Comment

Binsoi Over a year ago

Hi Jamie, i'm using rest_framework.authtoken.models Token. i can see 3 fields which is key, created_at and user_id. my question is how can i add an extra field like company_id to that model? Thanks

Collectives™ on Stack Overflow

How to use custom token model in Django Rest Framework

2 Answers 2

3 Comments

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

3 Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related