Merge pull request #2233 from aswanthabam/feature-interests

Added user interest add and update endpoints

api/register/register_views.py

@@ -4,15 +4,67 @@
 from db.organization import Country, Department, District, Organization, State, Zone
 from django.utils.decorators import method_decorator
 from db.task import InterestGroup
-from db.user import Role, User
+from db.user import Role, User, UserInterests
 from utils.response import CustomResponse
 from utils.types import OrganizationType
-from utils.utils import send_template_mail
 from . import serializers
 from .register_helper import get_auth_token
 from django.views.decorators.cache import cache_page
 from django.core.cache import cache
 from mu_celery.task import send_email
+from utils.permission import CustomizePermission, JWTUtils
+from decouple import config
+import requests
+
+
+class UserInterestAPI(APIView):
+    permission_classes = [CustomizePermission]
+
+    def put(self, request):
+        if not JWTUtils.is_jwt_authenticated(request):
+            return CustomResponse(
+                general_message="Unauthorized access"
+            ).get_failure_response()
+        user_id = JWTUtils.fetch_user_id(request)
+        if not (user := cache.get(f"db_user_{user_id}")):
+            user = User.objects.filter(id=user_id).first()
+        user_interest = UserInterests.objects.filter(user=user).first()
+        if not user_interest:
+            return CustomResponse(
+                general_message="User interests not found"
+            ).get_failure_response()
+        serializer = serializers.UserInterestSerializer(
+            instance=user_interest, data=request.data, context={"user": user}
+        )
+        if serializer.is_valid():
+            serializer.update(user_interest, serializer.validated_data)
+            return CustomResponse(
+                general_message="Updated interests"
+            ).get_success_response()
+        return CustomResponse(general_message=serializer.errors).get_failure_response()
+
+    def post(self, request):
+        if not JWTUtils.is_jwt_authenticated(request):
+            return CustomResponse(
+                general_message="Unauthorized access"
+            ).get_failure_response()
+        user_id = JWTUtils.fetch_user_id(request)
+        if not (user := cache.get(f"db_user_{user_id}")):
+            user = User.objects.filter(id=user_id).first()
+        user_interest = UserInterests.objects.filter(user=user).first()
+        if user_interest:
+            return CustomResponse(
+                general_message="User interests already exist"
+            ).get_failure_response()
+        serializer = serializers.UserInterestSerializer(
+            data=request.data, context={"user": user}
+        )
+        if serializer.is_valid():
+            serializer.save()
+            return CustomResponse(
+                general_message="Added interests"
+            ).get_success_response()
+        return CustomResponse(general_message=serializer.errors).get_failure_response()
 
 
 class UserRegisterValidateAPI(APIView):

api/register/serializers.py

@@ -14,8 +14,24 @@
     UserOrganizationLink,
     Zone,
 )
-from db.task import InterestGroup, Level, MucoinInviteLog, UserIgLink, UserLvlLink, Wallet
-from db.user import Role, Socials, User, UserMentor, UserReferralLink, UserRoleLink, UserSettings
+from db.task import (
+    InterestGroup,
+    Level,
+    MucoinInviteLog,
+    UserIgLink,
+    UserLvlLink,
+    Wallet,
+)
+from db.user import (
+    Role,
+    Socials,
+    User,
+    UserMentor,
+    UserReferralLink,
+    UserRoleLink,
+    UserSettings,
+    UserInterests,
+)
 from utils.exception import CustomException
 from utils.types import OrganizationType, RoleType
 from utils.utils import DateTimeUtils
@@ -163,7 +179,6 @@ class Meta:
         ]
 
 
-
 class ReferralSerializer(serializers.ModelSerializer):
     user = serializers.CharField(required=False)
     muid = serializers.CharField(required=False)
@@ -326,6 +341,7 @@ class Meta:
             "area_of_interest",
         ]
 
+
 class RegisterSerializer(serializers.Serializer):
     user = UserSerializer()
     organization = UserOrgLinkSerializer(required=False)
@@ -351,7 +367,7 @@ def create(self, validated_data):
 
             if mentor := validated_data.pop("mentor", None):
                 mentor["user"] = user
-                MentorSerializer().create(mentor)    
+                MentorSerializer().create(mentor)
 
         return user
 
@@ -399,3 +415,72 @@ class Meta:
 
     def get_location(self, obj):
         return f"{obj.name}, {obj.zone.state.name}, {obj.zone.state.country.name}"
