lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Escape illegal XML in Feed

Browse source
This commit is contained in:
cuom1999 2021-01-07 20:22:53 -06:00
parent 236cc7051b
commit dbaf181a50
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
judge/feed.py
View file

@ -8,6 +8,13 @@ from django.utils.feedgenerator import Atom1Feed
from judge.jinja2.markdown import markdown from judge.jinja2.markdown import markdown
from judge.models import BlogPost, Comment, Problem 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): class ProblemFeed(Feed):
title = 'Recently Added %s Problems' % settings.SITE_NAME title = 'Recently Added %s Problems' % settings.SITE_NAME
@ -86,6 +93,7 @@ class BlogFeed(Feed):
summary = cache.get(key) summary = cache.get(key)
if summary is None: if summary is None:
summary = str(markdown(post.summary or post.content, 'blog')) summary = str(markdown(post.summary or post.content, 'blog'))
summary = escape_xml_illegal_chars(summary)
cache.set(key, summary, 86400) cache.set(key, summary, 86400)
return summary return summary