add Google oauth

Author: yasssmk

backend/backend/settings.py

@@ -32,6 +32,7 @@
 
 ALLOWED_HOSTS = []
 
+SITE_ID = 1
 
 # Application definition
 
@@ -42,12 +43,20 @@
     "django.contrib.sessions",
     "django.contrib.messages",
     "django.contrib.staticfiles",
+    'django.contrib.sites',
     "rest_framework",
     "myapp",
     "corsheaders",
     'rest_framework_simplejwt.token_blacklist',
     'drf_yasg',
-    'backtest'
+    'backtest',
+    "rest_framework.authtoken",
+    'allauth',
+    'allauth.account',
+    'allauth.socialaccount',
+    'allauth.socialaccount.providers.google',
+    'dj_rest_auth',
+    'dj_rest_auth.registration',
 ]
 
 MIDDLEWARE = [
@@ -142,6 +151,12 @@
 
 AUTH_USER_MODEL = 'myapp.CustomUser'
 
+ACCOUNT_USER_MODEL_USERNAME_FIELD = None
+ACCOUNT_USERNAME_REQUIRED = False
+ACCOUNT_AUTHENTICATION_METHOD = 'email'
+ACCOUNT_EMAIL_REQUIRED = True
+ACCOUNT_EMAIL_VERIFICATION = 'optional'
+
 REST_FRAMEWORK = {
     'DEFAULT_AUTHENTICATION_CLASSES': [
         'rest_framework_simplejwt.authentication.JWTAuthentication',
@@ -203,4 +218,17 @@
             'MAX_ENTRIES': 1000,  # Maximum number of entries in the cache
         },
     }
+}
+
+SOCIALACCOUNT_PROVIDERS = {
+    'google': {
+        'SCOPE': [
+            'profile',
+            'email',
+        ],
+        'AUTH_PARAMS': {
+            'access_type': 'online',
+        },
+        'OAUTH_PKCE_ENABLED': True,
+    }
 }

backend/myapp/admin.py

@@ -1,3 +1,32 @@
 from django.contrib import admin
+from django.contrib.auth.admin import UserAdmin
+from django.contrib.auth.forms import UserChangeForm, UserCreationForm
+from .models import CustomUser
 
 # Register your models here.
+class CustomUserAdmin(UserAdmin):
+    add_form = UserCreationForm
+    form = UserChangeForm
+    model = CustomUser
+    list_display = ['email', 'first_name', 'last_name', 'is_staff']
+    search_fields = ['email', 'first_name', 'last_name']
+    ordering = ['email']  # Change 'username' to 'email'
+
+    # Fields to be used in displaying the User model.
+    fieldsets = (
+        (None, {'fields': ('email', 'password')}),
+        ('Personal info', {'fields': ('first_name', 'last_name')}),
+        ('Permissions', {'fields': ('is_active', 'is_staff', 'is_superuser')}),
+        ('Important dates', {'fields': ('last_login', 'date_joined')}),
+    )
+
+    # Fields to be used when creating a user.
+    add_fieldsets = (
+        (None, {
+            'classes': ('wide',),
+            'fields': ('email', 'password1', 'password2', 'first_name', 'last_name', 'is_active', 'is_staff', 'is_superuser'),
+        }),
+    )
+
+# Register CustomUserAdmin with the admin site
+admin.site.register(CustomUser, CustomUserAdmin)

backend/myapp/migrations/0002_alter_customuser_first_name_and_more.py

@@ -0,0 +1,23 @@
+# Generated by Django 5.0.6 on 2024-05-14 14:37
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('myapp', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='customuser',
+            name='first_name',
+            field=models.CharField(max_length=128),
+        ),
+        migrations.AlterField(
+            model_name='customuser',
+            name='last_name',
+            field=models.CharField(max_length=128),
+        ),
+    ]

backend/myapp/migrations/0003_alter_customuser_last_login.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.0.6 on 2024-05-16 12:16
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('myapp', '0002_alter_customuser_first_name_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='customuser',
+            name='last_login',
+            field=models.DateTimeField(auto_now=True),
+        ),
+    ]

