Assign project to partner program

Author: yakser

partner_programs/admin.py

@@ -47,6 +47,7 @@ class PartnerProgramUserProfileAdmin(admin.ModelAdmin):
     list_filter = ("partner_program",)
     raw_id_fields = (
         "user",
+        "project",
         "partner_program",
     )
     date_hierarchy = "datetime_created"

projects/helpers.py

@@ -2,6 +2,7 @@
 
 from django.contrib.auth import get_user_model
 
+from partner_programs.models import PartnerProgram, PartnerProgramUserProfile
 from projects.constants import RECOMMENDATIONS_COUNT
 from projects.models import Project, ProjectLink, Achievement
 
@@ -84,3 +85,16 @@ def update_links(links, pk):
             for link in links
         ]
     )
+
+
+def update_partner_program(
+    partner_program_id: int, user: "User", instance: "Project"
+) -> None:
+    if partner_program_id:
+        partner_program = PartnerProgram.objects.get(pk=partner_program_id)
+        partner_program_profile = PartnerProgramUserProfile.objects.get(
+            user=user,
+            partner_program=partner_program,
+        )
+        partner_program_profile.project = instance
+        partner_program_profile.save()

projects/views.py

@@ -9,9 +9,14 @@
 from core.permissions import IsStaffOrReadOnly
 from core.serializers import SetLikedSerializer
 from core.services import add_view, set_like
+from partner_programs.models import PartnerProgram, PartnerProgramUserProfile
 from projects.filters import ProjectFilter
 from projects.constants import VERBOSE_STEPS
-from projects.helpers import get_recommended_users, check_related_fields_update
+from projects.helpers import (
+    get_recommended_users,
+    check_related_fields_update,
+    update_partner_program,
+)
 from projects.models import Project, Achievement, ProjectNews
 from projects.pagination import ProjectNewsPagination
 from projects.permissions import (
@@ -51,6 +56,21 @@ def create(self, request, *args, **kwargs):
         serializer.validated_data["leader"] = request.user
 
         self.perform_create(serializer)
+
+        try:
+            partner_program_id = request.data.get("partner_program_id")
+            update_partner_program(partner_program_id, request.user, serializer.instance)
+        except PartnerProgram.DoesNotExist:
+            return Response(
+                {"detail": "Partner program with this id does not exist"},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+        except PartnerProgramUserProfile.DoesNotExist:
+            return Response(
+                {"detail": "User is not a member of this partner program"},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+
         headers = self.get_success_headers(serializer.data)
         return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
 
@@ -101,6 +121,20 @@ def retrieve(self, request, *args, **kwargs):
         return Response(serializer.data)
 
     def put(self, request, pk, **kwargs):
+        # fixme: add partner_program_id to docs
+        try:
+            partner_program_id = request.data.get("partner_program_id")
+            update_partner_program(partner_program_id, request.user, self.get_object())
+        except PartnerProgram.DoesNotExist:
+            return Response(
+                {"detail": "Partner program with this id does not exist"},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+        except PartnerProgramUserProfile.DoesNotExist:
+            return Response(
+                {"detail": "User is not a member of this partner program"},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
         check_related_fields_update(request.data, pk)
         return super(ProjectDetail, self).put(request, pk)
 

users/views.py

@@ -136,6 +136,7 @@ class CurrentUserPrograms(RetrieveAPIView):
 
     def get(self, request, *args, **kwargs):
         user = User.objects.get(id=request.user.id)
+        # fixme: mb hide finished programs
         programs = [
             profile.partner_program for profile in user.partner_program_profiles.all()
         ]