| 1 |
from django.conf import settings |
|---|
| 2 |
from django.contrib.comments.models import Comment, FreeComment |
|---|
| 3 |
from django.contrib.syndication.feeds import Feed |
|---|
| 4 |
from django.contrib.sites.models import Site |
|---|
| 5 |
|
|---|
| 6 |
class LatestFreeCommentsFeed(Feed): |
|---|
| 7 |
"""Feed of latest free comments on the current site.""" |
|---|
| 8 |
|
|---|
| 9 |
comments_class = FreeComment |
|---|
| 10 |
|
|---|
| 11 |
def title(self): |
|---|
| 12 |
if not hasattr(self, '_site'): |
|---|
| 13 |
self._site = Site.objects.get_current() |
|---|
| 14 |
return u"%s comments" % self._site.name |
|---|
| 15 |
|
|---|
| 16 |
def link(self): |
|---|
| 17 |
if not hasattr(self, '_site'): |
|---|
| 18 |
self._site = Site.objects.get_current() |
|---|
| 19 |
return "http://%s/" % (self._site.domain) |
|---|
| 20 |
|
|---|
| 21 |
def description(self): |
|---|
| 22 |
if not hasattr(self, '_site'): |
|---|
| 23 |
self._site = Site.objects.get_current() |
|---|
| 24 |
return u"Latest comments on %s" % self._site.name |
|---|
| 25 |
|
|---|
| 26 |
def get_query_set(self): |
|---|
| 27 |
return self.comments_class.objects.filter(site__pk=settings.SITE_ID, is_public=True) |
|---|
| 28 |
|
|---|
| 29 |
def items(self): |
|---|
| 30 |
return self.get_query_set()[:40] |
|---|
| 31 |
|
|---|
| 32 |
class LatestCommentsFeed(LatestFreeCommentsFeed): |
|---|
| 33 |
"""Feed of latest comments on the current site.""" |
|---|
| 34 |
|
|---|
| 35 |
comments_class = Comment |
|---|
| 36 |
|
|---|
| 37 |
def get_query_set(self): |
|---|
| 38 |
qs = super(LatestCommentsFeed, self).get_query_set() |
|---|
| 39 |
qs = qs.filter(is_removed=False) |
|---|
| 40 |
if settings.COMMENTS_BANNED_USERS_GROUP: |
|---|
| 41 |
where = ['user_id NOT IN (SELECT user_id FROM auth_users_group WHERE group_id = %s)'] |
|---|
| 42 |
params = [settings.COMMENTS_BANNED_USERS_GROUP] |
|---|
| 43 |
qs = qs.extra(where=where, params=params) |
|---|
| 44 |
return qs |
|---|