Refactor feed generation a bit

Author: nomeata

codespeed/feeds.py

@@ -1,25 +1,26 @@
 from django.contrib.syndication.views import Feed
 from codespeed.models import Report
 from django.conf import settings
+from django.db.models import Q
 
-
-class LatestEntries(Feed):
+class ResultFeed(Feed):
     title = settings.WEBSITE_NAME
     link = "/changes/"
-    description = "Last benchmark runs"
 
     def items(self):
-        return Report.objects.filter(
-            revision__branch__name=settings.DEF_BRANCH
-        ).order_by('-revision__date')[:10]
+        return Report.objects\
+            .filter(self.result_filter())\
+            .order_by('-revision__date')[:10]
 
-class LatestSignificantEntries(Feed):
-    title = settings.WEBSITE_NAME
-    link = "/changes/"
+class LatestEntries(ResultFeed):
+    description = "Last benchmark runs"
+
+    def result_filter(self):
+        return Q(revision__branch__name=settings.DEF_BRANCH)
+
+class LatestSignificantEntries(ResultFeed):
     description = "Last significant benchmark runs"
 
-    def items(self):
-        return Report.objects.filter(
-            revision__branch__name=settings.DEF_BRANCH,
-            colorcode__in = ('red','green')
-        ).order_by('-revision__date')[:10]
+    def result_filter(self):
+        return Q(revision__branch__name=settings.DEF_BRANCH,
+                 colorcode__in = ('red','green'))