This commit is contained in:
behruz-dev
2025-10-15 14:23:28 +05:00
parent 5dfda150f2
commit ffefa6ff67

View File

@@ -1,4 +1,5 @@
from django.db import transaction
from django.db.models import Q
from rest_framework import serializers
@@ -293,8 +294,20 @@ class ProjectsSerializer(serializers.ModelSerializer):
class ProjectFoldersSerializer(serializers.ModelSerializer):
projects = ProjectsSerializer(many=True)
projects = serializers.SerializerMethodField(method_name='get_projects')
class Meta:
model = ProjectFolder
fields = ['id', 'name', 'projects']
def get_projects(self, obj):
user = self.context.get('user')
is_mine = self.context.get('is_mine')
if is_mine and is_mine.lower() == 'true':
projects = obj.projects.filter(
Q(boss=user) |
Q(foreman=user) |
Q(other_members=user)
).distinct()
return ProjectsSerializer(projects, many=True).data
return ProjectsSerializer(obj.projects, many=True).data