Apply Pre-Commit Changes

Signed-off-by: Shubhanshu Saxena <[email protected]>

Author: shubhanshu02

.gitignore

@@ -346,4 +346,4 @@ $RECYCLE.BIN/
 # End of https://www.gitignore.io/api/vue,node,vuejs,linux,macos,python,django,windows
 script.py
 crypt.py
-service_account.json
+service_account.json

Auth/admin.py

@@ -1,11 +1,13 @@
 from django.contrib import admin
 from .models import VerifiedAccount
+
 # Register your models here.
 # admin.site.register(VerifiedAccount)
 
+
 @admin.register(VerifiedAccount)
 class VerifiedAccountAdmin(admin.ModelAdmin):
-    list_display=('__str__','provider','is_verified')
-    search_fields=('uid',)
-    list_filter=('provider','is_verified')
-    raw_id_fields=('user',)
+    list_display = ("__str__", "provider", "is_verified")
+    search_fields = ("uid",)
+    list_filter = ("provider", "is_verified")
+    raw_id_fields = ("user",)

Auth/apps.py

@@ -2,4 +2,4 @@
 
 
 class AuthConfig(AppConfig):
-    name = 'Auth'
+    name = "Auth"

Auth/migrations/0001_initial.py

@@ -15,17 +15,41 @@ class Migration(migrations.Migration):
 
     operations = [
         migrations.CreateModel(
-            name='VerifiedAccount',
+            name="VerifiedAccount",
             fields=[
-                ('uid', models.CharField(editable=False, max_length=64, primary_key=True, serialize=False)),
-                ('provider', models.CharField(choices=[('facebook.com', 'Facebook'), ('google.com', 'Google'), ('github.com', 'Github'), ('email', 'Email')], max_length=15)),
-                ('provider_uid', models.CharField(max_length=64)),
-                ('is_verified', models.BooleanField(default=True)),
-                ('user', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, related_name='verified_account', related_query_name='account', to=settings.AUTH_USER_MODEL)),
+                (
+                    "uid",
+                    models.CharField(
+                        editable=False, max_length=64, primary_key=True, serialize=False
+                    ),
+                ),
+                (
+                    "provider",
+                    models.CharField(
+                        choices=[
+                            ("facebook.com", "Facebook"),
+                            ("google.com", "Google"),
+                            ("github.com", "Github"),
+                            ("email", "Email"),
+                        ],
+                        max_length=15,
+                    ),
+                ),
+                ("provider_uid", models.CharField(max_length=64)),
+                ("is_verified", models.BooleanField(default=True)),
+                (
+                    "user",
+                    models.OneToOneField(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="verified_account",
+                        related_query_name="account",
+                        to=settings.AUTH_USER_MODEL,
+                    ),
+                ),
             ],
         ),
         migrations.AlterUniqueTogether(
-            name='verifiedaccount',
-            unique_together={('provider', 'provider_uid')},
+            name="verifiedaccount",
+            unique_together={("provider", "provider_uid")},
         ),
     ]

Auth/migrations/0002_auto_20190331_0158.py

@@ -6,13 +6,21 @@
 class Migration(migrations.Migration):
 
     dependencies = [
-        ('Auth', '0001_initial'),
+        ("Auth", "0001_initial"),
     ]
 
     operations = [
         migrations.AlterField(
-            model_name='verifiedaccount',
-            name='provider',
-            field=models.CharField(choices=[('facebook.com', 'Facebook'), ('google.com', 'Google'), ('github.com', 'Github'), ('password', 'Email')], max_length=15),
+            model_name="verifiedaccount",
+            name="provider",
+            field=models.CharField(
+                choices=[
+                    ("facebook.com", "Facebook"),
+                    ("google.com", "Google"),
+                    ("github.com", "Github"),
+                    ("password", "Email"),
+                ],
+                max_length=15,
+            ),
         ),
     ]

Auth/migrations/0003_auto_20190331_0210.py

@@ -6,13 +6,13 @@
 class Migration(migrations.Migration):
 
     dependencies = [
-        ('Auth', '0002_auto_20190331_0158'),
+        ("Auth", "0002_auto_20190331_0158"),
     ]
 
     operations = [
         migrations.AlterField(
-            model_name='verifiedaccount',
-            name='provider_uid',
+            model_name="verifiedaccount",
+            name="provider_uid",
             field=models.CharField(blank=True, max_length=64, null=True),
         ),
     ]

Auth/models.py

@@ -1,36 +1,41 @@
 from django.db import models
 from django.contrib.auth.models import User
 from .utils import FirebaseAPI
+
 # Create your models here.
 
+
 class VerifiedAccount(models.Model):
