Initial commit from ZIP

evg-outbo-api/outbo/apps/projects/viewsets.py

@@ -0,0 +1,82 @@
+from django.db.models import Q
+from django_filters import rest_framework as filters
+
+from outbo.apps.core.viewsets import (
+    AutoPermissionViewSetMixin,
+    MultipleSerializerViewSet,
+    CreateModelMixin,
+    PartialUpdateModelMixin,
+    RetrieveModelMixin,
+    ListModelMixin,
+    DestroyModelMixin,
+    NestedViewSetMixin,
+)
+from outbo.apps.projects.filters import ProjectFilterSet
+from outbo.apps.projects.models import Project, Member
+from outbo.apps.projects.models.member import MembershipType
+from outbo.apps.projects.queries import read_project_query
+from outbo.apps.projects.serializers import ProjectSerializer, MemberSerializer
+
+
+class ProjectViewSet(
+    RetrieveModelMixin,
+    ListModelMixin,
+    CreateModelMixin,
+    PartialUpdateModelMixin,
+    AutoPermissionViewSetMixin,
+    MultipleSerializerViewSet,
+):
+    queryset = Project.objects.all()
+    serializer_class_set = ProjectSerializer
+    filter_backends = (filters.DjangoFilterBackend,)
+    filterset_class = ProjectFilterSet
+
+    def get_queryset(self):
+        qs = super().get_queryset()
+        if self.action == "list":
+            qs_filter = read_project_query(self.request.user)
+            qs = qs.filter(qs_filter)
+            # TODO: try to refactor this poor loops
+            if self.request.user.is_observer or self.request.user.is_trader:
+                projects_id = []
+                for project in Project.objects.filter(published=True):
+                    members = project.members.filter(role=self.request.user.user_role)
+                    for member in members:
+                        if member.user.organization == self.request.user.organization:
+                            projects_id.append(project.id)
+                qs_id = qs.values_list("id", flat=True)
+                qs = Project.objects.filter(Q(id__in=projects_id) | Q(id__in=qs_id))
+            if self.request.user.is_trader and self.request.user.category is not None:
+                qs = qs.filter(category=self.request.user.category)
+        return qs
+
+    def partial_update(self, request, *args, **kwargs):
+        return super().partial_update(request=request, *args, **kwargs)
+
+
+class ProjectMemberNestedViewSet(
+    ListModelMixin,
+    CreateModelMixin,
+    DestroyModelMixin,
+    NestedViewSetMixin,
+    AutoPermissionViewSetMixin,
+    MultipleSerializerViewSet,
+):
+    queryset = Member.objects.all()
+    serializer_class_set = MemberSerializer
+
+    def perform_create(self, *, serializer: MemberSerializer.Create):
+        user = serializer.validated_data["user"]
+        role = None
+        if user.is_agency:
+            role = MembershipType.AGENCY
+        elif user.is_client:
+            role = MembershipType.CLIENT
+        elif user.is_supplier:
+            role = MembershipType.SUPPLIER
+        parents = self.get_parents_query_dict()
+        project = Project.objects.get(pk=parents["project_id"])
+        membership = Member.objects.filter(project=project, user=user).first()
+        if membership:
+            serializer.instance = membership
+        return serializer.save(project=project, role=role)