110 lines
3.4 KiB
Python
110 lines
3.4 KiB
Python
from django.conf import settings
|
|
from django.contrib.auth.models import AnonymousUser
|
|
from django.contrib.syndication.views import Feed
|
|
from django.core.cache import cache
|
|
from django.utils import timezone
|
|
from django.utils.feedgenerator import Atom1Feed
|
|
|
|
from judge.jinja2.markdown import markdown
|
|
from judge.models import BlogPost, Comment, Problem
|
|
|
|
import re
|
|
|
|
|
|
# https://lsimons.wordpress.com/2011/03/17/stripping-illegal-characters-out-of-xml-in-python/
|
|
def escape_xml_illegal_chars(val, replacement='?'):
|
|
_illegal_xml_chars_RE = re.compile(u'[\x00-\x08\x0b\x0c\x0e-\x1F\uD800-\uDFFF\uFFFE\uFFFF]')
|
|
return _illegal_xml_chars_RE.sub(replacement, val)
|
|
|
|
class ProblemFeed(Feed):
|
|
title = 'Recently Added %s Problems' % settings.SITE_NAME
|
|
link = '/'
|
|
description = 'The latest problems added on the %s website' % settings.SITE_LONG_NAME
|
|
|
|
def items(self):
|
|
return Problem.objects.filter(is_public=True, is_organization_private=False).defer('description')\
|
|
.order_by('-date', '-id')[:25]
|
|
|
|
def item_title(self, problem):
|
|
return problem.name
|
|
|
|
def item_description(self, problem):
|
|
key = 'problem_feed:%d' % problem.id
|
|
desc = cache.get(key)
|
|
if desc is None:
|
|
desc = str(markdown(problem.description, 'problem'))[:500] + '...'
|
|
desc = escape_xml_illegal_chars(desc)
|
|
cache.set(key, desc, 86400)
|
|
return desc
|
|
|
|
def item_pubdate(self, problem):
|
|
return problem.date
|
|
|
|
item_updateddate = item_pubdate
|
|
|
|
|
|
class AtomProblemFeed(ProblemFeed):
|
|
feed_type = Atom1Feed
|
|
subtitle = ProblemFeed.description
|
|
|
|
|
|
class CommentFeed(Feed):
|
|
title = 'Latest %s Comments' % settings.SITE_NAME
|
|
link = '/'
|
|
description = 'The latest comments on the %s website' % settings.SITE_LONG_NAME
|
|
|
|
def items(self):
|
|
return Comment.most_recent(AnonymousUser(), 25)
|
|
|
|
def item_title(self, comment):
|
|
return '%s -> %s' % (comment.author.user.username, comment.page_title)
|
|
|
|
def item_description(self, comment):
|
|
key = 'comment_feed:%d' % comment.id
|
|
desc = cache.get(key)
|
|
if desc is None:
|
|
desc = str(markdown(comment.body, 'comment'))
|
|
desc = escape_xml_illegal_chars(desc)
|
|
cache.set(key, desc, 86400)
|
|
return desc
|
|
|
|
def item_pubdate(self, comment):
|
|
return comment.time
|
|
|
|
item_updateddate = item_pubdate
|
|
|
|
|
|
class AtomCommentFeed(CommentFeed):
|
|
feed_type = Atom1Feed
|
|
subtitle = CommentFeed.description
|
|
|
|
|
|
class BlogFeed(Feed):
|
|
title = 'Latest %s Blog Posts' % settings.SITE_NAME
|
|
link = '/'
|
|
description = 'The latest blog posts from the %s' % settings.SITE_LONG_NAME
|
|
|
|
def items(self):
|
|
return BlogPost.objects.filter(visible=True, publish_on__lte=timezone.now()).order_by('-sticky', '-publish_on')
|
|
|
|
def item_title(self, post):
|
|
return post.title
|
|
|
|
def item_description(self, post):
|
|
key = 'blog_feed:%d' % post.id
|
|
summary = cache.get(key)
|
|
if summary is None:
|
|
summary = str(markdown(post.summary or post.content, 'blog'))
|
|
summary = escape_xml_illegal_chars(summary)
|
|
cache.set(key, summary, 86400)
|
|
return summary
|
|
|
|
def item_pubdate(self, post):
|
|
return post.publish_on
|
|
|
|
item_updateddate = item_pubdate
|
|
|
|
|
|
class AtomBlogFeed(BlogFeed):
|
|
feed_type = Atom1Feed
|
|
subtitle = BlogFeed.description
|