-    AUTH_FACEBOOK_PROVIDER = 'facebook.com'
-    AUTH_GOOGLE_PROVIDER = 'google.com'
-    AUTH_GITHUB_PROVIDER  = 'github.com'
-    AUTH_EMAIL_PROVIDER = 'password'
+    AUTH_FACEBOOK_PROVIDER = "facebook.com"
+    AUTH_GOOGLE_PROVIDER = "google.com"
+    AUTH_GITHUB_PROVIDER = "github.com"
+    AUTH_EMAIL_PROVIDER = "password"
 
     AUTH_PROVIDERS_CHOICE = (
-        (AUTH_FACEBOOK_PROVIDER, 'Facebook'),
-        (AUTH_GOOGLE_PROVIDER, 'Google'),
-        (AUTH_GITHUB_PROVIDER, 'Github'),
-        (AUTH_EMAIL_PROVIDER, 'Email'),
+        (AUTH_FACEBOOK_PROVIDER, "Facebook"),
+        (AUTH_GOOGLE_PROVIDER, "Google"),
+        (AUTH_GITHUB_PROVIDER, "Github"),
+        (AUTH_EMAIL_PROVIDER, "Email"),
     )
 
     uid = models.CharField(max_length=64, primary_key=True, editable=False)
     user = models.OneToOneField(
-        User, on_delete=models.CASCADE,
-        related_name='verified_account', related_query_name='account')
+        User,
+        on_delete=models.CASCADE,
+        related_name="verified_account",
+        related_query_name="account",
+    )
     provider = models.CharField(choices=AUTH_PROVIDERS_CHOICE, max_length=15)
     provider_uid = models.CharField(max_length=64, null=True, blank=True)
     is_verified = models.BooleanField(default=True)
 
     class Meta:
-        unique_together = ('provider', 'provider_uid')
-        app_label='Auth'
+        unique_together = ("provider", "provider_uid")
+        app_label = "Auth"
 
     def __str__(self):
-        return f'{self.user} from {self.provider}'
-    
+        return f"{self.user} from {self.provider}"
+
     def get_verified_status(self):
         if self.is_verified:
             return True

Auth/serializers.py

@@ -1,10 +1,12 @@
 from rest_framework import serializers
 from .utils import FirebaseAPI
 from .models import *
-from website.models import Profile,ValidReferral
+from website.models import Profile, ValidReferral
 from django.conf import settings
 from rest_framework.exceptions import ParseError
 import requests
+
+
 class LoginSerializer(serializers.Serializer):
     id_token = serializers.CharField(max_length=2400)
     provider_token = serializers.CharField(max_length=2400, required=False)
@@ -13,73 +15,76 @@ def validate_access_token(self, access_token):
         return FirebaseAPI.verify_id_token(access_token)
 
     def validate(self, attrs):
-        id_token = attrs.get('id_token', None)
-        provider_token = attrs.get('provider_token', None)
+        id_token = attrs.get("id_token", None)
+        provider_token = attrs.get("provider_token", None)
 
         user = None
 
         if id_token:
             jwt = self.validate_access_token(id_token)
-            uid = jwt['uid']
-            provider=FirebaseAPI.get_provider(jwt)
-            
+            uid = jwt["uid"]
+            provider = FirebaseAPI.get_provider(jwt)
+
             try:
                 account = VerifiedAccount.objects.get(pk=uid)
             except VerifiedAccount.DoesNotExist:
-                raise serializers.ValidationError('No such account exists')
-            
+                raise serializers.ValidationError("No such account exists")
+
             user = account.user
-            # add the verification status to the validated data 
-            attrs['is_verified']=account.get_verified_status()   
-            profile=user.profile
+            # add the verification status to the validated data
+            attrs["is_verified"] = account.get_verified_status()
+            profile = user.profile
             # because we also need the frontend to know if the profile is complete
             # this line automatically applies referral code if needed
-            attrs['is_profile_complete']=profile.get_or_set_profile_status()
+            attrs["is_profile_complete"] = profile.get_or_set_profile_status()
             if provider_token:
                 account.provider_token = provider_token
                 account.save()
         else:
-            raise ParseError('Provide access_token or username to continue.')
+            raise ParseError("Provide access_token or username to continue.")
         # Did we get back an active user?
         if user:
             if not user.is_active:
-                raise serializers.ValidationError('User account is disabled.')
+                raise serializers.ValidationError("User account is disabled.")
         else:
             raise serializers.ValidationError(
-                'Unable to log in with provided credentials.')
+                "Unable to log in with provided credentials."
+            )
 
-        attrs['user'] = user
+        attrs["user"] = user
         return attrs
 
 
 class RegisterSerializer(serializers.Serializer):
