Add CurrentUserPrograms view

Author: yakser

partner_programs/serializers.py

@@ -126,3 +126,13 @@ class PartnerProgramUserSerializer(serializers.Serializer):
 
 class PartnerProgramDataSchemaSerializer(serializers.Serializer):
     data_schema = serializers.JSONField(required=True)
+
+
+class UserProgramsSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = PartnerProgram
+        fields = [
+            "id",
+            "name",
+            "tag",
+        ]

users/urls.py

@@ -18,6 +18,7 @@
     RegisteredEventsList,
     SetUserOnboardingStage,
     ResendVerifyEmail,
+    CurrentUserPrograms,
 )
 
 app_name = "users"
@@ -35,6 +36,7 @@
     path("users/<int:pk>/set_onboarding_stage/", SetUserOnboardingStage.as_view()),
     path("users/reset-password/", EmailResetPassword.as_view()),
     path("users/current/", CurrentUser.as_view()),
+    path("users/current/programs/", CurrentUserPrograms.as_view()),
     path("users/current/events/", RegisteredEventsList.as_view()),
     path("users/achievements/", AchievementList.as_view()),
     path("users/achievements/<int:pk>/", AchievementDetail.as_view()),

users/views.py

@@ -15,6 +15,7 @@
     ListCreateAPIView,
     RetrieveUpdateDestroyAPIView,
     UpdateAPIView,
+    RetrieveAPIView,
 )
 from rest_framework.permissions import AllowAny, IsAuthenticated
 from rest_framework.request import Request
@@ -25,6 +26,8 @@
 from core.permissions import IsOwnerOrReadOnly
 from events.models import Event
 from events.serializers import EventsListSerializer
+from partner_programs.models import PartnerProgram
+from partner_programs.serializers import UserProgramsSerializer
 from projects.serializers import ProjectListSerializer
 from users.helpers import (
     reset_email,
@@ -126,6 +129,20 @@ def patch(self, request, pk):
         return super().patch(request, pk)
 
 
+class CurrentUserPrograms(RetrieveAPIView):
+    queryset = PartnerProgram.objects.all()
+    permission_classes = [IsAuthenticated]
+    serializer_class = UserProgramsSerializer
+
+    def get(self, request, *args, **kwargs):
+        user = User.objects.get(id=request.user.id)
+        programs = [
+            profile.partner_program for profile in user.partner_program_profiles.all()
+        ]
+        serializer = self.get_serializer(programs, many=True)
+        return Response(serializer.data, status=status.HTTP_200_OK)
+
+
 class CurrentUser(GenericAPIView):
     queryset = User.objects.get_users_for_detail_view()
     permission_classes = [IsAuthenticated]