Changeset 402:801a271ca7dc for djangobb_forum/models.py

Timestamp:

Mar 4, 2012 3:35:02 PM (16 months ago)

Author:

Igor Yanchenko <yanchenko.igor@…>

Branch:

default

Message:

moved reply_count_minus and reply_count_plus to manager and use_for_related_fields set True

File:

• djangobb_forum/models.py (modified) (2 diffs)

djangobb_forum/models.py

@@ -316 +316 @@
 
 class ProfileManager(models.Manager):
+    use_for_related_fields = True
     def get_query_set(self):
         qs = super(ProfileManager, self).get_query_set()
         if forum_settings.REPUTATION_SUPPORT:
-            qs = qs.extra(select={'reply_total':'Select sum(sign) from djangobb_forum_reputation where to_user_id = djangobb_forum_profile.user_id group by to_user_id'})
+            qs = qs.extra(select={
+                'reply_total':'Select sum(sign) from djangobb_forum_reputation where to_user_id = djangobb_forum_profile.user_id group by to_user_id',
+                'reply_count_minus':"Select sum(sign) from djangobb_forum_reputation where to_user_id = djangobb_forum_profile.user_id and sign = '-1' group by to_user_id",
+                'reply_count_plus':"Select sum(sign) from djangobb_forum_reputation where to_user_id = djangobb_forum_profile.user_id and sign = '1' group by to_user_id",
+                })
         return qs
 
@@ -364 +369 @@
             return  None
 
-    def reply_count_minus(self):
-        return Reputation.objects.filter(to_user__id=self.user_id, sign=-1).count()
-
-    def reply_count_plus(self):
-        return Reputation.objects.filter(to_user__id=self.user_id, sign=1).count()
-
-
 class PostTracking(models.Model):
     """