+
+
+class UserInterestSerializer(serializers.ModelSerializer):
+    id = serializers.CharField(read_only=True)
+    user = serializers.CharField(read_only=True)
+    choosen_interests = serializers.JSONField()
+    other_interests = serializers.JSONField(required=False)
+    choosen_endgoals = serializers.JSONField()
+    other_endgoals = serializers.JSONField(required=False)
+    created_at = serializers.DateTimeField(read_only=True)
+    updated_at = serializers.DateTimeField(read_only=True)
+
+    def create(self, validated_data):
+        validated_data["created_at"] = validated_data["updated_at"] = (
+            DateTimeUtils.get_current_utc_time()
+        )
+        if user := self.context.get("user"):
+            validated_data["user"] = user
+        else:
+            return serializers.ValidationError("User not found")
+        return super().create(validated_data)
+
+    def update(self, instance, validated_data):
+        if validated_data.get("choosen_interests", None):
+            instance.choosen_interests = validated_data.get("choosen_interests", [])
+        if validated_data.get("other_interests", None):
+            instance.other_interests = validated_data.get("other_interests", [])
+        if validated_data.get("choosen_endgoals", None):
+            instance.choosen_endgoals = validated_data.get("choosen_endgoals", [])
+        if validated_data.get("other_endgoals", None):
+            instance.other_endgoals = validated_data.get("other_endgoals", [])
+        instance.updated_at = DateTimeUtils.get_current_utc_time()
+        return instance.save()
+
+    def validate_choosen_interests(self, interests):
+        if not all(
+            interest in ("maker", "software", "creative", "manager", "other")
+            for interest in interests
+        ):
+            raise serializers.ValidationError("Invalid interests selected.")
+        return list(set(interests))
+
+    def validate_choosen_endgoals(self, end_goals):
+        if not all(
+            goal
+            in (
+                "job",
+                "higher_education",
+                "gig_work",
+                "entrepreneurship",
+                "other",
+            )
+            for goal in end_goals
+        ):
+            raise serializers.ValidationError("Invalid end goals selected.")
+        return list(set(end_goals))
+
+    class Meta:
+        model = UserInterests
+        fields = [
+            "id",
+            "user",
+            "choosen_interests",
+            "other_interests",
+            "choosen_endgoals",
+            "other_endgoals",
+            "created_at",
+            "updated_at",
+        ]

api/register/urls.py

@@ -8,9 +8,7 @@
     path("role/list/", register_views.RoleAPI.as_view()),
     path("colleges/", register_views.CollegesAPI.as_view()),
     path("department/list/", register_views.DepartmentAPI.as_view()),
-
     path("location/", register_views.LocationSearchView.as_view()),
-
     path("country/list/", register_views.CountryAPI.as_view()),
     path("state/list/", register_views.StateAPI.as_view()),
     path("district/list/", register_views.DistrictAPI.as_view()),
@@ -20,9 +18,10 @@
     path("schools/list/", register_views.SchoolAPI.as_view()),
     path("area-of-interest/list/", register_views.AreaOfInterestAPI.as_view()),
     path("lc/user-validation/", register_views.LearningCircleUserViewAPI.as_view()),
-    path('email-verification/', register_views.UserEmailVerificationAPI.as_view()),
-    path('user-country/', register_views.UserCountryAPI.as_view()),
-    path('user-state/', register_views.UserStateAPI.as_view()),
-    path('user-zone/', register_views.UserZoneAPI.as_view()),
-
-]
+    path("email-verification/", register_views.UserEmailVerificationAPI.as_view()),
+    path("user-country/", register_views.UserCountryAPI.as_view()),
+    path("user-state/", register_views.UserStateAPI.as_view()),
+    path("user-zone/", register_views.UserZoneAPI.as_view()),
+    path("interests /", register_views.UserInterestAPI.as_view()),
+    # path("connect-discord/", register_views.ConnectDiscordAPI.as_view()),
+]

db/user.py

@@ -6,6 +6,7 @@
 from django.conf import settings
 
 from .managers import user_manager
+
 # from .task import UserIgLink
 from decouple import config as decouple_config
 
@@ -56,6 +57,19 @@ def save(self, *args, **kwargs):
 
         return super().save(*args, **kwargs)
 
+class UserInterests(models.Model):
+    id = models.CharField(primary_key=True, max_length=36, default=uuid.uuid4)
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    choosen_interests = models.JSONField(max_length=100)
+    other_interests = models.JSONField(max_length=100, blank=True, null=True)
+    choosen_endgoals = models.JSONField(max_length=100)
+    other_endgoals = models.JSONField(max_length=100, blank=True, null=True)
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+
+    class Meta:
+        managed = False
+        db_table = 'user_interests'
 
 class UserMentor(models.Model):
     id = models.CharField(primary_key=True, max_length=36, default=uuid.uuid4)