From c670b3a17b3362b072c170cf6ab75d550f2e0a2f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jes=C3=BAs=20Espino?= Date: Sat, 26 Jul 2014 10:23:11 +0200 Subject: [PATCH] Small optimization on project stats --- taiga/projects/milestones/models.py | 2 +- taiga/projects/models.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/taiga/projects/milestones/models.py b/taiga/projects/milestones/models.py index 0c7161d1..684a62ff 100644 --- a/taiga/projects/milestones/models.py +++ b/taiga/projects/milestones/models.py @@ -72,7 +72,7 @@ class Milestone(WatchedModelMixin, models.Model): super().save(*args, **kwargs) def _get_user_stories_points(self, user_stories): - role_points = [us.role_points.all() for us in user_stories] + role_points = [us.role_points.all().select_related('points') for us in user_stories] flat_role_points = itertools.chain(*role_points) flat_role_dicts = map(lambda x: {x.role_id: x.points.value if x.points.value else 0}, flat_role_points) return dict_sum(*flat_role_dicts) diff --git a/taiga/projects/models.py b/taiga/projects/models.py index 0ccd51ca..766fcfdf 100644 --- a/taiga/projects/models.py +++ b/taiga/projects/models.py @@ -237,7 +237,7 @@ class Project(ProjectDefaults, TaggedMixin, models.Model): rp_query.delete() def _get_user_stories_points(self, user_stories): - role_points = [us.role_points.all() for us in user_stories] + role_points = [us.role_points.all().select_related('points') for us in user_stories] flat_role_points = itertools.chain(*role_points) flat_role_dicts = map(lambda x: {x.role_id: x.points.value if x.points.value else 0}, flat_role_points) return dict_sum(*flat_role_dicts)