backend/myapp/migrations/0004_customuser_auth_provider_alter_customuser_last_login.py

@@ -0,0 +1,23 @@
+# Generated by Django 5.0.6 on 2024-05-16 12:30
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('myapp', '0003_alter_customuser_last_login'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='customuser',
+            name='auth_provider',
+            field=models.CharField(default='email', max_length=50),
+        ),
+        migrations.AlterField(
+            model_name='customuser',
+            name='last_login',
+            field=models.DateTimeField(auto_now=True, null=True),
+        ),
+    ]

backend/myapp/migrations/0005_alter_customuser_last_login.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.0.6 on 2024-05-16 12:40
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('myapp', '0004_customuser_auth_provider_alter_customuser_last_login'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='customuser',
+            name='last_login',
+            field=models.DateTimeField(blank=True, null=True, verbose_name='last login'),
+        ),
+    ]

backend/myapp/migrations/0006_remove_customuser_auth_provider.py

@@ -0,0 +1,17 @@
+# Generated by Django 5.0.6 on 2024-05-17 13:19
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('myapp', '0005_alter_customuser_last_login'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='customuser',
+            name='auth_provider',
+        ),
+    ]

backend/myapp/migrations/0007_customuser_username.py

@@ -0,0 +1,20 @@
+# Generated by Django 5.0.6 on 2024-05-17 14:37
+
+import django.contrib.auth.validators
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('myapp', '0006_remove_customuser_auth_provider'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='customuser',
+            name='username',
+            field=models.CharField(default='placeholder', error_messages={'unique': 'A user with that username already exists.'}, help_text='Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only.', max_length=150, unique=True, validators=[django.contrib.auth.validators.UnicodeUsernameValidator()], verbose_name='username'),
+            preserve_default=False,
+        ),
+    ]

backend/myapp/migrations/0008_remove_customuser_username.py

@@ -0,0 +1,17 @@
+# Generated by Django 5.0.6 on 2024-05-17 15:16
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('myapp', '0007_customuser_username'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='customuser',
+            name='username',
+        ),
+    ]

backend/myapp/migrations/0009_alter_customuser_options.py

@@ -0,0 +1,17 @@
+# Generated by Django 5.0.6 on 2024-05-22 14:48
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('myapp', '0008_remove_customuser_username'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='customuser',
+            options={},
+        ),
+    ]

backend/myapp/models.py

@@ -1,5 +1,5 @@
 from django.db import models
-from django.contrib.auth.models import AbstractUser, BaseUserManager
+from django.contrib.auth.models import AbstractUser, BaseUserManager, AbstractBaseUser, PermissionsMixin
 
 class CustomUserManager(BaseUserManager):
     def create_user(self, email, password=None,  first_name=None, last_name=None, **extra_fields):
@@ -23,7 +23,7 @@ def create_superuser(self, email, password=None, **extra_fields):
         return self.create_user(email, password, **extra_fields)
 
 
-class CustomUser(AbstractUser):
+class CustomUser(AbstractBaseUser, PermissionsMixin):
     email = models.EmailField(unique=True)
     password = models.CharField(max_length=128)
     first_name = models.CharField(max_length=128, null=False)
@@ -32,7 +32,6 @@ class CustomUser(AbstractUser):
     is_staff = models.BooleanField(default=False)
     is_superuser = models.BooleanField(default=False)
     date_joined = models.DateTimeField(auto_now_add=True)
-    username = None
 
     objects = CustomUserManager()
 

backend/myapp/urls.py

@@ -1,7 +1,7 @@
 from django.contrib import admin
 from rest_framework import routers
 from django.urls import path, include
