changes

junior/serializers.py

@@ -0,0 +1,92 @@
+"""Serializer file for junior"""
+"""Import Django 3rd party app"""
+from rest_framework import serializers
+from django.contrib.auth.models import User
+from django.db import transaction
+import random
+"""Import django app"""
+from junior.models import Junior
+
+class ListCharField(serializers.ListField):
+    """Serializer for Array field"""
+    child = serializers.CharField()
+
+    def to_representation(self, data):
+        """to represent the data"""
+        return data
+
+    def to_internal_value(self, data):
+        """internal value"""
+        if isinstance(data, list):
+            return data
+        raise serializers.ValidationError({"details":"Invalid input. Expected a list of strings."})
+
+
+class CreateJuniorSerializer(serializers.ModelSerializer):
+    """Create junior serializer"""
+    first_name = serializers.SerializerMethodField('get_first_name')
+    last_name = serializers.SerializerMethodField('get_last_name')
+    email = serializers.SerializerMethodField('get_email')
+    phone = serializers.CharField(max_length=20, required=False)
+    country_code = serializers.IntegerField(required=False)
+    dob = serializers.DateField(required=False)
+    referral_code = serializers.CharField(max_length=100, required=False)
+    guardian_code = ListCharField(required=False)
+
+    class Meta(object):
+        """Meta info"""
+        model = Junior
+        fields = ['first_name', 'last_name', 'email', 'phone', 'gender', 'country_code', 'dob', 'referral_code',
+                  'passcode', 'is_complete_profile', 'guardian_code']
+
+    def get_first_name(self,obj):
+        """first name of junior"""
+        return obj.auth.first_name
+
+    def get_last_name(self,obj):
+        """last name of junior"""
+        return obj.auth.last_name
+
+    def get_email(self,obj):
+        """email of junior"""
+        return obj.auth.email
+
+    def create(self, validated_data):
+        """Create junior profile"""
+        user = User.objects.filter(username=self.context['user']).last()
+        if user:
+            """Save first and last name of junior"""
+            user.first_name = self.context.get('first_name', user.first_name)
+            user.last_name = self.context.get('last_name', user.last_name)
+            user.save()
+        """Create junior data"""
+        junior, created = Junior.objects.get_or_create(auth=self.context['user'])
+        if created:
+            """Create referral code and junior code"""
+            junior.referral_code = ''.join([str(random.randrange(9)) for _ in range(4)])
+            junior.junior_code = ''.join([str(random.randrange(9)) for _ in range(4)])
+        if junior:
+            """update details according to the data get from request"""
+            junior.gender = validated_data.get('gender',junior.gender)
+            """Update guardian code"""
+            junior.guardian_code = validated_data.get('guardian_code', junior.guardian_code)
+            junior.dob = validated_data.get('dob',junior.dob)
+            junior.passcode = validated_data.get('passcode', junior.passcode)
+            """Update country code and phone number"""
+            junior.phone = validated_data.get('phone', junior.phone)
+            junior.country_code = validated_data.get('country_code', junior.country_code)
+            junior.referral_code_used = validated_data.get('referral_code_used', junior.referral_code_used)
+            """Complete profile of the junior if below all data are filled"""
+            complete_profile_field = all([junior.phone, junior.gender, junior.family_name,
+                                          junior.dob, junior.country_code, user.first_name, user.last_name])
+            junior.is_complete_profile = False
+            if complete_profile_field:
+                junior.is_complete_profile = True
+            junior.save()
+        return junior
+
+    def save(self, **kwargs):
+        """Save the data into junior table"""
+        with transaction.atomic():
+            instance = super().save(**kwargs)
+            return instance