fixing user.projcts.[role] at /auth/user/{id} route

Author: VeryBigSad

projects/managers.py

@@ -4,6 +4,16 @@
 User = get_user_model()
 
 
+class CollaboratorManager(Manager):
+    def get_draft_projects_for_user(self):
+        return (
+            super()
+            .get_queryset()
+            .select_related("project")
+            .filter(project__is_draft=True)
+        )
+
+
 class ProjectManager(Manager):
     def get_projects_for_list_view(self):
         return self.get_queryset().filter(draft=False)

projects/models.py

@@ -9,7 +9,7 @@
 from files.models import UserFile
 from industries.models import Industry
 from projects.constants import VERBOSE_STEPS
-from projects.managers import AchievementManager, ProjectManager
+from projects.managers import AchievementManager, CollaboratorManager, ProjectManager
 from users.models import CustomUser
 
 User = get_user_model()
@@ -215,6 +215,8 @@ class Collaborator(models.Model):
         verbose_name="Дата изменения", null=False, auto_now=True
     )
 
+    objects = CollaboratorManager()
+
     class Meta:
         verbose_name = "Коллаборатор"
         verbose_name_plural = "Коллабораторы"

users/serializers.py

@@ -5,7 +5,6 @@
 from core.services import get_views_count
 from core.utils import get_user_online_cache_key
 from projects.models import Project
-from projects.serializers import CollaboratorSerializer
 from .models import CustomUser, Expert, Investor, Member, Mentor, UserAchievement
 
 
@@ -90,17 +89,13 @@ class UserDetailSerializer(serializers.ModelSerializer):
     is_online = serializers.SerializerMethodField()
     projects = serializers.SerializerMethodField()
 
-    @classmethod
-    def get_projects(cls, user: CustomUser):
+    def get_projects(self, user: CustomUser):
         return UserProjectsSerializer(
             [
                 collab.project
-                for collab in user.collaborations.select_related("project").filter(
-                    project__is_draft=False
-                )
-                # TODO: put this in user collaborations manager or something
+                for collab in user.collaborations.filter(project__draft=False)
             ],
-            context={"request": cls.context.get("request")},
+            context={"request": self.context.get("request")},
             many=True,
         ).data
 
@@ -222,6 +217,9 @@ class UserProjectsSerializer(serializers.ModelSerializer):
 
     @classmethod
     def get_collaborator(cls, project):
+        # TODO: fix me, import in a functon
+        from projects.serializers import CollaboratorSerializer
+
         user = cls.context.get("request").user
 
         return CollaboratorSerializer(