1-2d done

Author: Toruitas

djsr/authentication/models.py

@@ -3,4 +3,4 @@
 
 
 class CustomUser(AbstractUser):
-    fav_color = models.CharField(blank=True, max_length=120)
+    fav_color = models.CharField(blank=True, max_length=120)

djsr/authentication/serializers.py

@@ -0,0 +1,13 @@
+# djsr/authentication/serializers.py
+from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
+
+
+class MyTokenObtainPairSerializer(TokenObtainPairSerializer):
+
+    @classmethod
+    def get_token(cls, user):
+        token = super(MyTokenObtainPairSerializer, cls).get_token(user)
+
+        # Add custom claims
+        token['fav_color'] = user.fav_color
+        return token

djsr/authentication/urls.py

@@ -1,7 +1,8 @@
 from django.urls import path
 from rest_framework_simplejwt import views as jwt_views
+from .views import ObtainTokenPairWithColorView
 
 urlpatterns = [
-    path('token/obtain/', jwt_views.TokenObtainPairView.as_view(), name='token_create'),  # override sjwt stock token
+    path('token/obtain/', ObtainTokenPairWithColorView.as_view(), name='token_create'),
     path('token/refresh/', jwt_views.TokenRefreshView.as_view(), name='token_refresh'),
 ]

djsr/authentication/views.py

@@ -1,3 +1,8 @@
 from django.shortcuts import render
 
-# Create your views here.
+from rest_framework_simplejwt.views import TokenObtainPairView
+from .serializers import MyTokenObtainPairSerializer
+
+
+class ObtainTokenPairWithColorView(TokenObtainPairView):
+    serializer_class = MyTokenObtainPairSerializer