From 4368323165138ec45656c0a80ce4489aa130eeb2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?David=20Barrag=C3=A1n=20Merino?= Date: Tue, 24 Feb 2015 20:43:59 +0100 Subject: [PATCH] US #55: Custom fields - Improve API queries --- taiga/projects/custom_attributes/api.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/taiga/projects/custom_attributes/api.py b/taiga/projects/custom_attributes/api.py index d0868690..f87edf54 100644 --- a/taiga/projects/custom_attributes/api.py +++ b/taiga/projects/custom_attributes/api.py @@ -85,6 +85,11 @@ class UserStoryCustomAttributesValuesViewSet(BaseCustomAttributesValuesViewSet): lookup_field = "user_story_id" content_object = "user_story" + def get_queryset(self): + qs = self.model.objects.all() + qs = qs.select_related("user_story", "user_story__project") + return qs + class TaskCustomAttributesValuesViewSet(BaseCustomAttributesValuesViewSet): model = models.TaskCustomAttributesValues @@ -93,6 +98,11 @@ class TaskCustomAttributesValuesViewSet(BaseCustomAttributesValuesViewSet): lockup_fields = "task_id" content_object = "task" + def get_queryset(self): + qs = self.model.objects.all() + qs = qs.select_related("task", "task__project") + return qs + class IssueCustomAttributesValuesViewSet(BaseCustomAttributesValuesViewSet): model = models.IssueCustomAttributesValues @@ -100,3 +110,8 @@ class IssueCustomAttributesValuesViewSet(BaseCustomAttributesValuesViewSet): permission_classes = (permissions.IssueCustomAttributesValuesPermission,) lockup_fields = "issue_id" content_object = "issue" + + def get_queryset(self): + qs = self.model.objects.all() + qs = qs.select_related("issue", "issue__project") + return qs