-    
+
     id_token = serializers.CharField(max_length=2400, required=True)
     # provider_token = serializers.CharField(max_length=2400, required=False)
     first_name = serializers.CharField(max_length=40, allow_blank=False)
     last_name = serializers.CharField(max_length=100, allow_blank=True, required=False)
-    applied_referral_code = serializers.CharField(max_length=500,required=False, allow_blank=True)
+    applied_referral_code = serializers.CharField(
+        max_length=500, required=False, allow_blank=True
+    )
     # g_recaptcha_response = serializers.CharField(max_length=500, required=True)
 
     def validate_id_token(self, access_token):
         return FirebaseAPI.verify_id_token(access_token)
 
-    def validate_first_name(self,name):
-        if name==None or name=='':
+    def validate_first_name(self, name):
+        if name == None or name == "":
             raise serializers.ValidationError("First Name cannot be blank")
         return name
 
-    def validate_applied_referral_code(self,code):
-        if code is None or code=="":
+    def validate_applied_referral_code(self, code):
+        if code is None or code == "":
             return None
         try:
-            referred_by=Profile.objects.get(referral_code=code)
+            referred_by = Profile.objects.get(referral_code=code)
             if not referred_by.get_or_set_profile_status():
                 raise serializers.ValidationError("Invalid Referral Code")
         except:
             raise serializers.ValidationError("Invalid Referral Code")
         return referred_by
-        
+
     # def validate_g_recaptcha_response(self, token):
     #     data= {
     #         'secret':settings.GOOGLE_RECAPTCHA_SECRET_KEY,
@@ -91,57 +96,56 @@ def validate_applied_referral_code(self,code):
     #         return
     #     raise serializers.ValidationError("Captcha could not be verified. Please try again.")
 
-    def get_user(self, data,jwt):
+    def get_user(self, data, jwt):
         user = User()
-        uid = jwt['uid']
-        email = jwt.get('email', '')
+        uid = jwt["uid"]
+        email = jwt.get("email", "")
         user.username = uid
         user.email = email
-        user.first_name = data.get('first_name')
-        user.last_name = data.get('last_name',"")
-        user.gender = data.get('gender')
+        user.first_name = data.get("first_name")
+        user.last_name = data.get("last_name", "")
+        user.gender = data.get("gender")
         return user
 
-    
     def save(self):
         data = self.validated_data
-        jwt = data.get('id_token')
+        jwt = data.get("id_token")
         # print(jwt)
-        uid = jwt['uid']
+        uid = jwt["uid"]
         provider = FirebaseAPI.get_provider(jwt)
         # provider_uid = None
         # if provider !=VerifiedAccount.AUTH_EMAIL_PROVIDER:
         #     provider_uid = FirebaseAPI.get_provider_uid(jwt, provider)
-        user = self.get_user(data,jwt)
+        user = self.get_user(data, jwt)
         try:
             user.validate_unique()
         except Exception as e:
             raise serializers.ValidationError(detail="User already exists")
         user.save()
         account, _ = VerifiedAccount.objects.get_or_create(
-            uid=uid, user=user, provider=provider)
+            uid=uid, user=user, provider=provider
+        )
 
         if provider == VerifiedAccount.AUTH_EMAIL_PROVIDER:
-            account.is_verified=False
+            account.is_verified = False
             account.save()
-        
-        referred_by = data.get('applied_referral_code', None)
 
-        profile,_ = Profile.objects.get_or_create(user=user,referred_by=referred_by)
+        referred_by = data.get("applied_referral_code", None)
+
+        profile, _ = Profile.objects.get_or_create(user=user, referred_by=referred_by)
         profile.name = user.first_name + " " + user.last_name
         profile.save()
         return user
 
 
 class RegisterResponseSerializer(serializers.Serializer):
     user_id = serializers.IntegerField()
-    token = serializers.CharField(max_length = 500)
+    token = serializers.CharField(max_length=500)
     verification_status = serializers.BooleanField()
 
 
-
 class LoginResponseSerializer(serializers.Serializer):
-    user_id= serializers.IntegerField()
+    user_id = serializers.IntegerField()
     token = serializers.CharField(max_length=500)
     verification_status = serializers.BooleanField()
-    is_profile_complete = serializers.BooleanField()
+    is_profile_complete = serializers.BooleanField()

Auth/urls.py

@@ -1,6 +1,7 @@
 from django.urls import path
 from .views import *
+
 urlpatterns = [
-    path('login/',LoginView.as_view()),
-    path('register/',RegisterView.as_view()),
+    path("login/", LoginView.as_view()),
+    path("register/", RegisterView.as_view()),
 ]