Merge pull request #7 from L3-iGrant/feature/2-buyer-login-api

Add #2 Buyer login API

igrant_user/models.py

@@ -10,6 +10,8 @@ class IGrantUser(AbstractBaseUser, PermissionsMixin):
     class UserType(models.TextChoices):
         COMPANY = "COMPANY", _("Company")
         ISSUER = "ISSUER", _("Issuer")
+        BUYER  = "BUYER", _("Buyer")
+        SELLER = "SELLER", _("Seller")
 
     class Orgs(models.TextChoices):
         DEFAULT = "NIL", _("Nil")

igrant_user/urls.py

@@ -1,8 +1,11 @@
 from django.contrib import admin
 from django.urls import path, include
-from .views import UserList, UserDetail
+from .views import UserList, UserDetail, MyTokenObtainPairView
+from rest_framework_simplejwt.views import TokenRefreshView
 
 urlpatterns = [
         path('/', UserList.as_view()),
         path('/<int:pk>/', UserDetail.as_view()),
+        path('/login/', MyTokenObtainPairView.as_view(), name='token_obtain_pair'),
+        path('/api/token/refresh/', TokenRefreshView.as_view(), name='token_refresh'),
 ]

igrant_user/views.py

@@ -3,6 +3,8 @@
 from .serializers import IGrantUserSerializer
 from rest_framework import permissions
 from .permissions import IsOwnerOrReadOnly
+from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
+from rest_framework_simplejwt.views import TokenObtainPairView
 
 
 # Create your views here.
@@ -19,3 +21,18 @@ class UserDetail(generics.RetrieveAPIView):
     queryset = IGrantUser.objects.all()
     serializer_class = IGrantUserSerializer
     permission_classes = [permissions.IsAuthenticated, IsOwnerOrReadOnly]
+
+
+class MyTokenObtainPairSerializer(TokenObtainPairSerializer):
+    @classmethod
+    def get_token(cls, user):
+        token = super().get_token(user)
+
+        # Add custom claims
+        token['email'] = user.email
+        token['user_type'] = user.user_type
+
+        return token
+
+class MyTokenObtainPairView(TokenObtainPairView):
+    serializer_class = MyTokenObtainPairSerializer

pob_backend/settings.py

@@ -133,6 +133,7 @@
 
 REST_FRAMEWORK = {
     "DEFAULT_AUTHENTICATION_CLASSES": [
+        "rest_framework_simplejwt.authentication.JWTAuthentication",
         "rest_framework.authentication.SessionAuthentication",
         "rest_framework.authentication.TokenAuthentication",
     ],

pob_backend/urls.py

@@ -18,7 +18,6 @@
 
 urlpatterns = [
     path("admin/", admin.site.urls),
-    path(r"rest-auth/", include("rest_auth.urls")),
     path(r"users", include("igrant_user.urls")),
     path(r"connections", include("connections.urls")),
     path(r"certificates", include("certificate.urls")),