Added login user

Author: TanyaAng

styleguide_example/blog_examples/google_login_server_flow/raw/apis.py

@@ -1,3 +1,4 @@
+from django.contrib.auth import login
 from django.shortcuts import redirect
 from rest_framework import serializers, status
 from rest_framework.response import Response
@@ -6,6 +7,7 @@
 from styleguide_example.blog_examples.google_login_server_flow.raw.service import (
     GoogleRawLoginFlowService,
 )
+from styleguide_example.users.selectors import user_list
 
 
 class PublicApi(APIView):
@@ -63,6 +65,15 @@ def get(self, request, *args, **kwargs):
         id_token_decoded = google_tokens.decode_id_token()
         user_info = google_login_flow.get_user_info(google_tokens=google_tokens)
 
+        user_email = id_token_decoded["email"]
+        request_user_list = user_list(filters={"email": user_email})
+        user = request_user_list.get() if request_user_list else None
+
+        if user is None:
+            return Response({"error": f"User with email {user_email} is not found."}, status=status.HTTP_404_NOT_FOUND)
+
+        login(request, user)
+
         result = {
             "id_token_decoded": id_token_decoded,
             "user_info": user_info,

styleguide_example/blog_examples/google_login_server_flow/sdk/apis.py

@@ -1,3 +1,4 @@
+from django.contrib.auth import login
 from django.shortcuts import redirect
 from rest_framework import serializers, status
 from rest_framework.response import Response
@@ -6,6 +7,7 @@
 from styleguide_example.blog_examples.google_login_server_flow.sdk.services import (
     GoogleSdkLoginFlowService,
 )
+from styleguide_example.users.selectors import user_list
 
 
 class PublicApi(APIView):
@@ -63,6 +65,15 @@ def get(self, request, *args, **kwargs):
         id_token_decoded = google_tokens.decode_id_token()
         user_info = google_login_flow.get_user_info(google_tokens=google_tokens)
 
+        user_email = id_token_decoded["email"]
+        request_user_list = user_list(filters={"email": user_email})
+        user = request_user_list.get() if request_user_list else None
+
+        if user is None:
+            return Response({"error": f"User with email {user_email} is not found."}, status=status.HTTP_404_NOT_FOUND)
+
+        login(request, user)
+
         result = {
             "id_token_decoded": id_token_decoded,
             "user_info": user_info,