Skip to content

Instantly share code, notes, and snippets.

@shafi2263
Last active February 9, 2020 08:37
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save shafi2263/6bb24d5f17acda858ecc61dcec5a6819 to your computer and use it in GitHub Desktop.
Save shafi2263/6bb24d5f17acda858ecc61dcec5a6819 to your computer and use it in GitHub Desktop.
from django.contrib.auth import get_user_model
from rest_framework_simplejwt.serializers import TokenObtainSerializer
from rest_framework_simplejwt.tokens import RefreshToken
class MyTokenObtainSerializer(TokenObtainSerializer):
@classmethod
def get_token(cls, user):
return RefreshToken.for_user(user)
def validate(self, attrs):
data = super().validate(attrs)
refresh = self.get_token(self.user)
data['access'] = str(refresh.access_token)
return data
def main():
User = get_user_model()
user = User.objects.last()
user.set_password('Asdf1321!')
user.save()
data = {
'username': user.username,
'password': 'Asdf1321!'
}
serializer = MyTokenObtainSerializer(data=data)
serializer.is_valid(raise_exception=True)
print(serializer.validated_data)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment