add project specific news feeds

1 files changed, 26 insertions, 0 deletions

diff --git a/sublab_project/projects/feeds.py b/sublab_project/projects/feeds.py
new file mode 100644
index 0000000..dfc1595
--- /dev/null
+++ b/sublab_project/projects/feeds.py
@@ -0,0 +1,26 @@
+from django.utils.feedgenerator import Atom1Feed
+from django.shortcuts import get_object_or_404
+
+from news.models import News
+from projects.models import Project
+from news.feeds import NewsFeed
+
+class ProjectNewsFeed(NewsFeed):
+    def get_object(self, *args, **kwargs):
+        return get_object_or_404(Project, **kwargs)
+
+    def title(self, obj):
+        return 'sublab.org - %s' % obj.name
+
+    def description(self, obj):
+        return obj.description
+
+    def link(self, obj):
+        return obj.get_absolute_url()
+
+    def items(self, obj):
+        return obj.news.all()[:10]
+
+class ProjectNewsFeedAtom(ProjectNewsFeed):
+    feed_type = Atom1Feed
+    subtitle = ProjectNewsFeed.description