-from .views import Register, api_home, UserView,Logout
+from .views import Register, api_home, UserView,Logout, GoogleLogin
 from rest_framework_simplejwt.views import (
     TokenObtainPairView,
     TokenRefreshView,
@@ -13,5 +13,6 @@
     path("register", Register.as_view(), name="register"),
     path("user", UserView.as_view(), name="userview"),
     path("logout", Logout.as_view(), name='logout'),
-    path("", api_home)
+    path("", api_home),
+    path('google-login/', GoogleLogin.as_view(), name='google_login')
 ]

backend/myapp/views.py

@@ -8,8 +8,11 @@
 from rest_framework_simplejwt.authentication import JWTAuthentication
 from rest_framework.permissions import IsAuthenticated
 from rest_framework_simplejwt.tokens import AccessToken
-from drf_yasg import openapi
 from drf_yasg.utils import swagger_auto_schema
+from dj_rest_auth.registration.views import SocialLoginView
+from allauth.socialaccount.providers.google.views import GoogleOAuth2Adapter
+from allauth.socialaccount.models import SocialApp
+import requests
 
 # Create your views here.
 
@@ -74,4 +77,81 @@ def post(self, request):
             return Response({"success": "User logged out successfully."}, status=status.HTTP_200_OK)
         except Exception as e:
             print(request.data)
-            return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+            return Response({"error": str(e)}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        
+
+
+class GoogleLogin(SocialLoginView):
+    adapter_class = GoogleOAuth2Adapter
+
+    def post(self, request, *args, **kwargs):
+
+        code = request.data.get('code')
+        if not code:
+            return Response({'detail': 'Code is missing'}, status=status.HTTP_400_BAD_REQUEST)
+
+        try:
+            # Retrieve the Google social app
+            app = SocialApp.objects.get(provider="google")
+
+            # Exchange the authorization code for an access token
+            token_url = "https://oauth2.googleapis.com/token"
+            token_data = {
+                "code": code,
+                "client_id": app.client_id,
+                "client_secret": app.secret,
+                "redirect_uri": "http://localhost:3000",  # Make sure this matches the redirect_uri in your frontend config
+                "grant_type": "authorization_code",
+            }
+            token_response = requests.post(token_url, data=token_data)
+            token_response_data = token_response.json()
+            access_token = token_response_data.get('access_token')
+
+            if not access_token:
+                return Response({'detail': 'Failed to obtain access token from Google'}, status=status.HTTP_400_BAD_REQUEST)
+
+            # Retrieve user info from Google
+            user_info_url = "https://www.googleapis.com/oauth2/v2/userinfo"
+            user_info_response = requests.get(user_info_url, headers={"Authorization": f"Bearer {access_token}"})
+            user_info = user_info_response.json()
+
+            # Extract user details
+            email = user_info.get('email')
+            first_name = user_info.get('given_name')
+            last_name = user_info.get('family_name')
+            password=user_info.get('id')+user_info.get('email')+user_info.get('locale')
+
+
+            if not email:
+                return Response({'detail': 'Failed to obtain user info from Google'}, status=status.HTTP_400_BAD_REQUEST)
+
+            # Create or get the user from your database
+            user_data = {
+                            'email': email,
+                            'password': password,
+                            'first_name': first_name,
+                            'last_name': last_name,
+                            
+                        }
+            print("User data to be serialized:", user_data)
+
+            if not CustomUser.objects.filter(email=email).exists():
+                serializer = UserSerializer(data=user_data)
+                if serializer.is_valid():
+                    user = CustomUser.objects.create_user(email=email, password=password, last_name=last_name, first_name=first_name)
+                else:
+                    return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
+            else:
+                user = CustomUser.objects.get(email=email)
+
+            # Generate JWT token using MyTokenObtainPairSerializer
+            refresh = RefreshToken.for_user(user)
+            return Response({
+                'refresh': str(refresh),
+                'access': str(refresh.access_token),
+            }, status=status.HTTP_200_OK)
+
+        except SocialApp.DoesNotExist:
+            return Response({'detail': 'Google SocialApp configuration is missing.'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        except Exception as e:
+            return Response({'detail': str(e)}, status=status.HTTP_400_BAD_REQUEST)