Initial Commit

zod_bank/account/serializers.py

@@ -0,0 +1,104 @@
+from rest_framework import serializers
+from django.contrib.auth.models import User
+from guardian.models import Guardian
+from junior.models import Junior
+from account.models import UserProfile, UserEmailOtp, UserPhoneOtp
+from base.constants import GUARDIAN, JUNIOR, SUPERUSER
+from django.db import transaction
+from base.messages import ERROR_CODE_REQUIRED, ERROR_CODE, SUCCESS_CODE, STATUS_CODE_ERROR
+from django.core.exceptions import ObjectDoesNotExist
+from django.contrib.auth import authenticate
+from rest_framework import viewsets, status
+from rest_framework.decorators import action
+from django.contrib.auth import authenticate, login
+from rest_framework_simplejwt.tokens import RefreshToken
+
+class SuperUserSerializer(serializers.ModelSerializer):
+    user_type = serializers.SerializerMethodField('get_user_type')
+
+    def get_user_type(self, obj):
+        """user type"""
+        return SUPERUSER
+
+    class Meta(object):
+        """Meta info"""
+        model = User
+        fields = ['id', 'username', 'email', 'first_name', 'last_name', 'is_active', 'user_type']
+
+
+class GuardianSerializer(serializers.ModelSerializer):
+    """guardian serializer"""
+    user_type = serializers.SerializerMethodField('get_user_type')
+    email = serializers.SerializerMethodField('get_auth')
+    first_name = serializers.SerializerMethodField('get_first_name')
+    last_name = serializers.SerializerMethodField('get_last_name')
+    auth_token = serializers.SerializerMethodField('get_auth_token')
+
+    def get_auth_token(self, obj):
+        refresh = RefreshToken.for_user(obj.user)
+        access_token = str(refresh.access_token)
+        return access_token
+
+    def get_user_type(self, obj):
+        """user type"""
+        return GUARDIAN
+
+    def get_auth(self, obj):
+        """user email address"""
+        return obj.user.username
+
+    def get_first_name(self, obj):
+        """user first name"""
+        return obj.user.first_name
+
+    def get_last_name(self, obj):
+        """user last name"""
+        return obj.user.last_name
+
+    class Meta(object):
+        """Meta info"""
+        model = Guardian
+        fields = ['auth_token', 'email', 'first_name', 'last_name', 'country_code', 'phone', 'family_name', 'gender', 'dob',
+                  'referral_code', 'is_active', 'is_complete_profile', 'passcode',
+                  'created_at', 'updated_at', 'user_type']
+
+
+class JuniorSerializer(serializers.ModelSerializer):
+    """junior serializer"""
+    user_type = serializers.SerializerMethodField('get_user_type')
+    email = serializers.SerializerMethodField('get_auth')
+    first_name = serializers.SerializerMethodField('get_first_name')
+    last_name = serializers.SerializerMethodField('get_last_name')
+    auth_token = serializers.SerializerMethodField('get_auth_token')
+
+    def get_auth_token(self, obj):
+        refresh = RefreshToken.for_user(obj.auth)
+        access_token = str(refresh.access_token)
+        return access_token
+
+    def get_user_type(self, obj):
+        return JUNIOR
+
+    def get_auth(self, obj):
+        return obj.auth.username
+
+    def get_first_name(self, obj):
+        return obj.auth.first_name
+
+    def get_last_name(self, obj):
+        return obj.auth.last_name
+
+    class Meta(object):
+        """Meta info"""
+        model = Junior
+        fields = ['auth_token', 'email', 'first_name', 'last_name', 'country_code', 'phone', 'gender', 'dob',
+                  'guardian_code', 'referral_code','is_active', 'is_complete_profile', 'created_at',
+                  'updated_at', 'user_type']
+
+class EmailVerificationSerializer(serializers.ModelSerializer):
+    """Email verification serializer"""
+    class Meta(object):
+        """Meta info"""
+        model = UserEmailOtp
+        fields = '__all__'
+