45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
from django.conf import settings
|
|
from django.contrib.comments.models import Comment, FreeComment
|
|
from django.contrib.syndication.feeds import Feed
|
|
from django.contrib.sites.models import Site
|
|
|
|
class LatestFreeCommentsFeed(Feed):
|
|
"""Feed of latest free comments on the current site."""
|
|
|
|
comments_class = FreeComment
|
|
|
|
def title(self):
|
|
if not hasattr(self, '_site'):
|
|
self._site = Site.objects.get_current()
|
|
return u"%s comments" % self._site.name
|
|
|
|
def link(self):
|
|
if not hasattr(self, '_site'):
|
|
self._site = Site.objects.get_current()
|
|
return "http://%s/" % (self._site.domain)
|
|
|
|
def description(self):
|
|
if not hasattr(self, '_site'):
|
|
self._site = Site.objects.get_current()
|
|
return u"Latest comments on %s" % self._site.name
|
|
|
|
def get_query_set(self):
|
|
return self.comments_class.objects.filter(site__pk=settings.SITE_ID, is_public=True)
|
|
|
|
def items(self):
|
|
return self.get_query_set()[:40]
|
|
|
|
class LatestCommentsFeed(LatestFreeCommentsFeed):
|
|
"""Feed of latest comments on the current site."""
|
|
|
|
comments_class = Comment
|
|
|
|
def get_query_set(self):
|
|
qs = super(LatestCommentsFeed, self).get_query_set()
|
|
qs = qs.filter(is_removed=False)
|
|
if settings.COMMENTS_BANNED_USERS_GROUP:
|
|
where = ['user_id NOT IN (SELECT user_id FROM auth_users_group WHERE group_id = %s)']
|
|
params = [settings.COMMENTS_BANNED_USERS_GROUP]
|
|
qs = qs.extra(where=where, params=params)
|
|
return qs
|